def enable_project_access_from_anywhere(ctx, project_name): """Add 0.0.0.0/0 to the IP whitelist of the Atlas Project.""" project = ctx.obj.groups.byName[project_name].get().data cmd.ensure_connect_from_anywhere(client=ctx.obj, project_id=project.id)
def __init__(self, *, client, test_locator_token, configuration, xunit_output, persist_clusters, workload_startup_time): self.cases = [] self.client = client self.config = configuration self.xunit_logger = SingleTestXUnitLogger( output_directory=xunit_output) self.persist_clusters = persist_clusters self.workload_startup_time = workload_startup_time for full_path in self.find_spec_tests(test_locator_token): # Step-1: load test specification. with open(full_path, 'r') as spec_file: test_spec = JSONObject.from_dict( yaml.load(spec_file, Loader=yaml.FullLoader)) # Step-2: generate test name. test_name = get_test_name_from_spec_file(full_path) # Step-3: generate unique cluster name. cluster_name = get_cluster_name(test_name, self.config.name_salt) self.cases.append( AtlasTestCase(client=self.client, test_name=test_name, cluster_name=cluster_name, specification=test_spec, configuration=self.config)) # Set up Atlas for tests. # Step-1: ensure validity of the organization. # Note: organizations can only be created by via the web UI. org_name = self.config.organization_name LOGGER.info("Verifying organization {!r}".format(org_name)) org = get_one_organization_by_name(client=self.client, organization_name=org_name) LOGGER.info("Successfully verified organization {!r}".format(org_name)) # Step-2: check that the project exists or else create it. pro_name = self.config.project_name LOGGER.info("Verifying project {!r}".format(pro_name)) project = ensure_project(client=self.client, project_name=pro_name, organization_id=org.id) LOGGER.info("Successfully verified project {!r}".format(pro_name)) # Step-3: create a user under the project. # Note: all test operations will be run as this user. uname = self.config.database_username LOGGER.info("Verifying user {!r}".format(uname)) ensure_admin_user(client=self.client, project_id=project.id, username=uname, password=self.config.database_password) LOGGER.info("Successfully verified user {!r}".format(uname)) # Step-4: populate project IP whitelist to allow access from anywhere. LOGGER.info("Enabling access from anywhere on project " "{!r}".format(pro_name)) ensure_connect_from_anywhere(client=self.client, project_id=project.id) LOGGER.info("Successfully enabled access from anywhere on project " "{!r}".format(pro_name)) # Step-5: log test plan. LOGGER.info(self.get_printable_test_plan())