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
|
import unittest
from unittest.mock import patch
import pynetbox
nb = pynetbox.api("http://localhost:8000")
class DetailEndpointTestCase(unittest.TestCase):
def test_detail_endpoint_create_single(self):
# Prefixes
with patch(
"pynetbox.core.query.Request._make_call",
return_value={"id": 123, "prefix": "1.2.3.0/24"},
):
prefix_obj = nb.ipam.prefixes.get(123)
self.assertEqual(prefix_obj.prefix, "1.2.3.0/24")
with patch(
"pynetbox.core.query.Request._make_call",
return_value={"address": "1.2.3.1/24"},
):
ip_obj = prefix_obj.available_ips.create()
self.assertEqual(ip_obj.address, "1.2.3.1/24")
# IP Ranges
with patch(
"pynetbox.core.query.Request._make_call",
return_value={"id": 321, "display": "1.2.4.1-254/24"},
):
ip_range_obj = nb.ipam.ip_ranges.get(321)
self.assertEqual(ip_range_obj.display, "1.2.4.1-254/24")
with patch(
"pynetbox.core.query.Request._make_call",
return_value={"address": "1.2.4.2/24"},
):
ip_obj = ip_range_obj.available_ips.create()
self.assertEqual(ip_obj.address, "1.2.4.2/24")
def test_detail_endpoint_create_list(self):
# Prefixes
with patch(
"pynetbox.core.query.Request._make_call",
return_value={"id": 123, "prefix": "1.2.3.0/24"},
):
prefix_obj = nb.ipam.prefixes.get(123)
self.assertEqual(prefix_obj.prefix, "1.2.3.0/24")
with patch(
"pynetbox.core.query.Request._make_call",
return_value=[{"address": "1.2.3.1/24"}, {"address": "1.2.3.2/24"}],
):
ip_list = prefix_obj.available_ips.create([{} for _ in range(2)])
self.assertTrue(isinstance(ip_list, list))
self.assertEqual(len(ip_list), 2)
# IP Ranges
with patch(
"pynetbox.core.query.Request._make_call",
return_value={"id": 321, "display": "1.2.4.1-254/24"},
):
ip_range_obj = nb.ipam.ip_ranges.get(321)
self.assertEqual(ip_range_obj.display, "1.2.4.1-254/24")
with patch(
"pynetbox.core.query.Request._make_call",
return_value=[{"address": "1.2.4.2/24"}, {"address": "1.2.4.3/24"}],
):
ip_list = ip_range_obj.available_ips.create([{} for _ in range(2)])
self.assertTrue(isinstance(ip_list, list))
self.assertEqual(len(ip_list), 2)
|
