def test_stop_vmss_with_criteria_no_matches(init_client, cmclient, fetch_instances, fetch_vmss): with pytest.raises(FailedActivity) as x: client = MagicMock() cmclient.return_value = client init_client.return_value = cmclient vmss_list = [resource_vmss] fetch_vmss.return_value = vmss_list instances_list = [ resource_vmss_instance_1, resource_vmss_instance_2, resource_vmss_instance_3 ] fetch_instances.return_value = instances_list criteria = [{'instanceId': '6'}, {"name": "chaos-vmss-instance-6"}] stop_vmss(None, None, criteria, None) cmclient.virtual_machine_scale_set_vms.power_off.assert_called_with( resource_vmss['resourceGroup'], resource_vmss['name'], resource_vmss_instance_3['instanceId']) assert "No virtual machine scale set instances found for criteria" in str( x.value)
def test_stop_vmss_having_no_vmss(fetch): with pytest.raises(FailedActivity) as x: resource_list = [] fetch.return_value = resource_list stop_vmss(None, None, None) assert "No virtual machine scale sets found" in str(x)
def test_stop_vmss(client, fetch_instances, fetch_vmss): vmss_list = [resource_vmss] fetch_vmss.return_value = vmss_list instances_list = [resource_vmss_instance] fetch_instances.return_value = instances_list client.return_value = MockComputeManagementClient() stop_vmss(None, None, None)
def test_stop_vmss_having_no_vmss_instances(fetch_instances, fetch_vmss): with pytest.raises(FailedActivity) as x: vmss_list = [resource_vmss] fetch_vmss.return_value = vmss_list instances_list = [] fetch_instances.return_value = instances_list stop_vmss(None, None, None) assert "No virtual machine scale set instances found" in str(x)
def test_stop_vmss(client, fetch_instances, fetch_vmss): scale_set = vmss_provider.provide_scale_set() scale_sets = [scale_set] instance = vmss_provider.provide_instance() instances = [instance] fetch_vmss.return_value = scale_sets fetch_instances.return_value = instances client.return_value = MockComputeManagementClient() stop_vmss(None, None, None, None)
def test_stop_vmss_with_critiria_having_no_vmss_instances( init_client, cmclient, fetch_instances, fetch_vmss): with pytest.raises(FailedActivity) as x: client = MagicMock() cmclient.return_value = client init_client.return_value = cmclient vmss_list = [resource_vmss] fetch_vmss.return_value = vmss_list instances_list = [] fetch_instances.return_value = instances_list criteria = [{'instanceId': '6'}, {"name": "chaos-vmss-instance-6"}] stop_vmss(None, None, criteria, None) assert "No virtual machine scale set instances found" in str(x.value)
def test_stop_vmss_with_criteria(init_client, cmclient, fetch_instances, fetch_vmss): client = MagicMock() cmclient.return_value = client init_client.return_value = cmclient vmss_list = [resource_vmss] fetch_vmss.return_value = vmss_list instances_list = [ resource_vmss_instance_1, resource_vmss_instance_2, resource_vmss_instance_3 ] fetch_instances.return_value = instances_list criteria = [{'instanceId': '6'}, {"name": "chaos-vmss-instance-3"}] stop_vmss(None, None, criteria, None) cmclient.virtual_machine_scale_set_vms.power_off.assert_called_with( resource_vmss['resourceGroup'], resource_vmss['name'], resource_vmss_instance_3['instanceId'])