def expected_instance_template(gce_project_name, name, project_name, disk_size_gb=None, service_account=None, tls_cert=False): """Get the expected instance template for a project.""" gce_project = compute_engine_projects.load_project(gce_project_name) expected = copy.deepcopy(gce_project.get_instance_template(name)) expected['properties']['metadata']['items'].append({ 'key': 'task-tag', 'value': project_name, }) if disk_size_gb: disk = expected['properties']['disks'][0] disk['initializeParams']['diskSizeGb'] = disk_size_gb if service_account: expected['properties']['serviceAccounts'][0]['email'] = service_account if tls_cert: expected['properties']['metadata']['items'].extend([{ 'key': 'tls-cert', 'value': project_name + '_cert', }, { 'key': 'tls-key', 'value': project_name + '_key', }]) return expected
def expected_host_instance_template(gce_project_name, name): """Get the expected instance template for a project.""" gce_project = compute_engine_projects.load_project(gce_project_name) return copy.deepcopy(gce_project.get_instance_template(name))
def __init__(self, project_id): self.gce_project = compute_engine_projects.load_project(project_id) self.thread_pool = None self.pending_updates = []
def test_load_test_project(self): """Test that test config (project test-clusterfuzz) loads without any exceptions.""" self.assertIsNotNone( compute_engine_projects.load_project('test-clusterfuzz'))