chore: add unit and system tests for UUID

Signed-off-by: Sri Harsha CH <sriharshach@google.com>

Author: harshachinta

google/cloud/spanner_v1/streamed.py

@@ -394,6 +394,7 @@ def _merge_struct(lhs, rhs, type_):
     TypeCode.PROTO: _merge_string,
     TypeCode.INTERVAL: _merge_string,
     TypeCode.ENUM: _merge_string,
+    TypeCode.UUID: _merge_string,
 }
 
 

tests/system/test_session_api.py

@@ -20,6 +20,8 @@
 import struct
 import threading
 import time
+import uuid
+
 import pytest
 
 import grpc
@@ -3056,6 +3058,16 @@ def test_execute_sql_returning_transfinite_floats(sessions_database, not_postgre
         assert math.isnan(float_array[2])
 
 
+def test_execute_sql_w_uuid_bindings(sessions_database, database_dialect):
+    _bind_test_helper(
+        sessions_database,
+        database_dialect,
+        spanner_v1.param_types.UUID,
+        uuid.uuid4(),
+        [uuid.uuid4(), uuid.uuid4()],
+    )
+
+
 def test_partition_query(sessions_database, not_emulator, not_experimental_host):
     row_count = 40
     sql = f"SELECT * FROM {_sample_data.TABLE}"

tests/unit/test__helpers.py

@@ -14,6 +14,8 @@
 
 
 import unittest
+import uuid
+
 import mock
 
 from opentelemetry.sdk.resources import Resource
@@ -786,6 +788,18 @@ def test_w_proto_enum(self):
             self._callFUT(value_pb, field_type, field_name, column_info), VALUE
         )
 
+    def test_w_uuid(self):
+        from google.protobuf.struct_pb2 import Value
+        from google.cloud.spanner_v1 import Type
+        from google.cloud.spanner_v1 import TypeCode
+
+        VALUE = uuid.uuid4()
+        field_type = Type(code=TypeCode.UUID)
+        field_name = "uuid_column"
+        value_pb = Value(string_value=str(VALUE))
+
+        self.assertEqual(self._callFUT(value_pb, field_type, field_name), VALUE)
+
 
 class Test_parse_list_value_pbs(unittest.TestCase):
     def _callFUT(self, *args, **kw):