X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=utils%2Ftest%2Ftestapi%2Fopnfv_testapi%2Ftests%2Funit%2Fresources%2Ftest_base.py;h=dcec4e958cac941867439e999682b5dbafab1c13;hb=822cd5c32c9a129aa8aad9952f25a37123c96733;hp=67831f508e53bf7a2003205721380ea49ea8ef82;hpb=cbcabf6643e68c019b407c95b1526e77337f7697;p=releng.git

diff --git a/utils/test/testapi/opnfv_testapi/tests/unit/resources/test_base.py b/utils/test/testapi/opnfv_testapi/tests/unit/resources/test_base.py
index 67831f508..dcec4e958 100644
--- a/utils/test/testapi/opnfv_testapi/tests/unit/resources/test_base.py
+++ b/utils/test/testapi/opnfv_testapi/tests/unit/resources/test_base.py
@@ -41,18 +41,14 @@ class TestBase(testing.AsyncHTTPTestCase):
         self.config_patcher = mock.patch(
             'argparse.ArgumentParser.parse_known_args',
             return_value=(argparse.Namespace(config_file=config), None))
-        self.db_patcher = mock.patch('opnfv_testapi.cmd.server.get_db',
-                                     self._fake_pymongo)
+        self.db_patcher = mock.patch('opnfv_testapi.db.api.DB',
+                                     fake_pymongo)
         self.config_patcher.start()
         self.db_patcher.start()
 
     def set_config_file(self):
         self.config_file = 'normal.ini'
 
-    @staticmethod
-    def _fake_pymongo():
-        return fake_pymongo
-
     def get_app(self):
         from opnfv_testapi.cmd import server
         return server.make_app()