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'))
