def test_node_remove_specific_host(self, m_client, m_cont_running, m_veth):
"""
Test the client removes the specific host when node_remove called, and
that endpoints are removed when remove_endpoints flag is set.
"""
# Call method under test
endpoint1 = Mock()
endpoint1.name = "vethname1"
endpoint1.ipv4_nets = {IPNetwork("1.2.3.4/32")}
endpoint1.ipv6_nets = set()
endpoint2 = Mock()
endpoint2.name = "vethname2"
endpoint2.ipv4_nets = set()
endpoint2.ipv6_nets = set()
m_client.get_endpoints.return_value = [endpoint1, endpoint2]
# This should not cause a failure with specific host
m_cont_running.return_value = True
node.node_remove(True, "other-host")
# Assert
m_client.get_endpoints.assert_called_once_with(hostname="other-host")
m_client.release_ips.assert_called_once_with({IPAddress("1.2.3.4")})
m_client.remove_ipam_host.assert_called_once_with("other-host")
m_client.remove_host.assert_called_once_with("other-host")
m_veth.assert_has_calls([call("vethname1"), call("vethname2")])
def test_node_remove(self, m_client, m_cont_running, m_veth):
"""
Test the client removes the host when node_remove called, and that
endpoints are removed when remove_endpoints flag is set.
"""
# Call method under test
endpoint1 = Mock()
endpoint1.name = "vethname1"
endpoint1.ipv4_nets = {IPNetwork("1.2.3.4/32")}
endpoint1.ipv6_nets = set()
endpoint2 = Mock()
endpoint2.name = "vethname2"
endpoint2.ipv4_nets = set()
endpoint2.ipv6_nets = {IPNetwork("aa:bb::cc/128")}
m_client.get_endpoints.return_value = [endpoint1, endpoint2]
node.node_remove(True, False)
# Assert
m_client.get_endpoints.assert_called_once_with(hostname=node.hostname)
m_client.release_ips.assert_called_once_with(
{IPAddress("1.2.3.4"),
IPAddress("aa:bb::cc")})
m_client.remove_ipam_host.assert_called_once_with(node.hostname)
m_veth.assert_has_calls([call("vethname1"), call("vethname2")])
m_cont_running.assert_has_calls(
[call("calico-node"),
call("calico-libnetwork")])
m_client.remove_host.assert_called_once_with(node.hostname)
def test_node_remove_specific_host(self, m_client, m_cont_running, m_veth):
"""
Test the client removes the specific host when node_remove called, and
that endpoints are removed when remove_endpoints flag is set.
"""
# Call method under test
endpoint1 = Mock()
endpoint1.name = "vethname1"
endpoint1.ipv4_nets = {IPNetwork("1.2.3.4/32")}
endpoint1.ipv6_nets = set()
endpoint2 = Mock()
endpoint2.name = "vethname2"
endpoint2.ipv4_nets = set()
endpoint2.ipv6_nets = set()
m_client.get_endpoints.return_value = [endpoint1, endpoint2]
# This should not cause a failure with specific host
m_cont_running.return_value = True
node.node_remove(True, "other-host")
# Assert
m_client.get_endpoints.assert_called_once_with(hostname="other-host")
m_client.release_ips.assert_called_once_with({IPAddress("1.2.3.4")})
m_client.remove_ipam_host.assert_called_once_with("other-host")
m_client.remove_host.assert_called_once_with("other-host")
m_veth.assert_has_calls([call("vethname1"), call("vethname2")])
def test_node_remove(self, m_client, m_cont_running, m_veth):
"""
Test the client removes the host when node_remove called, and that
endpoints are removed when remove_endpoints flag is set.
"""
# Call method under test
endpoint1 = Mock()
endpoint1.name = "vethname1"
endpoint1.ipv4_nets = {IPNetwork("1.2.3.4/32")}
endpoint1.ipv6_nets = set()
endpoint2 = Mock()
endpoint2.name = "vethname2"
endpoint2.ipv4_nets = set()
endpoint2.ipv6_nets = {IPNetwork("aa:bb::cc/128")}
m_client.get_endpoints.return_value = [endpoint1, endpoint2]
node.node_remove(True, False)
# Assert
m_client.get_endpoints.assert_called_once_with(hostname=node.hostname)
m_client.release_ips.assert_called_once_with({IPAddress("1.2.3.4"),
IPAddress("aa:bb::cc")})
m_client.remove_ipam_host.assert_called_once_with(node.hostname)
m_veth.assert_has_calls([call("vethname1"), call("vethname2")])
m_cont_running.assert_has_calls([call("calico-node")])
m_client.remove_host.assert_called_once_with(node.hostname)
def test_node_remove(self, m_client, m_cont_running, m_veth):
"""
Test the client removes the host when node_remove called, and that
endpoints are removed when remove_endpoints flag is set.
"""
# Call method under test
endpoint1 = Mock()
endpoint1.name = "vethname1"
endpoint2 = Mock()
endpoint2.name = "vethname2"
m_client.get_endpoints.return_value = [endpoint1, endpoint2]
node.node_remove(True)
# Assert
m_client.get_endpoints.assert_called_once_with(hostname=node.hostname)
m_client.remove_host.assert_called_once_with(node.hostname)
m_veth.assert_has_calls([call("vethname1"), call("vethname2")])
m_cont_running.assert_has_calls([call("calico-node"), call("calico-libnetwork")])
def test_node_remove(self, m_client, m_cont_running, m_veth):
"""
Test the client removes the host when node_remove called, and that
endpoints are removed when remove_endpoints flag is set.
"""
# Call method under test
endpoint1 = Mock()
endpoint1.name = "vethname1"
endpoint2 = Mock()
endpoint2.name = "vethname2"
m_client.get_endpoints.return_value = [endpoint1, endpoint2]
node.node_remove(True)
# Assert
m_client.get_endpoints.assert_called_once_with(hostname=node.hostname)
m_client.remove_host.assert_called_once_with(node.hostname)
m_veth.assert_has_calls([call("vethname1"), call("vethname2")])
m_cont_running.assert_has_calls(
[call("calico-node"),
call("calico-libnetwork")])