def test_to_json(self):
"""
Test to_json() method.
"""
endpoint1 = Endpoint("aabbccddeeff112233",
"active",
"11-22-33-44-55-66",
if_name="eth0")
assert_equal(endpoint1.ep_id, "aabbccddeeff112233")
assert_equal(endpoint1.state, "active")
assert_equal(endpoint1.mac, "11-22-33-44-55-66")
assert_equal(endpoint1.profile_id, None)
expected = {"state": "active",
"name": "caliaabbccddeef",
"mac": "11-22-33-44-55-66",
"container:if_name": "eth0",
"profile_id": None,
"ipv4_nets": [],
"ipv6_nets": [],
"ipv4_gateway": None,
"ipv6_gateway": None}
assert_dict_equal(json.loads(endpoint1.to_json()), expected)
endpoint1.profile_id = "TEST12"
endpoint1.ipv4_nets.add(IPNetwork("192.168.1.23/32"))
endpoint1.ipv4_gateway = IPAddress("192.168.1.1")
expected["profile_id"] = "TEST12"
expected["ipv4_nets"] = ["192.168.1.23/32"]
expected["ipv4_gateway"] = "192.168.1.1"
assert_dict_equal(json.loads(endpoint1.to_json()), expected)
def test_get_endpoint_exists(self):
"""
Test get_endpoint() for an endpoint that exists.
"""
ep = Endpoint(TEST_ENDPOINT_ID, "active", "11-22-33-44-55-66",
if_name="eth1")
self.etcd_client.read.side_effect = mock_read_for_endpoint(ep)
ep2 = self.datastore.get_endpoint(TEST_HOST,
TEST_CONT_ID,
TEST_ENDPOINT_ID)
assert_equal(ep.to_json(), ep2.to_json())
assert_equal(ep.ep_id, ep2.ep_id)
