From 49aa84cd7485b8b1bbdc77ccca861f68f1f9e90c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Knut=20Olav=20L=C3=B8ite?= Date: Thu, 19 Jun 2025 06:58:35 +0200 Subject: [PATCH] test: native enums --- google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py | 2 +- test/mockserver_tests/test_basics.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py b/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py index 8e935fb2..a93a3435 100644 --- a/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py +++ b/google/cloud/sqlalchemy_spanner/sqlalchemy_spanner.py @@ -798,7 +798,7 @@ class SpannerDialect(DefaultDialect): supports_sequences = True sequences_optional = False supports_identity_columns = True - supports_native_enum = True + supports_native_enum = False supports_native_boolean = True supports_native_decimal = True supports_statement_cache = True diff --git a/test/mockserver_tests/test_basics.py b/test/mockserver_tests/test_basics.py index 3e422885..e47b3b8d 100644 --- a/test/mockserver_tests/test_basics.py +++ b/test/mockserver_tests/test_basics.py @@ -26,7 +26,7 @@ String, func, text, - BigInteger, + BigInteger, Enum, ) from sqlalchemy.orm import Session, DeclarativeBase, Mapped, mapped_column from sqlalchemy.testing import eq_, is_instance_of @@ -120,6 +120,7 @@ def test_create_table(self): metadata, Column("user_id", Integer, primary_key=True), Column("user_name", String(16), nullable=False), + Column("status", Enum("pending_activation", "active", "disabled")) ) metadata.create_all(engine) requests = self.database_admin_service.requests @@ -130,7 +131,8 @@ def test_create_table(self): "CREATE TABLE users (\n" "\tuser_id INT64 NOT NULL " "GENERATED BY DEFAULT AS IDENTITY (BIT_REVERSED_POSITIVE), \n" - "\tuser_name STRING(16) NOT NULL\n" + "\tuser_name STRING(16) NOT NULL, \n" + "\tstatus STRING(18)\n" ") PRIMARY KEY (user_id)", requests[0].statements[0], )