Update system tests for proper multi-use snapshot semantics.

googleapis · Jul 24, 2017 · 5dc09f8 · 5dc09f8
1 parent 8244af9
commit 5dc09f8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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.