| Adam Israel | dcdf82b | 2017-08-15 15:26:43 -0400 | [diff] [blame] | 1 | import pytest |
| 2 | import uuid |
| 3 | |
| 4 | from .. import base |
| 5 | from juju.controller import Controller |
| 6 | from juju.errors import JujuAPIError |
| 7 | |
| 8 | |
| 9 | @base.bootstrapped |
| 10 | @pytest.mark.asyncio |
| 11 | async def test_add_user(event_loop): |
| 12 | async with base.CleanController() as controller: |
| 13 | username = 'test{}'.format(uuid.uuid4()) |
| 14 | await controller.add_user(username) |
| 15 | result = await controller.get_user(username) |
| 16 | res_ser = result.serialize()['results'][0].serialize() |
| 17 | assert res_ser['result'] is not None |
| 18 | |
| 19 | |
| 20 | @base.bootstrapped |
| 21 | @pytest.mark.asyncio |
| 22 | async def test_disable_enable_user(event_loop): |
| 23 | async with base.CleanController() as controller: |
| 24 | username = 'test-disable{}'.format(uuid.uuid4()) |
| 25 | await controller.add_user(username) |
| 26 | await controller.disable_user(username) |
| 27 | result = await controller.get_user(username) |
| 28 | res_ser = result.serialize()['results'][0].serialize() |
| 29 | assert res_ser['result'].serialize()['disabled'] is True |
| 30 | await controller.enable_user(username) |
| 31 | result = await controller.get_user(username) |
| 32 | res_ser = result.serialize()['results'][0].serialize() |
| 33 | assert res_ser['result'].serialize()['disabled'] is False |
| 34 | |
| 35 | |
| 36 | @base.bootstrapped |
| 37 | @pytest.mark.asyncio |
| 38 | async def test_change_user_password(event_loop): |
| 39 | async with base.CleanController() as controller: |
| 40 | username = 'test-password{}'.format(uuid.uuid4()) |
| 41 | await controller.add_user(username) |
| 42 | await controller.change_user_password(username, 'password') |
| 43 | try: |
| 44 | new_controller = Controller() |
| 45 | await new_controller.connect( |
| 46 | controller.connection.endpoint, username, 'password') |
| 47 | result = True |
| 48 | await new_controller.disconnect() |
| 49 | except JujuAPIError: |
| 50 | result = False |
| 51 | assert result is True |
| 52 | |
| 53 | |
| 54 | @base.bootstrapped |
| 55 | @pytest.mark.asyncio |
| 56 | async def test_grant(event_loop): |
| 57 | async with base.CleanController() as controller: |
| 58 | username = 'test-grant{}'.format(uuid.uuid4()) |
| 59 | await controller.add_user(username) |
| 60 | await controller.grant(username, 'superuser') |
| 61 | result = await controller.get_user(username) |
| 62 | result = result.serialize()['results'][0].serialize()['result']\ |
| 63 | .serialize() |
| 64 | assert result['access'] == 'superuser' |
| 65 | await controller.grant(username, 'login') |
| 66 | result = await controller.get_user(username) |
| 67 | result = result.serialize()['results'][0].serialize()['result']\ |
| 68 | .serialize() |
| 69 | assert result['access'] == 'login' |
| 70 | |
| 71 | |
| 72 | @base.bootstrapped |
| 73 | @pytest.mark.asyncio |
| 74 | async def test_get_models(event_loop): |
| 75 | async with base.CleanController() as controller: |
| 76 | result = await controller.get_models() |
| 77 | assert isinstance(result.serialize()['user-models'], list) |