c334389c69cb0d84e0a6a5fadcfffa0f48a0ea5e
2 from concurrent
.futures
import ThreadPoolExecutor
6 from juju
.controller
import Controller
7 from juju
.errors
import JujuAPIError
12 async def test_add_user(event_loop
):
13 async with base
.CleanController() as controller
:
14 await controller
.add_user('test')
15 result
= await controller
.get_user('test')
16 res_ser
= result
.serialize()['results'][0].serialize()
17 assert res_ser
['result'] is not None
22 async def test_disable_enable_user(event_loop
):
23 async with base
.CleanController() as controller
:
24 await controller
.add_user('test-disable')
25 await controller
.disable_user('test-disable')
26 result
= await controller
.get_user('test-disable')
27 res_ser
= result
.serialize()['results'][0].serialize()
28 assert res_ser
['result'].serialize()['disabled'] is True
29 await controller
.enable_user('test-disable')
30 result
= await controller
.get_user('test-disable')
31 res_ser
= result
.serialize()['results'][0].serialize()
32 assert res_ser
['result'].serialize()['disabled'] is False
37 async def test_change_user_password(event_loop
):
38 async with base
.CleanController() as controller
:
39 await controller
.add_user('test-password')
40 await controller
.change_user_password('test-password', 'password')
42 con
= await controller
.connect(controller
.connection
.endpoint
, 'test-password', 'password')
51 async def test_grant(event_loop
):
52 async with base
.CleanController() as controller
:
53 await controller
.add_user('test-grant')
54 await controller
.grant('test-grant', 'superuser')
55 result
= await controller
.get_user('test-grant')
56 result
= result
.serialize()['results'][0].serialize()['result'].serialize()
57 assert result
['access'] == 'superuser'
58 await controller
.grant('test-grant', 'login')
59 result
= await controller
.get_user('test-grant')
60 result
= result
.serialize()['results'][0].serialize()['result'].serialize()
61 assert result
['access'] == 'login'
66 async def test_get_models(event_loop
):
67 async with base
.CleanController() as controller
:
68 result
= await controller
.get_models()
69 assert isinstance(result
.serialize()['user-models'], list)