Add spanner system tests: (googleapis#4371)

tseaver · web-flow · commit 2bf694b1c779 · 2017-11-10T11:40:46.000-05:00
- Bind timestamp.
- Bind string array.
- Bind empty string array.
- Bind string to null.
- Query returning array of -Inf, +Inf, NaN.
diff --git a/spanner/tests/system/test_system.py b/spanner/tests/system/test_system.py
@@ -399,9 +399,9 @@ class TestSessionAPI(unittest.TestCase, _TestData):
         ([1], True, BYTES_1, SOME_DATE, 0.0, 19, u'dog', SOME_TIME),
         ([5, 10], True, BYTES_1, None, 1.25, 99, u'cat', None),
         ([], False, BYTES_2, None, float('inf'), 107, u'frog', None),
-        ([3, None, 9], False, None, None, float('-inf'), 207, None, None),
-        ([], False, None, None, float('nan'), 1207, None, None),
-        ([], False, None, None, OTHER_NAN, 2000, None, NANO_TIME),
+        ([3, None, 9], False, None, None, float('-inf'), 207, u'bat', None),
+        ([], False, None, None, float('nan'), 1207, u'owl', None),
+        ([], False, None, None, OTHER_NAN, 2000, u'virus', NANO_TIME),
     )
 
     @classmethod
@@ -956,8 +956,9 @@ def test_execute_sql_w_manual_consume(self):
         self.assertEqual(streamed._current_row, [])
         self.assertEqual(streamed._pending_chunk, None)
 
-    def _check_sql_results(self, snapshot, sql, params, param_types, expected):
-        if 'ORDER' not in sql:
+    def _check_sql_results(
+            self, snapshot, sql, params, param_types, expected, order=True):
+        if order and 'ORDER' not in sql:
             sql += ' ORDER BY eye_d'
         rows = list(snapshot.execute_sql(
             sql, params=params, param_types=param_types))
@@ -1041,6 +1042,14 @@ def test_execute_sql_w_query_param(self):
             expected=[(19,)],
         )
 
+        self._check_sql_results(
+            snapshot,
+            sql='SELECT eye_d FROM all_types WHERE exactly_hwhen = @hwhen',
+            params={'hwhen': self.SOME_TIME},
+            param_types={'hwhen': Type(code=TIMESTAMP)},
+            expected=[(19,)],
+        )
+
         self._check_sql_results(
             snapshot,
             sql=('SELECT eye_d FROM all_types WHERE approx_value >= @lower'
@@ -1051,24 +1060,6 @@ def test_execute_sql_w_query_param(self):
             expected=[(None,), (19,)],
         )
 
-        # Find -inf
-        self._check_sql_results(
-            snapshot,
-            sql='SELECT eye_d FROM all_types WHERE approx_value = @pos_inf',
-            params={'pos_inf': float('+inf')},
-            param_types={'pos_inf': Type(code=FLOAT64)},
-            expected=[(107,)],
-        )
-
-        # Find +inf
-        self._check_sql_results(
-            snapshot,
-            sql='SELECT eye_d FROM all_types WHERE approx_value = @neg_inf',
-            params={'neg_inf': float('-inf')},
-            param_types={'neg_inf': Type(code=FLOAT64)},
-            expected=[(207,)],
-        )
-
         self._check_sql_results(
             snapshot,
             sql='SELECT description FROM all_types WHERE eye_d = @my_id',
@@ -1093,8 +1084,6 @@ def test_execute_sql_w_query_param(self):
             expected=[(19,)],
         )
 
-        # NaNs cannot be searched for by equality.
-
         self._check_sql_results(
             snapshot,
             sql='SELECT eye_d FROM all_types WHERE exactly_hwhen = @hwhen',
@@ -1103,16 +1092,91 @@ def test_execute_sql_w_query_param(self):
             expected=[(19,)],
         )
 
-        array_type = Type(code=ARRAY, array_element_type=Type(code=INT64))
+        int_array_type = Type(code=ARRAY, array_element_type=Type(code=INT64))
+
         self._check_sql_results(
             snapshot,
             sql=('SELECT description FROM all_types '
                  'WHERE eye_d in UNNEST(@my_list)'),
             params={'my_list': [19, 99]},
-            param_types={'my_list': array_type},
+            param_types={'my_list': int_array_type},
             expected=[(u'dog',), (u'cat',)],
         )
 
+        str_array_type = Type(code=ARRAY, array_element_type=Type(code=STRING))
+
+        self._check_sql_results(
+            snapshot,
+            sql=('SELECT eye_d FROM all_types '
+                 'WHERE description in UNNEST(@my_list)'),
+            params={'my_list': []},
+            param_types={'my_list': str_array_type},
+            expected=[],
+        )
+
+        self._check_sql_results(
+            snapshot,
+            sql=('SELECT eye_d FROM all_types '
+                 'WHERE description in UNNEST(@my_list)'),
+            params={'my_list': [u'dog', u'cat']},
+            param_types={'my_list': str_array_type},
+            expected=[(19,), (99,)],
+        )
+
+        self._check_sql_results(
+            snapshot,
+            sql='SELECT @v',
+            params={'v': None},
+            param_types={'v': Type(code=STRING)},
+            expected=[(None,)],
+            order=False,
+        )
+
+    def test_execute_sql_w_query_param_transfinite(self):
+        session = self._db.session()
+        session.create()
+        self.to_delete.append(session)
+
+        with session.batch() as batch:
+            batch.delete(self.ALL_TYPES_TABLE, self.ALL)
+            batch.insert(
+                self.ALL_TYPES_TABLE,
+                self.ALL_TYPES_COLUMNS,
+                self.ALL_TYPES_ROWDATA)
+
+        snapshot = session.snapshot(
+            read_timestamp=batch.committed, multi_use=True)
+
+        # Find -inf
+        self._check_sql_results(
+            snapshot,
+            sql='SELECT eye_d FROM all_types WHERE approx_value = @pos_inf',
+            params={'pos_inf': float('+inf')},
+            param_types={'pos_inf': Type(code=FLOAT64)},
+            expected=[(107,)],
+        )
+
+        # Find +inf
+        self._check_sql_results(
+            snapshot,
+            sql='SELECT eye_d FROM all_types WHERE approx_value = @neg_inf',
+            params={'neg_inf': float('-inf')},
+            param_types={'neg_inf': Type(code=FLOAT64)},
+            expected=[(207,)],
+        )
+
+        rows = list(snapshot.execute_sql(
+            'SELECT'
+            ' [CAST("-inf" AS FLOAT64),'
+            ' CAST("+inf" AS FLOAT64),'
+            ' CAST("NaN" AS FLOAT64)]'))
+        self.assertEqual(len(rows), 1)
+        float_array, = rows[0]
+        self.assertEqual(float_array[0], float('-inf'))
+        self.assertEqual(float_array[1], float('+inf'))
+        # NaNs cannot be searched for by equality.
+        self.assertTrue(math.isnan(float_array[2]))
+
 
 class TestStreamingChunking(unittest.TestCase, _TestData):
 
