Run fixture sql injection via session

dropbox · May 13, 2015 · c890662 · c890662
1 parent 5c74ad2
commit c890662
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/model_tests/fixtures.py b/tests/model_tests/fixtures.py
@@ -24,9 +24,9 @@ def fin():
     return session
 
 @pytest.fixture
-def sample_data1(db_engine):
+def sample_data1(session):
     sql_file = open('tests/sample_data/sample_data1.sql')
     sql = sql_file.read()
     for statement in sql.split(";"):
         scrubbed = re.sub(r'[\t\n]', ' ', statement)
-        db_engine.execute(scrubbed)
+        session.execute(scrubbed)