def test_restart_machine_with_no_machines(fetch):
with pytest.raises(FailedActivity) as x:
resource_list = []
fetch.return_value = resource_list
restart_machines(None, None, None)
assert "No virtual machines found" in str(x.value)
def test_restart_two_machines(init, fetch):
client = MagicMock()
init.return_value = client
machines = [MACHINE_ALPHA, MACHINE_BETA]
fetch.return_value = machines
f = "where resourceGroup=='myresourcegroup' | sample 2"
restart_machines(f, CONFIG, SECRETS)
fetch.assert_called_with(f, CONFIG, SECRETS)
assert client.virtual_machines.restart.call_count == 2
def restart_node(filter: str = None,
configuration: Configuration = None,
secrets: Secrets = None):
"""
Restart a node at random from a managed Azure Kubernetes Service.
Parameters
----------
filter : str
Filter the managed AKS. If the filter is omitted all AKS in
the subscription will be selected as potential chaos candidates.
Filtering example:
'where resourceGroup=="myresourcegroup" and name="myresourcename"'
"""
logger.debug("Start restart_node: configuration='{}', filter='{}'".format(
configuration, filter))
query = node_resource_group_query(filter, configuration, secrets)
restart_machines(query, configuration, secrets)
