Fix sqlalchemy flaky tests

Author: mdesmet

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()