remove autoflush for queries during dual write (#20460)

This commit is contained in:
Elizabeth Thompson
2022-06-23 14:50:30 -07:00
committed by GitHub
parent 661ab35bd0
commit 44f0b511dd
3 changed files with 68 additions and 12 deletions

View File

@@ -718,6 +718,7 @@ def test_update_physical_sqlatable_columns(
metrics=[],
database=Database(database_name="my_database", sqlalchemy_uri="sqlite://"),
)
session.add(sqla_table)
session.flush()
@@ -735,8 +736,11 @@ def test_update_physical_sqlatable_columns(
assert session.query(Column).count() == 3
dataset = session.query(Dataset).one()
assert len(dataset.columns) == 2
for table_column, dataset_column in zip(sqla_table.columns, dataset.columns):
assert table_column.uuid == dataset_column.uuid
# check that both lists have the same uuids
assert [col.uuid for col in sqla_table.columns].sort() == [
col.uuid for col in dataset.columns
].sort()
# delete the column in the original instance
sqla_table.columns = sqla_table.columns[1:]