import unittest
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):
+ def test_user_info(self):
loop = asyncio.get_event_loop()
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)