Change TestExoticSchemaNameRender to make it fail like in #95

Pogchamp-company · Jan 31, 2025 · c704bfd · c704bfd
1 parent de2ab82
commit c704bfd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/sync_enum_values/test_exotic_schema_name.py b/tests/sync_enum_values/test_exotic_schema_name.py
@@ -22,7 +22,7 @@ def get_database_schema(self) -> MetaData:
             Column("id", Integer, primary_key=True),
             Column(
                 "dealer",
-                postgresql.ENUM(*self.old_variants, name=self.enum_name),
+                postgresql.ENUM(*self.old_variants, name=self.enum_name, schema=KEYWORD_SCHEMA_NAME),
             ),
             schema=KEYWORD_SCHEMA_NAME,
         )
@@ -37,7 +37,7 @@ def get_target_schema(self) -> MetaData:
             Column("id", Integer, primary_key=True),
             Column(
                 "dealer",
-                postgresql.ENUM(*self.new_variants, name=self.enum_name),
+                postgresql.ENUM(*self.new_variants, name=self.enum_name, schema=KEYWORD_SCHEMA_NAME),
             ),
             schema=KEYWORD_SCHEMA_NAME,
         )
@@ -48,7 +48,7 @@ def get_expected_upgrade(self) -> str:
         return f"""
             # ### commands auto generated by Alembic - please adjust! ###
             op.sync_enum_values(
-                enum_schema='public',
+                enum_schema='default',
                 enum_name='dealers',
                 new_values=['Walter White', 'Jesse Pinkman'],
                 affected_columns=[TableReference(table_schema='default', table_name='pricing', column_name='dealer')],
@@ -61,7 +61,7 @@ def get_expected_downgrade(self) -> Optional[str]:
         return f"""
             # ### commands auto generated by Alembic - please adjust! ###
             op.sync_enum_values(
-                enum_schema='public',
+                enum_schema='default',
                 enum_name='dealers',
                 new_values=['Walter White', 'Jesse Pinkman', 'Tuco'],
                 affected_columns=[TableReference(table_schema='default', table_name='pricing', column_name='dealer')],