def test_node_stop(self, m_docker_client, m_client):
"""
Test the client removes the host and stops the node when node_stop
called
"""
# Call method under test
node.node_stop(True)
# Assert
m_client.remove_host.assert_called_once_with(node.hostname)
m_docker_client.stop.assert_called_once_with('calico-node')
def test_node_stop(self, m_docker_client, m_client):
"""
Test the client removes the host and stops the node when node_stop
called
"""
# Call method under test
node.node_stop(True)
# Assert
m_client.remove_host.assert_called_once_with(node.hostname)
m_docker_client.stop.assert_called_once_with('calico-node')
def test_node_stop(self, m_docker_client, m_client):
"""
Test the client stops the node when node_stop called when there are
endpoints and the force flag is set.
"""
# Call method under test
m_client.get_endpoints.return_value = [Mock()]
node.node_stop(True)
# Assert
m_client.get_endpoints.assert_called_once_with(hostname=node.hostname)
m_docker_client.stop.assert_has_calls([call("calico-node"), call("calico-libnetwork")])
def test_node_stop(self, m_docker_client, m_client):
"""
Test the client stops the node when node_stop called when there are
endpoints and the force flag is set.
"""
# Call method under test
m_client.get_endpoints.return_value = [Mock()]
node.node_stop(True)
# Assert
m_client.get_endpoints.assert_called_once_with(hostname=node.hostname)
m_docker_client.stop.assert_has_calls([call('calico-node'),
call('calico-libnetwork')])
