Test bulk insert using SQLAlchemy core .values() call

Signed-off-by: Jesse Whitehouse <[email protected]>

Author: Jesse Whitehouse

tests/e2e/sqlalchemy/test_queries.py

@@ -79,6 +79,34 @@ def test_create_table_not_null(db_engine, metadata_obj):
 
     metadata_obj.drop_all()
 
+def test_bulk_insert_with_core(db_engine, metadata_obj, session):
+
+    import random
+
+    num_to_insert = random.choice(range(10_000, 20_000))
+
+    table_name = "PySQLTest_{}".format(datetime.datetime.utcnow().strftime("%s"))
+
+    names = ["Bim", "Miki", "Sarah", "Ira"]
+    
+    SampleTable = Table(
+         table_name,
+         metadata_obj,
+         Column("name", String(255)),
+         Column("number", Integer)
+    )
+
+    rows = [{"name": names[i%3], "number": random.choice(range(10000))} for i in range(num_to_insert)]
+
+    metadata_obj.create_all()
+    db_engine.execute(insert(SampleTable).values(rows))
+
+    rows = db_engine.execute(select(SampleTable)).fetchall()
+
+    assert len(rows) == num_to_insert
+
+
+
 
 
 def test_create_insert_drop_table_core(base, db_engine, metadata_obj: MetaData):