Pydbvolve issues 10

.gitignore

@@ -97,3 +97,5 @@ test_db.sqlite
 # cli testing
 pydbvolve-cli
 
+.pytest_cache
+

pydbvolve/__init__.py

@@ -105,7 +105,7 @@ def pre_statement(config, migration, statement):
 # End pre_statement
 
 
-def post_statement(config, migration, statement):
+def post_statement(config, migration, statement, cursor):
     """
     Called on SQL migrations only. Called after individual statement execution.
     Accepts config dict, migration dict, and statement string as arguments.
@@ -951,8 +951,7 @@ def run_sql_migration(config, migration):
 
             with conn.cursor() as cur:
                 cur.execute(stmt)
-
-            post_statement(config, migration, stmt)
+                post_statement(config, migration, stmt, cur)
 
     return True
 # End run_sql_migration

setup.py

@@ -2,11 +2,11 @@
 
 setup(name='pydbvolve',
       description='Database migrations',
-      long_description='Database migrations with Python3 using sql or python3 migrations. Evolve your database with python!',
+      long_description='Python Database Evolution. Database migrations with Python3 using sql or python3 migrations. Evolve your database with python!',
       url="https://github.com/Teamworksapp/pydbvolve",
       author="Teamworksapp",
       author_email="[email protected]",
-      version='1.0.1',
+      version='1.0.2',
       license='MIT',
       classifiers=[
         # How mature is this project? Common values are

tests/pydbvolve.conf

@@ -4,6 +4,8 @@ import os
 import re
 
 
+TEST_OUT={}
+
 # Could have used sqlite3.Row, but simpler is better, in this case. The main script works with dict types.
 def dict_factory(cur, row):
     return {col[0]: row[ix] for ix, col in enumerate(cur.description)}
@@ -62,4 +64,9 @@ def get_db_connection(config, credentials):
     return conn
 # End get_db_connection
 
+def post_statement(config, migration, statement, cursor):
+    TEST_OUT['rowcount'] = cursor.rowcount
+
+    return None
+
 

tests/unittests/test_03_migration_jobs.py

@@ -11,6 +11,7 @@
 
 TEST_CONFIG_FILE = os.path.join('tests', 'pydbvolve.conf')
 TEST_DB_FILE = os.path.join('tests', 'test_db.sqlite')
+TEST_OUT_FILE = os.path.join('tests', 'test_post_statement.txt')
 
 
 def _table_exists(conn, table_name):
@@ -753,5 +754,18 @@ def test_22_downgrade_baseline(capsys):
 # End test_21_upgrade_baseline_current
 
 
+def test_22_post_statement():
+    try:
+        os.unlink(TEST_DB_FILE)
+    except:
+        pass
+
+    rc = pydbvolve.run_migration(TEST_CONFIG_FILE, 'upgrade', 'r1.0.0', True, False)
+    assert (rc == 0)
+    assert ('rowcount' in pydbvolve.TEST_OUT)
+    assert (pydbvolve.TEST_OUT['rowcount'] == 1)
+
+    os.unlink(TEST_DB_FILE)
+# End test_22_post_statement