def test_happily_fetch_instances_with_instance_filter_for_instance0_or_instance_2(
mocked_fetch_instances):
# arrange
instance_0 = vmss_provider.provide_instance()
instance_0['instance_id'] = '0'
instance_0['name'] = 'chaos-pool_0'
instance_1 = vmss_provider.provide_instance()
instance_1['instance_id'] = '1'
instance_1['name'] = 'chaos-pool_1'
instance_2 = vmss_provider.provide_instance()
instance_2['instance_id'] = '2'
instance_2['name'] = 'chaos-pool_2'
instances = [instance_0, instance_1, instance_2]
mocked_fetch_instances.return_value = instances
scale_set = vmss_provider.provide_scale_set()
# fire
result = fetch_instances(scale_set,
"where instance_id=='0' or instance_id=='2'",
None)
# assert
assert len(result) == 2
assert result[0].get('name') == 'chaos-pool_0'
assert result[0].get('instance_id') == '0'
assert result[1].get('name') == 'chaos-pool_2'
assert result[1].get('instance_id') == '2'
def test_happily_fetch_empty_list_instances_with_empty_instance_filter(
mocked_fetch_instances):
mocked_fetch_instances.return_value = []
scale_set = vmss_provider.provide_scale_set()
result = fetch_instances(scale_set, None, None)
assert len(result) == 0
def test_sadly_fetch_instances_with_invalid_instance_criteria(
mocked_fetch_instances):
# arrange
instance_0 = vmss_provider.provide_instance()
instance_0['instance_id'] = '0'
instance_1 = vmss_provider.provide_instance()
instance_1['instance_id'] = '1'
instance_2 = vmss_provider.provide_instance()
instance_2['instance_id'] = '2'
instances = [instance_0, instance_1, instance_2]
mocked_fetch_instances.return_value = instances
scale_set = vmss_provider.provide_scale_set()
# fire
with pytest.raises(InterruptExecution) as x:
fetch_instances(scale_set, "invalid filter query syntax", None)
assert "invalid query" in x.value
def test_succesful_fetch_instances_without_instance_criteria(
mocked_fetch_instances):
instance = vmss_provider.provide_instance()
instances = [instance]
mocked_fetch_instances.return_value = instances
scale_set = vmss_provider.provide_scale_set()
result = fetch_instances(scale_set, None, None)
assert len(result) == 1
assert result[0].get('name') == 'chaos-pool_0'
assert result[0].get('instance_id') == '0'