-
Notifications
You must be signed in to change notification settings - Fork 181
/
Copy pathtest_wireless.py
87 lines (74 loc) · 2.92 KB
/
test_wireless.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
import unittest
from unittest.mock import patch
import pynetbox
import pynetbox.models.wireless
from .util import Response
api = pynetbox.api("http://localhost:8000")
nb_app = api.wireless
HEADERS = {"accept": "application/json"}
class Generic:
class Tests(unittest.TestCase):
name = ""
ret = pynetbox.core.response.Record
app = "wireless"
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_app, 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_app, 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_app, 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 WirelessLansTestCase(Generic.Tests):
name = "wireless_lans"
ret = pynetbox.models.wireless.WirelessLans
@patch(
"requests.sessions.Session.get",
return_value=Response(fixture="wireless/wireless_lan.json"),
)
def test_repr(self, _):
wireless_lan_obj = nb_app.wireless_lans.get(1)
self.assertEqual(type(wireless_lan_obj), pynetbox.models.wireless.WirelessLans)
self.assertEqual(str(wireless_lan_obj), "SSID 1")