forked from linode/linode_api4-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathnodebalancers_test.py
More file actions
112 lines (97 loc) · 4.11 KB
/
Copy pathnodebalancers_test.py
File metadata and controls
112 lines (97 loc) · 4.11 KB
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
from datetime import datetime
from test.base import ClientBaseCase
from linode_api4.objects import NodeBalancerConfig, NodeBalancerNode
from linode_api4.objects.base import MappedObject
class NodeBalancerConfigTest(ClientBaseCase):
"""
Tests methods of the NodeBalancerConfig class
"""
def test_get_config(self):
"""
Tests that a config is loaded correctly by ID
"""
config = NodeBalancerConfig(self.client, 65432, 123456)
self.assertEqual(config._populated, False)
self.assertEqual(config.port, 80)
self.assertEqual(config._populated, True)
self.assertEqual(config.check, "connection")
self.assertEqual(config.protocol, "http")
self.assertEqual(config.check_attempts, 2)
self.assertEqual(config.stickiness, "table")
self.assertEqual(config.check_interval, 5)
self.assertEqual(config.check_body, "")
self.assertEqual(config.check_passive, True)
self.assertEqual(config.algorithm, "roundrobin")
self.assertEqual(config.check_timeout, 3)
self.assertEqual(config.check_path, "/")
self.assertEqual(config.ssl_cert, None)
self.assertEqual(config.ssl_commonname, "")
self.assertEqual(config.nodebalancer_id, 123456)
self.assertEqual(config.cipher_suite, "recommended")
self.assertEqual(config.ssl_key, None)
self.assertEqual(config.nodes_status.up, 0)
self.assertEqual(config.nodes_status.down, 0)
self.assertEqual(config.ssl_fingerprint, "")
self.assertEqual(config.proxy_protocol, "none")
class NodeBalancerNodeTest(ClientBaseCase):
"""
Tests methods of the NodeBalancerNode class
"""
def test_get_node(self):
"""
Tests that a node is loaded correctly by ID
"""
node = NodeBalancerNode(self.client, 54321, (65432, 123456))
self.assertEqual(node._populated, False)
self.assertEqual(node.weight, 50)
self.assertEqual(node._populated, True)
self.assertEqual(node.id, 54321)
self.assertEqual(node.address, "192.168.210.120")
self.assertEqual(node.label, "node54321")
self.assertEqual(node.status, "UP")
self.assertEqual(node.mode, "accept")
self.assertEqual(node.config_id, 65432)
self.assertEqual(node.nodebalancer_id, 123456)
def test_create_node(self):
"""
Tests that a node can be created
"""
with self.mock_post('nodebalancers/123456/configs/65432/nodes/54321') as m:
config = NodeBalancerConfig(self.client, 65432, 123456)
node = config.node_create('node54321', '192.168.210.120',
weight=50, mode='accept')
self.assertIsNotNone(node)
self.assertEqual(node.id, 54321)
self.assertEqual(m.call_url, '/nodebalancers/123456/configs/65432/nodes')
self.assertEqual(m.call_data, {
"label": "node54321",
"address": "192.168.210.120",
"weight": 50,
"mode": "accept"
})
def test_update_node(self):
"""
Tests that a node can be updated
"""
with self.mock_put('nodebalancers/123456/configs/65432/nodes/54321') as m:
node = self.client.load(NodeBalancerNode, 54321, (65432, 123456))
node.label = "ThisNewLabel"
node.weight = 60
node.mode = "drain"
node.address = "192.168.210.121"
node.save()
self.assertEqual(m.call_url, '/nodebalancers/123456/configs/65432/nodes/54321')
self.assertEqual(m.call_data, {
"label": "ThisNewLabel",
"address": "192.168.210.121",
"mode": "drain",
"weight": 60
})
def test_delete_node(self):
"""
Tests that deleting a node creates the correct api request.
"""
with self.mock_delete() as m:
node = NodeBalancerNode(self.client, 54321, (65432, 123456))
node.delete()
self.assertEqual(m.call_url, '/nodebalancers/123456/configs/65432/nodes/54321')