test: add test to check that the PG.OID param type is correct

googleapis · Nov 13, 2023 · 4f09a0b · 4f09a0b
1 parent c8a5b65
commit 4f09a0b
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/unit/test_param_types.py b/tests/unit/test_param_types.py
@@ -71,3 +71,20 @@ def test_it(self):
         found = param_types.PG_JSONB
 
         self.assertEqual(found, expected)
+
+
+class Test_OidParamType(unittest.TestCase):
+    def test_it(self):
+        from google.cloud.spanner_v1 import Type
+        from google.cloud.spanner_v1 import TypeCode
+        from google.cloud.spanner_v1 import TypeAnnotationCode
+        from google.cloud.spanner_v1 import param_types
+
+        expected = Type(
+            code=TypeCode.INT64,
+            type_annotation=TypeAnnotationCode.PG_OID,
+        )
+
+        found = param_types.PG_OID
+
+        self.assertEqual(found, expected)