test(datatype): handle geometry

Author: Brooke-white

test/integration/datatype/_generate_test_datatype_tables.py

@@ -17,9 +17,78 @@
 """
 
 
+class RedshiftDatatypes(Enum):
+    """
+    All redshift specific datatypes are defined here
+    """
+
+    super = auto()
+    geometry = auto()
+
+    @classmethod
+    def list(cls) -> typing.List["RedshiftDatatypes"]:
+        return list(map(lambda p: p, cls))  # type: ignore
+
+
+redshift_test_data: typing.Dict[str, typing.Tuple[typing.Tuple[str, str], ...]] = {
+    RedshiftDatatypes.geometry.name: (
+        (
+            "ST_GeomFromText('LINESTRING(1 2,3 4,5 6,7 8,9 10,11 12,13 14,15 16,17 18,19 20)')",
+            (
+                "01020000000A000000000000000000F03F000000000000004000000000000008400000000000001040000000000000144000000000"
+                "000018400000000000001C400000000000002040000000000000224000000000000024400000000000002640000000000000284000"
+                "00000000002A400000000000002C400000000000002E40000000000000304000000000000031400000000000003240000000000000"
+                "33400000000000003440"
+            ),
+        ),
+        (
+            "ST_GeomFromText('POLYGON((0 0,0 1,1 1,1 0,0 0))',4326)",
+            (
+                "0103000020E61000000100000005000000000000000000000000000000000000000000000000000000000000000000F03F00000000"
+                "0000F03F000000000000F03F000000000000F03F000000000000000000000000000000000000000000000000"
+            ),
+        ),
+        (
+            (
+                "ST_GeomFromEWKB('0103000020E61000000100000005000000000000000000000000000000000000000000000000000000000000000000F03F000000"
+                "000000F03F000000000000F03F000000000000F03F000000000000000000000000000000000000000000000000')"
+            ),
+            (
+                "0103000020E61000000100000005000000000000000000000000000000000000000000000000000000000000000000F03F000000"
+                "000000F03F000000000000F03F000000000000F03F000000000000000000000000000000000000000000000000"
+            ),
+        ),
+        (
+            (
+                "ST_GeomFromEWKB('01020000000A000000000000000000F03F0000000000000040000000000000084000000000000010400000000000001440000000"
+                "00000018400000000000001C4000000000000020400000000000002240000000000000244000000000000026400000000000002840"
+                "0000000000002A400000000000002C400000000000002E400000000000003040000000000000314000000000000032400000000000"
+                "0033400000000000003440')"
+            ),
+            (
+                "01020000000A000000000000000000F03F0000000000000040000000000000084000000000000010400000000000001440000000"
+                "00000018400000000000001C4000000000000020400000000000002240000000000000244000000000000026400000000000002840"
+                "0000000000002A400000000000002C400000000000002E400000000000003040000000000000314000000000000032400000000000"
+                "0033400000000000003440"
+            ),
+        ),
+        (
+            (
+                "ST_GeomFromWKB('01030000000100000005000000000000000000000000000000000000000000000000000000000000000000F03"
+                "F000000000000F03F000000000000F03F000000000000F03F000000000000000000000000000000000000000000000000')"
+            ),
+            (
+                "01030000000100000005000000000000000000000000000000000000000000000000000000000000000000F03"
+                "F000000000000F03F000000000000F03F000000000000F03F000000000000000000000000000000000000000000000000"
+            ),
+        ),
+    )
+}
+
+
 class Datatypes(Enum):
     """
-    All supported datatypes are defined here, so a test table can be created for each
+    All non-redshift specific datatypes are defined here, so a test table can be created for each
     """
 
     int2 = auto()

test/integration/datatype/test_datatypes.py

@@ -9,6 +9,7 @@
     datatype_test_setup,
     datatype_test_teardown,
     get_table_name,
+    redshift_test_data,
     test_data,
 )
 
@@ -64,3 +65,23 @@ def test_datatype_recv_support(db_kwargs, datatype):
                     assert isclose(results[ridx][1].microsecond, exp_row[-1].microsecond, rel_tol=1e1)
                 else:
                     assert results[ridx][1] == exp_row[-1]
+
+
+redshift_datatype_testcases: typing.List[typing.Tuple] = []
+for datatype in redshift_test_data:
+    for test_case in redshift_test_data[datatype]:
+        redshift_datatype_testcases.append((datatype, test_case))
+
+
+@pytest.mark.parametrize("_input", redshift_datatype_testcases)
+def test_redshift_specific_recv_support(db_kwargs, _input):
+    datatype, data = _input
+    test_val, exp_val = data
+
+    with redshift_connector.connect(**db_kwargs) as con:
+        with con.cursor() as cursor:
+            cursor.execute("select {}".format(test_val))
+            results: typing.Tuple = cursor.fetchall()
+            assert len(results) == 1
+            assert len(results[0]) == 1
+            assert results[0][0] == exp_val

test/unit/datatype/test_data_in.py

@@ -23,6 +23,7 @@ class Datatypes(Enum):
     time = type_utils.time_in
     timetz = type_utils.timetz_in
     date = type_utils.date_in
+    geometry = type_utils.text_recv
 
 
 test_data: typing.Dict[Datatypes, typing.List[typing.Tuple]] = {
@@ -129,6 +130,20 @@ class Datatypes(Enum):
             date(year=2020, day=10, month=3),
         )
     ],
+    Datatypes.geometry: [
+        (
+            b"\x00\x01\x00\x00\x01R01020000000A000000000000000000F03F000000000000004000000000000008400000000000001040000000000000144000000000000018400000000000001C40000000000000204000000000000022400000000000002440000000000000264000000000000028400000000000002A400000000000002C400000000000002E4000000000000030400000000000003140000000000000324000000000000033400000000000003440",
+            6,
+            338,
+            "01020000000A000000000000000000F03F000000000000004000000000000008400000000000001040000000000000144000000000000018400000000000001C40000000000000204000000000000022400000000000002440000000000000264000000000000028400000000000002A400000000000002C400000000000002E4000000000000030400000000000003140000000000000324000000000000033400000000000003440",
+        ),
+        (
+            b"\x00\x01\x00\x00\x00\xc20103000020E61000000100000005000000000000000000000000000000000000000000000000000000000000000000F03F000000000000F03F000000000000F03F000000000000F03F000000000000000000000000000000000000000000000000",
+            6,
+            194,
+            "0103000020E61000000100000005000000000000000000000000000000000000000000000000000000000000000000F03F000000000000F03F000000000000F03F000000000000F03F000000000000000000000000000000000000000000000000",
+        ),
+    ],
 }