def test_tag_ondemand_instance():
instance = mock.MagicMock()
name = "name1"
fqdn = "FQDN1"
moz_instance_type = "type1"
with mock.patch("time.sleep"):
tag_ondemand_instance(instance, name, fqdn, moz_instance_type)
expected_calls = [
mock.call.add_tag("Name", name),
mock.call.add_tag("FQDN", fqdn),
mock.call.add_tag("moz-type", moz_instance_type),
mock.call.add_tag("moz-state", "ready"),
]
instance.assert_has_calls(expected_calls, any_order=True)
def do_request_ondemand_instance(region, price, ami_id, instance_type, ssh_key,
user_data, bdm, nc, profile,
moz_instance_type, name, fqdn):
conn = get_aws_connection(region)
res = conn.run_instances(
image_id=ami_id,
key_name=ssh_key,
instance_type=instance_type,
user_data=user_data,
block_device_map=bdm,
network_interfaces=nc,
instance_profile_name=profile,
# terminate the instances on shutdown
instance_initiated_shutdown_behavior="terminate",
)
return tag_ondemand_instance(res.instances[0], name, fqdn,
moz_instance_type)
def do_request_ondemand_instance(region, price, ami_id, instance_type, ssh_key,
user_data, bdm, nc, profile,
moz_instance_type, name, fqdn):
conn = get_aws_connection(region)
res = conn.run_instances(
image_id=ami_id,
key_name=ssh_key,
instance_type=instance_type,
user_data=user_data,
block_device_map=bdm,
network_interfaces=nc,
instance_profile_name=profile,
# terminate the instances on shutdown
instance_initiated_shutdown_behavior="terminate",
)
return tag_ondemand_instance(res.instances[0], name, fqdn,
moz_instance_type)
