fix test for sqlite

apache · Jun 13, 2022 · 4efb13a · 4efb13a
1 parent baa5cd8
commit 4efb13a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/integration_tests/fixtures/datasource.py b/tests/integration_tests/fixtures/datasource.py
@@ -182,8 +182,8 @@ def load_dataset_with_columns() -> Generator[SqlaTable, None, None]:
             "students",
             meta,
             Column("id", Integer, primary_key=True),
-            Column("name", String),
-            Column("lastname", String),
+            Column("name", String(255)),
+            Column("lastname", String(255)),
             Column("ds", Date),
         )
         meta.create_all(engine)
@@ -203,6 +203,8 @@ def load_dataset_with_columns() -> Generator[SqlaTable, None, None]:
         students_table = meta.tables.get("students")
         if students_table is not None:
             base = declarative_base()
+            # needed for sqlite
+            session.commit()
             base.metadata.drop_all(engine, [students_table], checkfirst=True)
         session.delete(dataset)
         session.delete(column)