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
from unittest.mock import patch
import pynetbox
from .util import Response
host = "http://localhost:8000"
def_kwargs = {
"token": "abc123",
}
# Keys are app names, values are arbitrarily selected endpoints
# We use dcim and ipam since they have unique app classes
# and circuits because it does not. We don't add other apps/endpoints
# beyond 'circuits' as they all use the same code as each other
endpoints = {
"dcim": "devices",
"ipam": "prefixes",
"circuits": "circuits",
}
class ApiTestCase(unittest.TestCase):
@patch(
"requests.sessions.Session.post",
return_value=Response(),
)
def test_get(self, *_):
api = pynetbox.api(host, **def_kwargs)
self.assertTrue(api)
@patch(
"requests.sessions.Session.post",
return_value=Response(),
)
def test_sanitize_url(self, *_):
api = pynetbox.api("http://localhost:8000/", **def_kwargs)
self.assertTrue(api)
self.assertEqual(api.base_url, "http://localhost:8000/api")
class ApiVersionTestCase(unittest.TestCase):
class ResponseHeadersWithVersion:
headers = {"API-Version": "1.999"}
ok = True
@patch(
"requests.sessions.Session.get",
return_value=ResponseHeadersWithVersion(),
)
def test_api_version(self, *_):
api = pynetbox.api(
host,
)
self.assertEqual(api.version, "1.999")
class ResponseHeadersWithoutVersion:
headers = {}
ok = True
@patch(
"requests.sessions.Session.get",
return_value=ResponseHeadersWithoutVersion(),
)
def test_api_version_not_found(self, *_):
api = pynetbox.api(
host,
)
self.assertEqual(api.version, "")
class ApiStatusTestCase(unittest.TestCase):
class ResponseWithStatus:
ok = True
def json(self):
return {
"netbox-version": "0.9.9",
}
@patch(
"requests.sessions.Session.get",
return_value=ResponseWithStatus(),
)
def test_api_status(self, *_):
api = pynetbox.api(
host,
)
self.assertEqual(api.status()["netbox-version"], "0.9.9")
class ApiCreateTokenTestCase(unittest.TestCase):
@patch(
"requests.sessions.Session.post",
return_value=Response(fixture="api/token_provision.json"),
)
def test_create_token(self, *_):
api = pynetbox.api(host)
token = api.create_token("user", "pass")
self.assertTrue(isinstance(token, pynetbox.core.response.Record))
self.assertEqual(token.key, "1234567890123456789012345678901234567890")
self.assertEqual(api.token, "1234567890123456789012345678901234567890")
|
