def test_instance_state(aws_client):
client = MagicMock()
aws_client.return_value = client
instance_ids = ['i-987654321fedcba', 'i-392024ac3252ecb']
client.describe_instances.return_value = {
'Reservations': [{
'Instances': [
{
'InstanceId': instance_ids[0],
'State': {
'Name': 'running'
}
},
{
'InstanceId': instance_ids[1],
'State': {
'Name': 'running'
}
},
]
}]
}
results = instance_state(state='running', instance_ids=instance_ids)
client.describe_instances.assert_called_with(InstanceIds=instance_ids)
assert results
def test_instance_state(aws_client):
client = MagicMock()
aws_client.return_value = client
instance_ids = ["i-987654321fedcba", "i-392024ac3252ecb"]
client.describe_instances.return_value = {
"Reservations": [{
"Instances": [
{
"InstanceId": instance_ids[0],
"State": {
"Name": "running"
}
},
{
"InstanceId": instance_ids[1],
"State": {
"Name": "running"
}
},
]
}]
}
results = instance_state(state="running", instance_ids=instance_ids)
client.describe_instances.assert_called_with(InstanceIds=instance_ids)
assert results
def test_instance_state_filters(aws_client):
client = MagicMock()
aws_client.return_value = client
filters = [{'Name': 'instance-type', 'Values': ['t2.large']}]
client.describe_instances.return_value = {
'Reservations': [{'Instances': [
{'InstanceId': 'i-987654321fedcba',
'State': {'Name': 'running'},
'InstanceType': 't2.large'}]}]}
results = instance_state(
state='running',
filters=filters)
client.describe_instances.assert_called_with(Filters=filters)
assert results
def test_instance_state_filters(aws_client):
client = MagicMock()
aws_client.return_value = client
filters = [{"Name": "instance-type", "Values": ["t2.large"]}]
client.describe_instances.return_value = {
"Reservations": [{
"Instances": [{
"InstanceId": "i-987654321fedcba",
"State": {
"Name": "running"
},
"InstanceType": "t2.large",
}]
}]
}
results = instance_state(state="running", filters=filters)
client.describe_instances.assert_called_with(Filters=filters)
assert results
def test_instance_state_no_query():
instance_ids = ['i-987654321fedcba', 'i-392024ac3252ecb']
with pytest.raises(FailedActivity) as x:
instance_state(state='running')
assert 'Probe "instance_state" missing required parameter ' \
'"instance_ids" or "filters"' in str(x)
def test_instance_state_no_state():
instance_ids = ['i-987654321fedcba', 'i-392024ac3252ecb']
with pytest.raises(TypeError) as x:
instance_state(instance_ids=instance_ids)
assert "missing 1 required positional argument: 'state'" in str(x)
def test_instance_state_no_query(aws_client):
with pytest.raises(FailedActivity) as x:
instance_state(state="running")
assert ('Probe "instance_state" missing required parameter '
'"instance_ids" or "filters"' in str(x.value))
def test_instance_state_no_state(aws_client):
instance_ids = ["i-987654321fedcba", "i-392024ac3252ecb"]
with pytest.raises(TypeError) as x:
instance_state(instance_ids=instance_ids)
assert "missing 1 required positional argument: 'state'" in str(x.value)
