Add functional tests for Model.add_machine()

[osm/N2VC.git] / tests / client / test_client.py

diff --git a/tests/client/test_client.py b/tests/client/test_client.py
index 876470e..653da5b 100644 (file)
--- a/tests/client/test_client.py
+++ b/tests/client/test_client.py
@@ -1,22 +1,24 @@
-import asyncio
-import unittest
+import pytest
 
 from juju.client.connection import Connection
-from juju.client.client import UserManager, Entity
+from juju.client import client
 
 from ..base import bootstrapped
 
 
 @bootstrapped
-class UserManagerTest(unittest.TestCase):
-    def test_connect_current(self):
-        loop = asyncio.get_event_loop()
-        conn = loop.run_until_complete(
-            Connection.connect_current())
+@pytest.mark.asyncio
+async def test_user_info(event_loop):
+    conn = await Connection.connect_current()
+    controller_conn = await conn.controller()
 
-        um = UserManager()
-        um.connect(conn)
-        result = loop.run_until_complete(
-            um.UserInfo([Entity('user-admin')], True))
+    um = client.UserManagerFacade()
+    um.connect(controller_conn)
+    result = await um.UserInfo(
+        [client.Entity('user-admin')], True)
+    await conn.close()
+    await controller_conn.close()
 
-        assert result
+    assert isinstance(result, client.UserInfoResults)
+    for r in result.results:
+        assert isinstance(r, client.UserInfoResult)