1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
# Software Name: MOON
# Version: 5.4
# SPDX-FileCopyrightText: Copyright (c) 2018-2020 Orange and its contributors
# SPDX-License-Identifier: Apache-2.0
# This software is distributed under the 'Apache License 2.0',
# the text of which is available at 'http://www.apache.org/licenses/LICENSE-2.0.txt'
# or see the "LICENSE" file for more details.
from falcon import HTTP_200, HTTP_204, HTTP_401
import hug
import base64
from uuid import uuid4
from helpers import data_builder as builder
def test_get_auth():
from moon_utilities.auth_functions import get_api_key_for_user
from moon_manager.api import auth
from moon_manager.api import policy
headers = {"Authorization": "Basic {}".format(base64.b64encode(b"admin:admin").decode("utf-8"))}
req = hug.test.get(auth, 'auth/', headers=headers)
assert req.status == HTTP_200
key = req.data
assert get_api_key_for_user("admin") == req.data
headers = {"x-api-key": key}
req = hug.test.get(policy, 'policies/', headers=headers)
assert req.status == HTTP_200
def test_del_auth():
from moon_utilities.auth_functions import get_api_key_for_user
from moon_manager.api import auth
from moon_manager.api import policy
headers = {"Authorization": "Basic {}".format(base64.b64encode(b"admin:admin").decode("utf-8"))}
req = hug.test.get(auth, 'auth/', headers=headers)
assert req.status == HTTP_200
key = req.data
headers = {"x-api-key": key}
req = hug.test.delete(auth, 'auth/', headers=headers)
assert req.status == HTTP_204
req = hug.test.get(policy, 'policies/', headers=headers)
assert req.status == HTTP_401
assert not get_api_key_for_user("admin")
def test_readd_auth():
from moon_utilities.auth_functions import get_api_key_for_user
from moon_manager.api import auth
from moon_manager.api import policy
headers = {"Authorization": "Basic {}".format(base64.b64encode(b"admin:admin").decode("utf-8"))}
req = hug.test.get(auth, 'auth/', headers=headers)
assert req.status == HTTP_200
key = req.data
headers = {"x-api-key": key}
req = hug.test.delete(auth, 'auth/', headers=headers)
assert req.status == HTTP_204
headers = {"Authorization": "Basic {}".format(base64.b64encode(b"admin:admin").decode("utf-8"))}
req = hug.test.get(auth, 'auth/', headers=headers)
assert req.status == HTTP_200
new_key = req.data
headers = {"x-api-key": new_key}
req = hug.test.get(policy, 'policies/', headers=headers)
assert req.status == HTTP_200
assert get_api_key_for_user("admin")
assert get_api_key_for_user("admin") == new_key
assert get_api_key_for_user("admin") != key
|
