Skip to content

Commit a75ccae

Browse files
committed
Fix constraint name token test case
1 parent 553811d commit a75ccae

File tree

1 file changed

+14
-19
lines changed

1 file changed

+14
-19
lines changed

tests/test_generators.py

+14-19
Original file line numberDiff line numberDiff line change
@@ -2399,29 +2399,24 @@ def test_constraint_name_token(self, generator: CodeGenerator) -> None:
23992399
validate_code(
24002400
generator.generate(),
24012401
"""\
2402-
from sqlalchemy import CheckConstraint, Column, Integer, MetaData
2403-
from sqlalchemy.orm import declarative_base
2404-
from sqlalchemy.sql.elements import conv
2405-
2406-
metadata = MetaData(
2407-
naming_convention={
2408-
"ck": "ck_%(table_name)s_%(constraint_name)s",
2409-
"pk": "pk_%(table_name)s",
2410-
}
2411-
)
2402+
from sqlalchemy import CheckConstraint, Column, Integer, MetaData
2403+
from sqlalchemy.orm import declarative_base
2404+
from sqlalchemy.sql.elements import conv
24122405
2413-
Base = declarative_base(metadata=metadata)
2406+
Base = declarative_base()
2407+
Base.metadata.naming_convention = \
2408+
{'ck': 'ck_%(table_name)s_%(constraint_name)s', 'pk': 'pk_%(table_name)s'}
24142409
24152410
2416-
class Simple(Base):
2417-
__tablename__ = 'simple'
2418-
__table_args__ = (
2419-
CheckConstraint('id > 0', name='idcheck'),
2420-
CheckConstraint('number > 0', name=conv('non_default_name'))
2421-
)
2411+
class Simple(Base):
2412+
__tablename__ = 'simple'
2413+
__table_args__ = (
2414+
CheckConstraint('id > 0', name='idcheck'),
2415+
CheckConstraint('number > 0', name=conv('non_default_name'))
2416+
)
24222417
2423-
id = Column(Integer, primary_key=True)
2424-
number = Column(Integer)
2418+
id = Column(Integer, primary_key=True)
2419+
number = Column(Integer)
24252420
""",
24262421
)
24272422

0 commit comments

Comments
 (0)