mirror of
https://github.com/apache/superset.git
synced 2026-04-19 16:14:52 +00:00
refactor: Ensure Flask framework leverages the Flask-SQLAlchemy session (Phase II) (#26909)
This commit is contained in:
@@ -24,9 +24,10 @@ def test_column_model(session: Session) -> None:
|
||||
"""
|
||||
Test basic attributes of a ``Column``.
|
||||
"""
|
||||
from superset import db
|
||||
from superset.columns.models import Column
|
||||
|
||||
engine = session.get_bind()
|
||||
engine = db.session.get_bind()
|
||||
Column.metadata.create_all(engine) # pylint: disable=no-member
|
||||
|
||||
column = Column(
|
||||
@@ -35,8 +36,8 @@ def test_column_model(session: Session) -> None:
|
||||
expression="ds",
|
||||
)
|
||||
|
||||
session.add(column)
|
||||
session.flush()
|
||||
db.session.add(column)
|
||||
db.session.flush()
|
||||
|
||||
assert column.id == 1
|
||||
assert column.uuid is not None
|
||||
|
||||
Reference in New Issue
Block a user