Merge pull request #23 from petevg/bug/instance-placement-bug

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

diff --git a/tests/client/test_client.py b/tests/client/test_client.py
index ed386d3..8598878 100644 (file)
--- a/tests/client/test_client.py
+++ b/tests/client/test_client.py
@@ -2,7 +2,7 @@ import asyncio
 import unittest
 
 from juju.client.connection import Connection
-from juju.client.client import UserManager, Entity
+from juju.client import client
 
 from ..base import bootstrapped
 
@@ -13,10 +13,14 @@ class UserManagerTest(unittest.TestCase):
         loop = asyncio.get_event_loop()
         conn = loop.run_until_complete(
             Connection.connect_current())
+        conn = loop.run_until_complete(
+            conn.controller())
 
-        um = UserManager()
+        um = client.UserManagerFacade()
         um.connect(conn)
         result = loop.run_until_complete(
-            um.UserInfo([Entity('user-admin')], True))
+            um.UserInfo([client.Entity('user-admin')], True))
 
-        assert result
+        self.assertIsInstance(result, client.UserInfoResults)
+        for r in result.results:
+            self.assertIsInstance(r, client.UserInfoResult)