Fix test bug.

brighthive · Mar 2, 2020 · 3d5da74 · 3d5da74
1 parent 7b451d7
commit 3d5da74
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -129,7 +129,7 @@ def upgrade_loop(self):
             try:
                 with self.app.app_context():
                     print("------------- running upgrade")
-                    self.data_model_manager.run_upgrade()
+                    self.data_model_manager.initalize_base_models()
 
                     if self.schema_dicts is None:
                         print("------------- running monitor data resources")
@@ -143,7 +143,7 @@ def upgrade_loop(self):
                             print("------------- running monitor data models")
                             self.data_model_manager.work_on_schema(schema_dict, "custom_schema")
 
-                    self.data_model_manager.run_upgrade()
+                    self.data_model_manager.initalize_base_models()
                     upgraded = True
                     return True