Skip to content

Commit 53de15d

Browse files
committed
Fix constraint name token test case
1 parent 0930157 commit 53de15d

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
@@ -2400,29 +2400,24 @@ def test_constraint_name_token(self, generator: CodeGenerator) -> None:
24002400
validate_code(
24012401
generator.generate(),
24022402
"""\
2403-
from sqlalchemy import CheckConstraint, Column, Integer, MetaData
2404-
from sqlalchemy.orm import declarative_base
2405-
from sqlalchemy.sql.elements import conv
2406-
2407-
metadata = MetaData(
2408-
naming_convention={
2409-
"ck": "ck_%(table_name)s_%(constraint_name)s",
2410-
"pk": "pk_%(table_name)s",
2411-
}
2412-
)
2403+
from sqlalchemy import CheckConstraint, Column, Integer, MetaData
2404+
from sqlalchemy.orm import declarative_base
2405+
from sqlalchemy.sql.elements import conv
24132406
2414-
Base = declarative_base(metadata=metadata)
2407+
Base = declarative_base()
2408+
Base.metadata.naming_convention = \
2409+
{'ck': 'ck_%(table_name)s_%(constraint_name)s', 'pk': 'pk_%(table_name)s'}
24152410
24162411
2417-
class Simple(Base):
2418-
__tablename__ = 'simple'
2419-
__table_args__ = (
2420-
CheckConstraint('id > 0', name='idcheck'),
2421-
CheckConstraint('number > 0', name=conv('non_default_name'))
2422-
)
2412+
class Simple(Base):
2413+
__tablename__ = 'simple'
2414+
__table_args__ = (
2415+
CheckConstraint('id > 0', name='idcheck'),
2416+
CheckConstraint('number > 0', name=conv('non_default_name'))
2417+
)
24232418
2424-
id = Column(Integer, primary_key=True)
2425-
number = Column(Integer)
2419+
id = Column(Integer, primary_key=True)
2420+
number = Column(Integer)
24262421
""",
24272422
)
24282423

0 commit comments

Comments
 (0)