add test for existing table

dimagi · May 8, 2018 · 6cd13c9 · 6cd13c9
1 parent 1f10dda
commit 6cd13c9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -13,9 +13,10 @@
 logging.getLogger().setLevel(logging.DEBUG)
 logging.getLogger().addHandler(logging.StreamHandler())
 
+
 @pytest.fixture(scope="class", params=[
     {
-        'url': "postgresql://postgres@/%s",
+        'url': "postgresql://postgres@localhost/%s",
         'admin_db': 'postgres'
     },
     {

diff --git a/tests/test_checkpointmanager.py b/tests/test_checkpointmanager.py
@@ -26,6 +26,12 @@ def test_create_checkpoint_table(self, manager):
         with manager:
             assert 'commcare_export_runs' in manager.metadata.tables
 
+    def test_checkpoint_table_exists(self, manager):
+        self.test_create_checkpoint_table(manager)
+        with manager:
+            manager.connection.execute(manager.sqlalchemy.sql.text('DROP TABLE alembic_version'))
+        manager.create_checkpoint_table()
+
     def test_get_time_of_last_run(self, manager):
         manager.create_checkpoint_table()
         with manager: