Fix sqlalchemy flaky tests

trinodb · Sep 13, 2022 · d358954 · d358954
1 parent 9d898a8
commit d358954
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/integration/test_sqlalchemy_integration.py b/tests/integration/test_sqlalchemy_integration.py
@@ -102,7 +102,9 @@ def test_insert(trino_connection):
  schema="test")
  metadata.create_all(engine)
  ins = users.insert()
- conn.execute(ins, {"id": 2, "name": "wendy", "fullname": "Wendy Williams"})
+ result = conn.execute(ins, {"id": 2, "name": "wendy", "fullname": "Wendy Williams"})
+ # TODO: Can we get rid of this?
+ result.fetchall()
  query = sqla.select(users)
  result = conn.execute(query)
  rows = result.fetchall()
@@ -126,11 +128,13 @@ def test_insert_multiple_statements(trino_connection):
  schema="test")
  metadata.create_all(engine)
  ins = users.insert()
- conn.execute(ins, [
+ result = conn.execute(ins, [
  {"id": 2, "name": "wendy", "fullname": "Wendy Williams"},
  {"id": 3, "name": "john", "fullname": "John Doe"},
  {"id": 4, "name": "mary", "fullname": "Mary Hopkins"},
  ])
+ # TODO: Can we get rid of this?
+ result.fetchall()
  query = sqla.select(users)
  result = conn.execute(query)
  rows = result.fetchall()