def setUpClass(cls): SQL_QUERY_TEST_HELPER.set_ip_addresses_in_env() if os.path.exists(TEARDOWN_LOCK_FILE_QUERY): print("Skip creating and setting up instances as they were created manually " "(helps to iterate on tests)") else: helper = CloudSqlQueryTestHelper() gcp_authenticator = GcpAuthenticator(gcp_key=GCP_CLOUDSQL_KEY) gcp_authenticator.gcp_store_authentication() try: gcp_authenticator.gcp_authenticate() helper.create_instances(instance_suffix=QUERY_SUFFIX) helper.setup_instances(instance_suffix=QUERY_SUFFIX) finally: gcp_authenticator.gcp_restore_authentication()
helper = CloudSqlQueryTestHelper() gcp_authenticator = GcpAuthenticator(GCP_CLOUDSQL_KEY) helper.log.info('Starting action: {}'.format(action)) gcp_authenticator.gcp_store_authentication() try: gcp_authenticator.gcp_authenticate() if action == 'before-tests': pass elif action == 'after-tests': pass elif action == 'create': helper.create_instances() elif action == 'delete': helper.delete_instances() elif action == 'create2': helper.create_instances(instance_suffix="2") elif action == 'delete2': helper.delete_instances(instance_suffix="2") elif action == 'setup-instances': helper.setup_instances() elif action == 'setup-instances2': helper.setup_instances(instance_suffix="2") elif action == 'delete-service-accounts-acls': helper.delete_service_account_acls() else: raise Exception("Unknown action: {}".format(action)) finally: gcp_authenticator.gcp_restore_authentication() helper.log.info('Finishing action: {}'.format(action))