Fix a bug in configuration of unit tests.

Change-Id: Ie2e73142f99bd7fcf015590673b25cde7c685f80

diff --git a/moonv4/moon_db/tests/unit_python/conftest.py b/moonv4/moon_db/tests/unit_python/conftest.py
index 6cce944..b9f38d7 100644 (file)
--- a/moonv4/moon_db/tests/unit_python/conftest.py
+++ b/moonv4/moon_db/tests/unit_python/conftest.py
@@ -192,9 +192,9 @@ def set_consul_and_db(monkeypatch):
                 "id": "1111111111111"
             }]}
         )
-        from moon_db.db_manager import init_engine, run
+        from moon_db.db_manager import init_engine, main
         engine = init_engine()
-        run("upgrade", logging.getLogger("db_manager"), engine)
+        main("upgrade", logging.getLogger("db_manager"), engine)
         yield m
         os.unlink(CONF['database']['url'].replace("sqlite:///", ""))
 

diff --git a/moonv4/moon_manager/tests/unit_python/conftest.py b/moonv4/moon_manager/tests/unit_python/conftest.py
index 58f04af..d6f518f 100644 (file)
--- a/moonv4/moon_manager/tests/unit_python/conftest.py
+++ b/moonv4/moon_manager/tests/unit_python/conftest.py
@@ -176,10 +176,10 @@ def no_requests(monkeypatch):
             }]}
         )
         print("Start populating the DB.")
-        from moon_db.db_manager import init_engine, run
+        from moon_db.db_manager import init_engine, main
         engine = init_engine()
         print("engine={}".format(engine))
-        run("upgrade", logging.getLogger("db_manager"), engine)
+        main("upgrade", logging.getLogger("db_manager"), engine)
         print("End populating the DB.")
         yield m