Add machine_type param and fix default types

pull/22326/head
Eric Gribkoff 5 years ago
parent 55ce8a2cdf
commit afa9d0d2d7
  1. 12
      tools/run_tests/run_xds_tests.py

@ -79,10 +79,11 @@ argp.add_argument('--zone', default='us-central1-a')
argp.add_argument('--secondary_zone', argp.add_argument('--secondary_zone',
default='us-west1-b', default='us-west1-b',
help='Zone to use for secondary TD locality tests') help='Zone to use for secondary TD locality tests')
argp.add_argument('--qps', default=10, help='Client QPS') argp.add_argument('--qps', default=10, type=int, help='Client QPS')
argp.add_argument( argp.add_argument(
'--wait_for_backend_sec', '--wait_for_backend_sec',
default=600, default=600,
type=int,
help='Time limit for waiting for created backend services to report ' help='Time limit for waiting for created backend services to report '
'healthy when launching or updated GCP resources') 'healthy when launching or updated GCP resources')
argp.add_argument( argp.add_argument(
@ -124,6 +125,9 @@ argp.add_argument('--xds_server',
argp.add_argument('--source_image', argp.add_argument('--source_image',
default='projects/debian-cloud/global/images/family/debian-9', default='projects/debian-cloud/global/images/family/debian-9',
help='Source image for VMs created during the test') help='Source image for VMs created during the test')
argp.add_argument('--machine_type',
default='e2-standard-2',
help='Machine type for VMs created during the test')
argp.add_argument( argp.add_argument(
'--tolerate_gcp_errors', '--tolerate_gcp_errors',
default=False, default=False,
@ -429,14 +433,14 @@ def test_secondary_locality_gets_requests_on_primary_failure(
patch_backend_instances(gcp, backend_service, [primary_instance_group]) patch_backend_instances(gcp, backend_service, [primary_instance_group])
def create_instance_template(gcp, name, network, source_image): def create_instance_template(gcp, name, network, source_image, machine_type):
config = { config = {
'name': name, 'name': name,
'properties': { 'properties': {
'tags': { 'tags': {
'items': ['allow-health-checks'] 'items': ['allow-health-checks']
}, },
'machineType': 'e2-standard-2', 'machineType': machine_type,
'serviceAccounts': [{ 'serviceAccounts': [{
'email': 'default', 'email': 'default',
'scopes': ['https://www.googleapis.com/auth/cloud-platform',] 'scopes': ['https://www.googleapis.com/auth/cloud-platform',]
@ -941,7 +945,7 @@ try:
raise Exception( raise Exception(
'Failed to find a valid ip:port for the forwarding rule') 'Failed to find a valid ip:port for the forwarding rule')
create_instance_template(gcp, template_name, args.network, create_instance_template(gcp, template_name, args.network,
args.source_image) args.source_image, args.machine_type)
instance_group = add_instance_group(gcp, args.zone, instance_group_name, instance_group = add_instance_group(gcp, args.zone, instance_group_name,
_INSTANCE_GROUP_SIZE) _INSTANCE_GROUP_SIZE)
patch_backend_instances(gcp, backend_service, [instance_group]) patch_backend_instances(gcp, backend_service, [instance_group])

Loading…
Cancel
Save