Add system tests exercising multi-use snapshots.

tseaver · tseaver · commit 26bdbc5bd095 · 2017-07-17T14:33:48.000-04:00
diff --git a/spanner/tests/system/test_system.py b/spanner/tests/system/test_system.py
@@ -574,6 +574,21 @@ def test_snapshot_read_w_various_staleness(self):
         rows = list(strong.read(self.TABLE, self.COLUMNS, self.ALL))
         self._check_row_data(rows, all_data_rows)
 
+    def test_multiuse_snapshot_read_isolation_strong(self):
+        ROW_COUNT = 40
+        session, committed = self._set_up_table(ROW_COUNT)
+        all_data_rows = list(self._row_data(ROW_COUNT))
+        strong = session.snapshot(multi_use=True)
+
+        before = list(strong.read(self.TABLE, self.COLUMNS, self.ALL))
+        self._check_row_data(before, all_data_rows)
+
+        with self._db.batch() as batch:
+            batch.delete(self.TABLE, self.ALL)
+
+        after = list(strong.read(self.TABLE, self.COLUMNS, self.ALL))
+        self._check_row_data(after, all_data_rows)
+
     def test_read_w_manual_consume(self):
         ROW_COUNT = 4000
         session, committed = self._set_up_table(ROW_COUNT)
@@ -723,6 +738,22 @@ def _check_sql_results(self, snapshot, sql, params, param_types, expected):
             sql, params=params, param_types=param_types))
         self._check_row_data(rows, expected=expected)
 
+    def test_multiuse_snapshot_execute_sql_isolation_strong(self):
+        ROW_COUNT = 40
+        SQL = 'SELECT * FROM {}'.format(self.TABLE)
+        session, committed = self._set_up_table(ROW_COUNT)
+        all_data_rows = list(self._row_data(ROW_COUNT))
+        strong = session.snapshot(multi_use=True)
+
+        before = list(strong.execute_sql(SQL))
+        self._check_row_data(before, all_data_rows)
+
+        with self._db.batch() as batch:
+            batch.delete(self.TABLE, self.ALL)
+
+        after = list(strong.execute_sql(SQL))
+        self._check_row_data(after, all_data_rows)
+
     def test_execute_sql_returning_array_of_struct(self):
         SQL = (
             "SELECT ARRAY(SELECT AS STRUCT C1, C2 "
