def step_uc_mongo_latest_user_write_access(context): context.analytical_test_data_s3_location["path"] = "data/uc_mongo_latest/" analytical_env_helper.assume_role_for_test( context.aws_acc, context.analytical_test_e2e_role, context.aws_session_timeout_seconds, )
def step_non_sc_role_assumed(context): team_databases = [ "uc_ris_redacted", "uc_ris_unredacted", "ucs_opsmi_redacted", "ucs_opsmi_unredacted", ] team_paths = [f"data/{db}/" for db in team_databases] context.analytical_test_data_s3_location["paths"] = team_paths for db in team_databases: tag_map = {"pii": "false", "db": db, "table": "test_table"} context.analytical_test_data_s3_location["path"] = f"data/{db}/" analytical_env_helper.setup_test_file_in_s3( context.analytical_test_data_s3_location["file_name"], context.analytical_test_data_s3_location["path"], context.published_bucket, context.timeout, tag_map, ) del context.analytical_test_data_s3_location["path"] analytical_env_helper.assume_role_for_test( context.aws_acc, context.analytical_test_e2e_role, context.aws_session_timeout_seconds, )
def step_uc_mongo_latest_assumed(context): context.analytical_test_data_s3_location["path"] = "data/uc_mongo_latest/" tag_map = {"pii": "true", "db": "uc_mongo_latest", "table": "test_table"} analytical_env_helper.setup_test_file_in_s3( context.analytical_test_data_s3_location["file_name"], context.analytical_test_data_s3_location["path"], context.published_bucket, context.timeout, tag_map, ) analytical_env_helper.assume_role_for_test( context.aws_acc, context.analytical_test_e2e_role, context.aws_session_timeout_seconds, )
def step_ucs_opsmi_unredacted_upload(context): context.analytical_test_data_s3_location["path"] = "data/uc_ris_redacted/" tag_map = {"pii": "false", "db": "uc_ris_redacted", "table": "test_table"} analytical_env_helper.setup_test_file_in_s3( context.analytical_test_data_s3_location["file_name"], context.analytical_test_data_s3_location["path"], context.published_bucket, context.timeout, tag_map, ) analytical_env_helper.assume_role_for_test( context.aws_acc, context.analytical_test_e2e_role, context.aws_session_timeout_seconds, )
def step_auditlog_unred_v_upload(context): context.analytical_test_data_s3_location[ "path"] = "data/uc/auditlog_unred_v/" tag_map = {"pii": "true", "db": "uc", "table": "auditlog_unred_v"} analytical_env_helper.setup_test_file_in_s3( context.analytical_test_data_s3_location["file_name"], context.analytical_test_data_s3_location["path"], context.published_bucket, context.timeout, tag_map, ) analytical_env_helper.assume_role_for_test( context.aws_acc, context.analytical_test_e2e_role, context.aws_session_timeout_seconds, )
def step_non_sc_role_assumed(context): restricted_tables = [ "auditlog_red_v", "auditlog_unred_v", "auditlog_sec_v" ] restricted_paths = [f"data/uc/{table}/" for table in restricted_tables] context.analytical_test_data_s3_location["paths"] = restricted_paths for table in restricted_tables: tag_map = {"pii": "true", "db": "uc", "table": table} context.analytical_test_data_s3_location["path"] = f"data/uc/{table}/" analytical_env_helper.setup_test_file_in_s3( context.analytical_test_data_s3_location["file_name"], context.analytical_test_data_s3_location["path"], context.published_bucket, context.timeout, tag_map, ) del context.analytical_test_data_s3_location["path"] analytical_env_helper.assume_role_for_test( context.aws_acc, context.analytical_test_e2e_role, context.aws_session_timeout_seconds, )