/
test_tenancy.py
104 lines (89 loc) · 3.04 KB
/
test_tenancy.py
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
import unittest
import six
import pynetbox
from .util import Response
if six.PY3:
from unittest.mock import patch
else:
from mock import patch
api = pynetbox.api(
"http://localhost:8000",
)
nb = api.tenancy
HEADERS = {
'accept': 'application/json;'
}
class Generic(object):
class Tests(unittest.TestCase):
name = ''
ret = pynetbox.core.response.Record
app = 'tenancy'
def test_get_all(self):
with patch(
'pynetbox.core.query.requests.sessions.Session.get',
return_value=Response(fixture='{}/{}.json'.format(
self.app,
self.name
))
) as mock:
ret = getattr(nb, self.name).all()
self.assertTrue(ret)
self.assertTrue(isinstance(ret, list))
self.assertTrue(isinstance(ret[0], self.ret))
mock.assert_called_with(
'http://localhost:8000/api/{}/{}/'.format(
self.app,
self.name.replace('_', '-')
),
params={},
json=None,
headers=HEADERS,
verify=True
)
def test_filter(self):
with patch(
'pynetbox.core.query.requests.sessions.Session.get',
return_value=Response(fixture='{}/{}.json'.format(
self.app,
self.name
))
) as mock:
ret = getattr(nb, self.name).filter(name='test')
self.assertTrue(ret)
self.assertTrue(isinstance(ret, list))
self.assertTrue(isinstance(ret[0], self.ret))
mock.assert_called_with(
'http://localhost:8000/api/{}/{}/'.format(
self.app,
self.name.replace('_', '-')
),
params={"name": "test"},
json=None,
headers=HEADERS,
verify=True
)
def test_get(self):
with patch(
'pynetbox.core.query.requests.sessions.Session.get',
return_value=Response(fixture='{}/{}.json'.format(
self.app,
self.name[:-1]
))
) as mock:
ret = getattr(nb, self.name).get(1)
self.assertTrue(ret)
self.assertTrue(isinstance(ret, self.ret))
mock.assert_called_with(
'http://localhost:8000/api/{}/{}/1/'.format(
self.app,
self.name.replace('_', '-')
),
params={},
json=None,
headers=HEADERS,
verify=True
)
class TenantsTestCase(Generic.Tests):
name = 'tenants'
class TenantGroupsTestCase(Generic.Tests):
name = 'tenant_groups'