From 2ea8daab4c01f82ece794d20a547cb4debb4aae4 Mon Sep 17 00:00:00 2001 From: Alan Wu Date: Sun, 12 Nov 2017 17:04:28 -0800 Subject: [PATCH] read multiple keys with index --- spanner/tests/system/test_system.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/spanner/tests/system/test_system.py b/spanner/tests/system/test_system.py index f0c9f9d5f6ee..6b345ad6c1fe 100644 --- 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