def test_adds_the_member_if_not_there(self):
"""Test that adds the member if not there."""
new_member_fqdn = "i.already.exist"
new_member_peer_url = f"https://{new_member_fqdn}:1234"
expected_member_id = "1234556789012345"
mock_run_sync = _get_mock_run_sync(side_effect=[
b"""
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
""", # noqa: E501
b"""Added :)""",
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={new_member_fqdn} peerURLs={new_member_peer_url}
""".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_member_id = controller.ensure_node_exists(
new_member_fqdn=new_member_fqdn,
member_peer_url=new_member_peer_url,
)
_assert_called_with_single_param(param="add", mock_obj=mock_run_sync)
assert gotten_member_id == expected_member_id
def test_uses_default_member_url_if_not_passed(self):
"""Test that uses default member url if not passed."""
new_member_fqdn = "i.already.exist"
expected_peer_url = f"https://{new_member_fqdn}:2380"
mock_run_sync = _get_mock_run_sync(side_effect=[
b"",
b"""Added :)""",
f"""
415090d15def9053: name={new_member_fqdn} peerURLs={expected_peer_url}
""".encode(),
])
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):
controller.ensure_node_exists(new_member_fqdn=new_member_fqdn)
_assert_called_with_single_param(param=expected_peer_url,
mock_obj=mock_run_sync)
def test_skips_addition_if_member_already_exists(self):
"""Test that skips addition if member already exists."""
existing_member_fqdn = "i.already.exist"
existing_member_peer_url = f"https://{existing_member_fqdn}:1234"
expected_member_id = "1234556789012345"
mock_run_sync = _get_mock_run_sync(return_value=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=true
{expected_member_id}: name={existing_member_fqdn} peerURLs={existing_member_peer_url}
""".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_member_id = controller.ensure_node_exists(
new_member_fqdn=existing_member_fqdn,
member_peer_url=existing_member_peer_url,
)
_assert_called_with_single_param(param="list", mock_obj=mock_run_sync)
_assert_not_called_with_single_param(param="add",
mock_obj=mock_run_sync)
assert gotten_member_id == expected_member_id