From 5dc09f808a61b5e8ba71e9a64aae5840a2a49027 Mon Sep 17 00:00:00 2001 From: Tres Seaver Date: Mon, 24 Jul 2017 15:14:20 -0400 Subject: [PATCH] Update system tests for proper multi-use snapshot semantics. --- spanner/tests/system/test_system.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/spanner/tests/system/test_system.py b/spanner/tests/system/test_system.py index 7228cc2771b6..28ef66862313 100644 --- a/spanner/tests/system/test_system.py +++ b/spanner/tests/system/test_system.py @@ -828,7 +828,7 @@ def test_read_w_ranges(self): START = 1000 END = 2000 session, committed = self._set_up_table(ROW_COUNT) - snapshot = session.snapshot(read_timestamp=committed) + snapshot = session.snapshot(read_timestamp=committed, multi_use=True) all_data_rows = list(self._row_data(ROW_COUNT)) closed_closed = KeyRange(start_closed=[START], end_closed=[END]) @@ -934,7 +934,8 @@ def test_execute_sql_w_query_param(self): self.ALL_TYPES_COLUMNS, self.ALL_TYPES_ROWDATA) - snapshot = session.snapshot(read_timestamp=batch.committed) + snapshot = session.snapshot( + read_timestamp=batch.committed, multi_use=True) # Cannot equality-test array values. See below for a test w/ # array of IDs.