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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
import unittest
from unittest.mock import patch
import pynetbox
from .util import Response
api = pynetbox.api(
"http://localhost:8000",
)
nb = api.users
HEADERS = {"accept": "application/json"}
class Generic:
class Tests(unittest.TestCase):
name = ""
ret = pynetbox.core.response.Record
app = "users"
def test_get_all(self):
with patch(
"requests.sessions.Session.get",
return_value=Response(fixture="{}/{}.json".format(self.app, self.name)),
) as mock:
ret = list(getattr(nb, self.name).all())
self.assertTrue(ret)
self.assertTrue(isinstance(ret[0], self.ret))
mock.assert_called_with(
"http://localhost:8000/api/{}/{}/".format(
self.app, self.name.replace("_", "-")
),
params={"limit": 0},
json=None,
headers=HEADERS,
)
def test_filter(self):
with patch(
"requests.sessions.Session.get",
return_value=Response(fixture="{}/{}.json".format(self.app, self.name)),
) as mock:
ret = list(getattr(nb, self.name).filter(name="test"))
self.assertTrue(ret)
self.assertTrue(isinstance(ret[0], self.ret))
mock.assert_called_with(
"http://localhost:8000/api/{}/{}/".format(
self.app, self.name.replace("_", "-")
),
params={"name": "test", "limit": 0},
json=None,
headers=HEADERS,
)
def test_get(self):
with patch(
"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,
)
class UsersTestCase(Generic.Tests):
name = "users"
@patch(
"requests.sessions.Session.get",
return_value=Response(fixture="users/user.json"),
)
def test_repr(self, _):
test = nb.users.get(1)
self.assertEqual(type(test), pynetbox.models.users.Users)
self.assertEqual(str(test), "user1")
class GroupsTestCase(Generic.Tests):
name = "groups"
class PermissionsTestCase(Generic.Tests):
name = "permissions"
@patch(
"requests.sessions.Session.get",
return_value=Response(fixture="users/permission.json"),
)
def test_username(self, _):
permission = nb.permissions.get(1)
self.assertEqual(len(permission.users), 1)
user = permission.users[0]
self.assertEqual(str(user), "user1")
@patch(
"requests.sessions.Session.get",
return_value=Response(fixture="users/permission.json"),
)
def test_constraints(self, _):
permission = nb.permissions.get(1)
self.assertIsInstance(permission.constraints, list)
self.assertIsInstance(permission.constraints[0], dict)
class UnknownModelTestCase(unittest.TestCase):
"""This test validates that an unknown model is returned as Record object
and that the __str__() method correctly uses the 'display' field of the
object (introduced as a standard field in NetBox 2.11.0).
"""
@patch(
"requests.sessions.Session.get",
return_value=Response(fixture="users/unknown_model.json"),
)
def test_unknown_model(self, _):
unknown_obj = nb.unknown_model.get(1)
self.assertEqual(type(unknown_obj), pynetbox.core.response.Record)
self.assertEqual(str(unknown_obj), "Unknown object")
|
