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 __init__(self, project_id): self.gce_project = compute_engine_projects.load_project(project_id) self.thread_pool = None self.pending_updates = []
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 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'))