uuids

.github/workflows/tests.yml

@@ -50,7 +50,7 @@ jobs:
       - name: Create src database
         shell: bash
         run: |
-          PGPASSWORD=password psql --host=localhost --username=postgres --file=tests/examples/src.dump
+          PGPASSWORD=password psql --host=localhost --username=postgres --set="ON_ERROR_STOP=1" --file=tests/examples/src.dump
       - name: Run Unit Tests
         shell: bash
         run: |

sqlsynthgen/make.py

@@ -15,6 +15,7 @@
 from mimesis.providers.base import BaseProvider
 from sqlacodegen.generators import DeclarativeGenerator
 from sqlalchemy import MetaData, UniqueConstraint, text
+from sqlalchemy.dialects import postgresql
 from sqlalchemy.sql import sqltypes
 
 from sqlsynthgen import providers
@@ -220,6 +221,7 @@ def _get_provider_for_column(column: Any) -> Tuple[List[str], str, List[str]]:
         (sqltypes.DateTime, False): "generic.datetime.datetime",
         (sqltypes.Numeric, False): "generic.numeric.float_number",
         (sqltypes.LargeBinary, False): "generic.bytes_provider.bytes",
+        (postgresql.UUID, False): "generic.cryptographic.uuid",
         (sqltypes.String, False): "generic.text.color",
         (sqltypes.String, True): "generic.person.password",
     }

tests/examples/src.dump

@@ -177,6 +177,16 @@ CREATE TABLE public.unique_constraint_test2 (
 
 ALTER TABLE public.unique_constraint_test2 OWNER TO postgres;
 
+--
+-- Name: data_type_test; Type: TABLE; Schema: public; Owner: postgres
+--
+
+CREATE TABLE public.data_type_test (
+    myuuid UUID NOT NULL
+);
+
+ALTER TABLE public.data_type_test OWNER TO postgres;
+
 --
 -- Data for Name: concept; Type: TABLE DATA; Schema: public; Owner: postgres
 --

tests/test_make.py

@@ -11,6 +11,7 @@
 from pydantic.tools import parse_obj_as
 from sqlalchemy import BigInteger, Column, String
 from sqlalchemy.dialects.mysql.types import INTEGER
+from sqlalchemy.dialects.postgresql import UUID
 
 from sqlsynthgen.make import (
     _get_provider_for_column,
@@ -174,6 +175,17 @@ def test_get_provider_for_column(self) -> None:
             ["100"],
         )
 
+        # UUID
+        (
+            _,
+            generator_function,
+            __,
+        ) = _get_provider_for_column(Column("myuuid", UUID))
+        self.assertEqual(
+            generator_function,
+            "generic.cryptographic.uuid",
+        )
+
 
 class TestMakeTables(SSGTestCase):
     """Test the make_tables function."""