def test_inherit_timeout_settings(self):
self.unit_get.return_value = "1.2.4.5"
self.relations_of_type.return_value = [{
"__unit__":
"haproxy/1",
"hostname":
"haproxy-1",
"private-address":
"1.2.4.4",
"all_services":
yaml.dump([
{
"service_name": "foo_service",
"service_host": "0.0.0.0",
"service_options": ["timeout server 5000"],
"service_port": 4242
},
])
}]
services_dict = {
"foo_service": {
"service_name": "foo_service",
"service_host": "0.0.0.0",
"service_port": 4242,
"service_options": ["timeout server 5000"],
"server_options": ["maxconn 4"],
"servers":
[("backend_1__8080", "1.2.3.4", 8080, ["maxconn 4"])],
},
}
expected = {
"foo_service": {
"service_name":
"foo_service",
"service_host":
"0.0.0.0",
"service_port":
4242,
"service_options": [
"balance leastconn", "mode tcp", "option tcplog",
"timeout server 5000"
],
"servers": [("haproxy-1", "1.2.4.4", 4243, ["check"]),
("haproxy-2", "1.2.4.5", 4243, ["check",
"backup"])],
},
"foo_service_be": {
"service_name": "foo_service_be",
"service_host": "0.0.0.0",
"service_port": 4243,
"service_options": ["timeout server 5000"],
"server_options": ["maxconn 4"],
"servers":
[("backend_1__8080", "1.2.3.4", 8080, ["maxconn 4"])],
},
}
self.assertEqual(expected, hooks.apply_peer_config(services_dict))
def test_inherit_timeout_settings(self):
self.unit_get.return_value = "1.2.4.5"
self.relations_of_type.return_value = [
{"__unit__": "haproxy/1",
"hostname": "haproxy-1",
"private-address": "1.2.4.4",
"all_services": yaml.dump([
{"service_name": "foo_service",
"service_host": "0.0.0.0",
"service_options": ["timeout server 5000"],
"service_port": 4242},
])
}
]
services_dict = {
"foo_service": {
"service_name": "foo_service",
"service_host": "0.0.0.0",
"service_port": 4242,
"service_options": ["timeout server 5000"],
"server_options": ["maxconn 4"],
"servers": [("backend_1__8080", "1.2.3.4",
8080, ["maxconn 4"])],
},
}
expected = {
"foo_service": {
"service_name": "foo_service",
"service_host": "0.0.0.0",
"service_port": 4242,
"service_options": ["balance leastconn",
"mode tcp",
"option tcplog",
"timeout server 5000"],
"servers": [
("haproxy-1", "1.2.4.4", 4243, ["check"]),
("haproxy-2", "1.2.4.5", 4243, ["check", "backup"])
],
},
"foo_service_be": {
"service_name": "foo_service_be",
"service_host": "0.0.0.0",
"service_port": 4243,
"service_options": ["timeout server 5000"],
"server_options": ["maxconn 4"],
"servers": [("backend_1__8080", "1.2.3.4",
8080, ["maxconn 4"])],
},
}
self.assertEqual(expected, hooks.apply_peer_config(services_dict))
def test_with_no_relation_data(self):
self.unit_get.return_value = "1.2.4.5"
self.relations_of_type.return_value = []
services_dict = {
"foo_service": {
"service_name": "foo_service",
"service_host": "0.0.0.0",
"service_port": 4242,
"service_options": ["balance leastconn"],
"server_options": ["maxconn 4"],
"servers": [("backend_1__8080", "1.2.3.4",
8080, ["maxconn 4"])],
},
}
expected = services_dict
self.assertEqual(expected, hooks.apply_peer_config(services_dict))
def test_with_no_relation_data(self):
self.unit_get.return_value = "1.2.4.5"
self.relations_of_type.return_value = []
services_dict = {
"foo_service": {
"service_name": "foo_service",
"service_host": "0.0.0.0",
"service_port": 4242,
"service_options": ["balance leastconn"],
"server_options": ["maxconn 4"],
"servers":
[("backend_1__8080", "1.2.3.4", 8080, ["maxconn 4"])],
},
}
expected = services_dict
self.assertEqual(expected, hooks.apply_peer_config(services_dict))
def test_with_missing_all_services(self):
self.unit_get.return_value = "1.2.4.5"
self.relations_of_type.return_value = [{
"__unit__": "haproxy/1",
"hostname": "haproxy-1",
"private-address": "1.2.4.4",
}]
services_dict = {
"foo_service": {
"service_name": "foo_service",
"service_host": "0.0.0.0",
"service_port": 4242,
"service_options": ["balance leastconn"],
"server_options": ["maxconn 4"],
"servers":
[("backend_1__8080", "1.2.3.4", 8080, ["maxconn 4"])],
},
}
expected = services_dict
self.assertEqual(expected, hooks.apply_peer_config(services_dict))
def test_with_missing_all_services(self):
self.unit_get.return_value = "1.2.4.5"
self.relations_of_type.return_value = [
{"__unit__": "haproxy/1",
"hostname": "haproxy-1",
"private-address": "1.2.4.4",
}
]
services_dict = {
"foo_service": {
"service_name": "foo_service",
"service_host": "0.0.0.0",
"service_port": 4242,
"service_options": ["balance leastconn"],
"server_options": ["maxconn 4"],
"servers": [("backend_1__8080", "1.2.3.4",
8080, ["maxconn 4"])],
},
}
expected = services_dict
self.assertEqual(expected, hooks.apply_peer_config(services_dict))