Fix bug 1505

[osm/N2VC.git] / n2vc / tests / unit / test_n2vc_juju_conn.py

diff --git a/n2vc/tests/unit/test_n2vc_juju_conn.py b/n2vc/tests/unit/test_n2vc_juju_conn.py
index 1f723fe..e5e26be 100644 (file)
--- a/n2vc/tests/unit/test_n2vc_juju_conn.py
+++ b/n2vc/tests/unit/test_n2vc_juju_conn.py
@@ -15,6 +15,9 @@
 
 import asyncio
 import logging
 
 import asyncio
 import logging

+from unittest.mock import Mock
+
+

 import asynctest
 from n2vc.n2vc_juju_conn import N2VCJujuConnector
 from osm_common import fslocal
 import asynctest
 from n2vc.n2vc_juju_conn import N2VCJujuConnector
 from osm_common import fslocal


@@ -51,6 +54,7 @@ class N2VCJujuConnTestCase(asynctest.TestCase):
 
         logging.disable(logging.CRITICAL)
 
 
         logging.disable(logging.CRITICAL)
 

+        N2VCJujuConnector.get_public_key = Mock()

         self.n2vc = N2VCJujuConnector(
             db=db,
             fs=fslocal.FsLocal(),
         self.n2vc = N2VCJujuConnector(
             db=db,
             fs=fslocal.FsLocal(),


@@ -61,6 +65,7 @@ class N2VCJujuConnTestCase(asynctest.TestCase):
             vca_config=vca_config,
             on_update_db=None,
         )
             vca_config=vca_config,
             on_update_db=None,
         )

+        N2VCJujuConnector.get_public_key.assert_not_called()

 
 
 @asynctest.mock.patch("n2vc.libjuju.Libjuju.get_metrics")
 
 
 @asynctest.mock.patch("n2vc.libjuju.Libjuju.get_metrics")