def test_query_instance_AbnormalData2Of2(self):
# Arrange
host = "unknown"
expected_result = "No instance found for the IP address unknown"
# Act
try:
client = Main.create_ec2_client()
result = Main.query_instance(client, host)
except Exception as e:
result = str(e)
# Assert
assert expected_result == result
def test_query_instance_NormalData_ByHostname_2Of2(self):
# Arrange
host = "lin_jump_pub_01"
expected_name = "lin_jump_pub_01"
expected_volume_id = "vol-0430d0945194a0a59"
expected_volume_name = "/dev/sda1"
# Act
client = Main.create_ec2_client()
result = Main.query_instance(client, host)
# Assert
assert expected_name == result.name
assert expected_volume_id == result.root_volume_id
assert expected_volume_name == result.root_volume_name
def test_query_instance_NormalData_ByHostname_1Of2(self):
# Arrange
host = "win_jump_pub_01"
expected_name = "win_jump_pub_01"
expected_volume_id = "vol-0ab3909a5663e3a0c"
expected_volume_name = "/dev/sda1"
# Act
client = Main.create_ec2_client()
result = Main.query_instance(client, host)
# Assert
assert expected_name == result.name
assert expected_volume_id == result.root_volume_id
assert expected_volume_name == result.root_volume_name