feat: add support for PG.OID in parameterized queries (#1035)

* feat: add support for PG.OID in parameterized queries * test: add tests for PG.OID bindings * test: add test to check that the PG.OID param type is correct * lint: fix lint * test: correct new test name --------- Co-authored-by: larkee <larkee@users.noreply.github.com>

Author: larkee

google/cloud/spanner_v1/param_types.py

@@ -33,6 +33,7 @@
 JSON = Type(code=TypeCode.JSON)
 PG_NUMERIC = Type(code=TypeCode.NUMERIC, type_annotation=TypeAnnotationCode.PG_NUMERIC)
 PG_JSONB = Type(code=TypeCode.JSON, type_annotation=TypeAnnotationCode.PG_JSONB)
+PG_OID = Type(code=TypeCode.INT64, type_annotation=TypeAnnotationCode.PG_OID)
 
 
 def Array(element_type):

tests/system/test_session_api.py

@@ -2348,6 +2348,18 @@ def test_execute_sql_w_jsonb_bindings(
  )
 
 
+def test_execute_sql_w_oid_bindings(
+ not_emulator, not_google_standard_sql, sessions_database, database_dialect
+):
+ _bind_test_helper(
+ sessions_database,
+ database_dialect,
+ spanner_v1.param_types.PG_OID,
+ 123,
+ [123, 456],
+ )
+
+
 def test_execute_sql_w_query_param_struct(sessions_database, not_postgres):
  name = "Phred"
  count = 123

tests/unit/test_param_types.py

@@ -70,3 +70,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)