Minor improvements to libjuju.py

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

diff --git a/n2vc/tests/unit/test_juju_watcher.py b/n2vc/tests/unit/test_juju_watcher.py
index 41065bf..d333b33 100644 (file)
--- a/n2vc/tests/unit/test_juju_watcher.py
+++ b/n2vc/tests/unit/test_juju_watcher.py
@@ -142,6 +142,7 @@ class StatusTest(TestCase):
         self.assertTrue(isinstance(value, str))
 
 
         self.assertTrue(isinstance(value, str))
 
 

+@asynctest.mock.patch("asyncio.sleep")

 class WaitForModelTest(asynctest.TestCase):
     @asynctest.mock.patch("juju.client.connector.Connector.connect")
     def setUp(self, mock_connect=None):
 class WaitForModelTest(asynctest.TestCase):
     @asynctest.mock.patch("juju.client.connector.Connector.connect")
     def setUp(self, mock_connect=None):


@@ -149,7 +150,7 @@ class WaitForModelTest(asynctest.TestCase):
         self.model = Model()
 
     @asynctest.mock.patch("juju.model.Model.block_until")
         self.model = Model()
 
     @asynctest.mock.patch("juju.model.Model.block_until")

-    def test_wait_for_model(self, mock_block_until):
+    def test_wait_for_model(self, mock_block_until, mock_sleep):

         self.loop.run_until_complete(
             JujuModelWatcher.wait_for_model(self.model, timeout=None)
         )
         self.loop.run_until_complete(
             JujuModelWatcher.wait_for_model(self.model, timeout=None)
         )


@@ -157,7 +158,7 @@ class WaitForModelTest(asynctest.TestCase):
 
     @asynctest.mock.patch("asyncio.ensure_future")
     @asynctest.mock.patch("asyncio.wait")
 
     @asynctest.mock.patch("asyncio.ensure_future")
     @asynctest.mock.patch("asyncio.wait")

-    def test_wait_for_model_exception(self, mock_wait, mock_ensure_future):
+    def test_wait_for_model_exception(self, mock_wait, mock_ensure_future, mock_sleep):

         task = Mock()
         mock_ensure_future.return_value = task
         mock_wait.side_effect = Exception
         task = Mock()
         mock_ensure_future.return_value = task
         mock_wait.side_effect = Exception