read multiple keys with index (googleapis#4386)

chemelnucfin · web-flow · commit 3fb9f167502e · 2017-12-05T13:09:38.000-08:00
diff --git a/spanner/tests/system/test_system.py b/spanner/tests/system/test_system.py
@@ -758,6 +758,19 @@ def test_empty_read_with_single_keys_index(self):
         rows = list(results_iter)
         self.assertEqual(rows, [])
 
+    def test_read_with_multiple_keys_index(self):
+        row_count = 10
+        columns = self.COLUMNS[1], self.COLUMNS[2]
+        session, committed = self._set_up_table(row_count)
+        self.to_delete.append(session)
+        expected = [[row[1], row[2]] for row in self._row_data(row_count)]
+        rows = list(session.read(self.TABLE,
+                                 columns,
+                                 KeySet(keys=expected),
+                                 index='name')
+        )
+        self.assertEqual(rows, expected)
+
     def test_snapshot_read_w_various_staleness(self):
         from datetime import datetime
         from google.cloud._helpers import UTC
