def test_skips_removal_if_member_does_not_exist(self):
"""Test that skips removal if member does not exist."""
non_existing_member_fqdn = "i.dont.exist"
expected_result = None
mock_run_sync = _get_mock_run_sync(return_value="""
415090d15def9053: name=toolsbeta-test-k8s-etcd-9.toolsbeta.eqiad1.wikimedia.cloud peerURLs=https://toolsbeta-test-k8s-etcd-9.toolsbeta.eqiad1.wikimedia.cloud:2380 clientURLs=https://toolsbeta-test-k8s-etcd-9.toolsbeta.eqiad1.wikimedia.cloud:2379 isLeader=true
""".encode() # noqa: E501
)
controller = EtcdctlController(remote_host=RemoteHosts(
config=mock.MagicMock(specset=Config),
hosts=NodeSet("test0.local.host")), )
with mock.patch.object(RemoteHosts, "run_sync", mock_run_sync):
gotten_result = controller.ensure_node_does_not_exist(
member_fqdn=non_existing_member_fqdn, )
_assert_called_with_single_param(param="list", mock_obj=mock_run_sync)
_assert_not_called_with_single_param(param="remove",
mock_obj=mock_run_sync)
assert gotten_result == expected_result
def test_removes_the_member_if_there_already(self):
"""Test that it removes the member if there already."""
member_fqdn = "i.already.exist"
expected_member_id = "1234556789012345"
mock_run_sync = _get_mock_run_sync(side_effect=[
f"""
415090d15def9053: name=toolsbeta-test-k8s-etcd-9.toolsbeta.eqiad1.wikimedia.cloud peerURLs=https://toolsbeta-test-k8s-etcd-9.toolsbeta.eqiad1.wikimedia.cloud:2380 clientURLs=https://toolsbeta-test-k8s-etcd-9.toolsbeta.eqiad1.wikimedia.cloud:2379 isLeader=false
{expected_member_id}: name={member_fqdn} peerURLs=http://some.url
""".encode(), # noqa: E501
"Removed :)",
])
controller = EtcdctlController(remote_host=RemoteHosts(
config=mock.MagicMock(specset=Config),
hosts=NodeSet("test0.local.host")), )
with mock.patch.object(RemoteHosts, "run_sync", mock_run_sync):
gotten_result = controller.ensure_node_does_not_exist(
member_fqdn=member_fqdn)
_assert_called_with_single_param(param="remove",
mock_obj=mock_run_sync)
assert gotten_result == expected_member_id