add unit tests

tested using
tox -e unit -- tests/unit/test_model.py::TestModelWaitForIdle

Author: sed-i

tests/integration/test_application.py

@@ -151,10 +151,10 @@ async def test_upgrade_local_charm(event_loop):
         tests_dir = Path(__file__).absolute().parent
         charm_path = tests_dir / 'upgrade-charm'
         app = await model.deploy('ubuntu', series='focal')
-        await model.wait_for_idle(wait_for_active=True)
+        await model.wait_for_idle(wait_for_status="active")
         assert app.data['charm-url'].startswith('cs:ubuntu')
         await app.upgrade_charm(path=charm_path)
-        await model.wait_for_idle()  # nb: can't use wait_for_active because test charm goes to "waiting"
+        await model.wait_for_idle(wait_for_status="waiting")
         assert app.data['charm-url'] == 'local:focal/ubuntu-0'
 
 

tests/integration/test_model.py

@@ -97,7 +97,7 @@ async def test_deploy_local_charm(event_loop):
     async with base.CleanModel() as model:
         await model.deploy(str(charm_path))
         assert 'charm' in model.applications
-        await model.wait_for_idle(wait_for_active=True)
+        await model.wait_for_idle(wait_for_status="active")
         assert model.units['charm/0'].workload_status == 'active'
 
 
@@ -113,7 +113,7 @@ async def test_wait_local_charm_blocked(event_loop):
         assert 'charm' in model.applications
         await model.wait_for_idle()
         with pytest.raises(JujuUnitError):
-            await model.wait_for_idle(wait_for_active=True,
+            await model.wait_for_idle(wait_for_status="active",
                                       raise_on_blocked=True,
                                       timeout=30)
 
@@ -130,7 +130,7 @@ async def test_wait_local_charm_waiting_timeout(event_loop):
         assert 'charm' in model.applications
         await model.wait_for_idle()
         with pytest.raises(asyncio.TimeoutError):
-            await model.wait_for_idle(wait_for_active=True, timeout=30)
+            await model.wait_for_idle(wait_for_status="active", timeout=30)
 
 
 @base.bootstrapped

tests/unit/test_model.py

@@ -262,3 +262,50 @@ async def test_with_posargs(self, mock_connect, mock_connect_model, _):
                                         macaroons='macaroons',
                                         loop='loop',
                                         max_frame_size='max_frame_size')
+
+
+class TestModelWaitForIdle(asynctest.TestCase):
+    async def test_no_args(self):
+        m = Model()
+        with self.assertWarns(DeprecationWarning):
+            # no apps so should return right away
+            await m.wait_for_idle(wait_for_active=True)
+
+    async def test_timeout(self):
+        m = Model()
+        import asyncio
+        with self.assertRaises(asyncio.TimeoutError) as cm:
+            # no apps so should timeout after timeout period
+            await m.wait_for_idle(apps=["nonexisting_app"], timeout=1, idle_period=0)
+        self.assertEqual(str(cm.exception), "Timed out waiting for model:\nnonexisting_app (missing)")
+
+    async def test_wait_for_active_status(self):
+        # create a custom apps mock
+        from types import SimpleNamespace
+        apps = {"dummy_app": SimpleNamespace(
+            status="active",
+            units=[SimpleNamespace(
+                name="mockunit/0",
+                workload_status="active",
+                workload_status_message="workload_status_message",
+                machine=None,
+                agent_status="idle",
+            )],
+        )}
+
+        from unittest.mock import patch, PropertyMock
+        with patch.object(Model, 'applications', new_callable=PropertyMock) as mock_apps:
+            mock_apps.return_value = apps
+            m = Model()
+
+            # pass "active" via `wait_for_status` (str)
+            await m.wait_for_idle(apps=["dummy_app"], wait_for_status="active", timeout=5, idle_period=0)
+
+            # pass "active" via `wait_for_active` (bool; deprecated)
+            await m.wait_for_idle(apps=["dummy_app"], wait_for_active=True, timeout=5, idle_period=0)
+
+            # use both `wait_for_status` and `wait_for_active` - `wait_for_active` takes precedence
+            await m.wait_for_idle(apps=["dummy_app"], wait_for_active=True, wait_for_status="doesn't matter", timeout=5,
+                                  idle_period=0)
+
+        mock_apps.assert_called()

tox.ini

@@ -10,6 +10,9 @@ skipsdist=True
 [pytest]
 markers =
     serial: mark a test that must run by itself
+filterwarnings =
+    ignore::DeprecationWarning:asynctest
+    ignore::DeprecationWarning:websockets
 
 [testenv]
 usedevelop=True
@@ -49,6 +52,14 @@ commands =
     pip install pylxd
     py.test --tb native -ra -v -n auto -k 'integration' -m 'not serial' {posargs}
 
+[testenv:unit]
+envdir = {toxworkdir}/py3
+commands =
+    # These need to be installed in a specific order
+    pip install urllib3==1.25.7
+    pip install pylxd
+    py.test --tb native -ra -v -n auto {toxinidir}/tests/unit {posargs}
+
 [testenv:serial]
 # tests that can't be run in parallel
 envdir = {toxworkdir}/py3