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)