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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
# Copyright 2015 Open Platform for NFV Project, Inc. and its contributors
# This software is distributed under the terms and conditions of the 'Apache-2.0'
# license which can be found in the file 'LICENSE' in this package distribution
# or at 'http://www.apache.org/licenses/LICENSE-2.0'.
import moon_db
import uuid
Connector = moon_db.Driver("sql", "mysql+pymysql://moonuser:password@localhost/moon")
Connector.driver.init_db()
def test_get_tenants():
t = Connector.driver.get_tenants_dict()
print(t)
assert type(t) == dict
def test_add_tenant():
new_tenant = {
"id": uuid.uuid4().hex,
"name": "demo",
"description": uuid.uuid4().hex,
"intra_authz_extension_id": "",
"intra_admin_extension_id": "",
}
data = Connector.driver.add_tenant_dict(tenant_id=new_tenant['id'],
tenant_dict=new_tenant)
data_id = list(data.keys())[0]
assert new_tenant["id"] == data_id
assert new_tenant["name"] == data[data_id]["name"]
assert new_tenant["intra_authz_extension_id"] == data[data_id]["intra_authz_extension_id"]
assert new_tenant["intra_admin_extension_id"] == data[data_id]["intra_admin_extension_id"]
data = Connector.driver.get_tenants_dict()
assert data != {}
def test_del_tenant():
new_tenant = {
"id": uuid.uuid4().hex,
"name": "demo",
"description": uuid.uuid4().hex,
"intra_authz_extension_id": "",
"intra_admin_extension_id": "",
}
data = Connector.driver.get_tenants_dict()
number_of_tenant = len(data.keys())
data = Connector.driver.add_tenant_dict(tenant_id=new_tenant['id'],
tenant_dict=new_tenant)
data_id = list(data.keys())[0]
assert new_tenant["name"] == data[data_id]["name"]
assert new_tenant["intra_authz_extension_id"] == data[data_id]["intra_authz_extension_id"]
assert new_tenant["intra_admin_extension_id"] == data[data_id]["intra_admin_extension_id"]
data = Connector.driver.get_tenants_dict()
assert len(data.keys()) == number_of_tenant+1
Connector.driver.del_tenant(data_id)
data = Connector.driver.get_tenants_dict()
assert len(data.keys()) == number_of_tenant
def test_set_tenant():
new_tenant = {
"id": uuid.uuid4().hex,
"name": "demo",
"description": uuid.uuid4().hex,
"intra_authz_extension_id": "123456",
"intra_admin_extension_id": "0987654",
}
data = Connector.driver.get_tenants_dict()
number_of_tenant = len(data.keys())
data = Connector.driver.add_tenant_dict(tenant_id=new_tenant['id'],
tenant_dict=new_tenant)
data_id = list(data.keys())[0]
assert new_tenant["name"] == data[data_id]["name"]
assert new_tenant["intra_authz_extension_id"] == data[data_id]["intra_authz_extension_id"]
assert new_tenant["intra_admin_extension_id"] == data[data_id]["intra_admin_extension_id"]
data = Connector.driver.get_tenants_dict()
assert len(data.keys()) == number_of_tenant+1
new_tenant["name"] = "demo2"
data = Connector.driver.set_tenant_dict(tenant_id=data_id, tenant_dict=new_tenant)
data_id = list(data.keys())[0]
assert new_tenant["name"] == data[data_id]["name"]
assert new_tenant["intra_authz_extension_id"] == data[data_id]["intra_authz_extension_id"]
assert new_tenant["intra_admin_extension_id"] == data[data_id]["intra_admin_extension_id"]
|
