def test_container_remove(self, m_netns, m_client, m_get_container_id,
m_enforce_root):
"""
Test for container_remove of calicoctl container command
"""
# Set up mock objects
m_get_container_id.return_value = 666
ipv4_nets = set()
ipv4_nets.add(IPNetwork(IPAddress('1.1.1.1')))
ipv6_nets = set()
m_endpoint = Mock(spec=Endpoint)
m_endpoint.ipv4_nets = ipv4_nets
m_endpoint.ipv6_nets = ipv6_nets
m_endpoint.endpoint_id = 12
m_endpoint.name = "eth1234"
ippool = IPPool('1.1.1.1/24')
m_client.get_endpoint.return_value = m_endpoint
m_client.get_ip_pools.return_value = [ippool]
# Call method under test
container.container_remove('container1')
# Assert
m_enforce_root.assert_called_once_with()
m_get_container_id.assert_called_once_with('container1')
m_client.get_endpoint.assert_called_once_with(
hostname=utils.hostname,
orchestrator_id=utils.DOCKER_ORCHESTRATOR_ID,
workload_id=666
)
self.assertEqual(m_client.unassign_address.call_count, 1)
m_netns.remove_veth.assert_called_once_with("eth1234")
m_client.remove_workload.assert_called_once_with(
utils.hostname, utils.DOCKER_ORCHESTRATOR_ID, 666)
def test_container_remove(self, m_netns, m_client, m_get_container_id,
m_enforce_root):
"""
Test for container_remove of calicoctl container command
"""
# Set up mock objects
m_get_container_id.return_value = 666
ipv4_nets = set()
ipv4_nets.add(IPNetwork(IPAddress('1.1.1.1')))
ipv6_nets = set()
m_endpoint = Mock(spec=Endpoint)
m_endpoint.ipv4_nets = ipv4_nets
m_endpoint.ipv6_nets = ipv6_nets
m_endpoint.endpoint_id = 12
m_endpoint.name = "eth1234"
ippool = IPPool('1.1.1.1/24')
m_client.get_endpoint.return_value = m_endpoint
m_client.get_ip_pools.return_value = [ippool]
# Call method under test
container.container_remove('container1')
# Assert
m_enforce_root.assert_called_once_with()
m_get_container_id.assert_called_once_with('container1')
m_client.get_endpoint.assert_called_once_with(
hostname=utils.hostname,
orchestrator_id=utils.ORCHESTRATOR_ID,
workload_id=666)
self.assertEqual(m_client.unassign_address.call_count, 1)
m_netns.remove_veth.assert_called_once_with("eth1234")
m_client.remove_workload.assert_called_once_with(
utils.hostname, utils.ORCHESTRATOR_ID, 666)
def test_container_remove(self, m_netns, m_client, m_enforce_root):
"""
Test for container_remove of calicoctl container command
"""
# Set up mock objects
ipv4_nets = set()
ipv4_nets.add(IPNetwork(IPAddress('1.1.1.1')))
ipv6_nets = set()
m_endpoint = Mock(spec=Endpoint)
m_endpoint.ipv4_nets = ipv4_nets
m_endpoint.ipv6_nets = ipv6_nets
m_endpoint.endpoint_id = 12
m_endpoint.name = "eth1234"
m_client.get_endpoints.return_value = [m_endpoint]
# Call method under test
container.container_remove('container1')
# Assert
m_enforce_root.assert_called_once_with()
m_client.get_endpoints.assert_called_once_with(
hostname=utils.hostname,
orchestrator_id=utils.DOCKER_ORCHESTRATOR_ID,
workload_id='container1'
)
self.assertEqual(m_client.release_ips.call_count, 1)
m_netns.remove_veth.assert_called_once_with("eth1234")
m_client.remove_workload.assert_called_once_with(
utils.hostname, utils.DOCKER_ORCHESTRATOR_ID, "container1")
def test_container_remove_with_lookup(self, m_netns, m_client,
m_get_workload_id, m_enforce_root):
"""
Test for container_remove of calicoctl container command
"""
# Set up mock objects
m_get_workload_id.return_value = "long_id"
m_client.get_endpoints.return_value = []
# Call method under test
container.container_remove('container1')
# Assert
m_enforce_root.assert_called_once_with()
self.assertEqual(m_client.release_ips.call_count, 0)
self.assertEqual(m_netns.remove_veth.call_count, 0)
self.assertEqual(m_client.get_endpoints.call_args_list,
[call(hostname=utils.hostname, workload_id='container1', orchestrator_id='docker'),
call(hostname=utils.hostname, workload_id='long_id', orchestrator_id='docker')])
m_client.remove_workload.assert_called_once_with(
utils.hostname, utils.DOCKER_ORCHESTRATOR_ID, "long_id")
