X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=tests%2Fclient%2Ftest_client.py;h=e31a69a3d589176d61cf609005cf06f91d15f8d4;hb=7291882b61e1482db38a9f37cc6a447439227d59;hp=ed386d323388dba917efd82fdf097b676c23afa6;hpb=386838416289efe693bd7b02b14ecc21c443dec2;p=osm%2FN2VC.git

diff --git a/tests/client/test_client.py b/tests/client/test_client.py
index ed386d3..e31a69a 100644
--- 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
 
@@ -14,9 +14,11 @@ class UserManagerTest(unittest.TestCase):
         conn = loop.run_until_complete(
             Connection.connect_current())
 
-        um = UserManager()
+        um = client.UserManager()
         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)