Automated change: Fix sanity tests
diff --git a/tools/run_tests/performance/loadtest_fork.py b/tools/run_tests/performance/loadtest_fork.py index 6db3c89..a69f2e0 100755 --- a/tools/run_tests/performance/loadtest_fork.py +++ b/tools/run_tests/performance/loadtest_fork.py
@@ -24,75 +24,86 @@ import argparse + def main() -> None: argp = argparse.ArgumentParser( - description='Creates a grpc_e2e_performance_gke_env.sh script providing the necessary ENVs to run OSS performance benchmark tests from some forked repos.') + description= + 'Creates a grpc_e2e_performance_gke_env.sh script providing the necessary ENVs to run OSS performance benchmark tests from some forked repos.' + ) argp.add_argument( '--grpc_test_infra_repo', default='grpc/test-infra', type=str, - help='The flag will supply the value for GRPC_TEST_INFRA_REPO environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the source code of test-infra repo to clone from, default to grpc/test-infra.', + help= + 'The flag will supply the value for GRPC_TEST_INFRA_REPO environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the source code of test-infra repo to clone from, default to grpc/test-infra.', ) argp.add_argument( '--grpc_test_infra_branch', default='master', type=str, - help='The flag will supply the value for GRPC_TEST_INFRA_BRANCH environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the branch of grpc test-infra repo to check out, default to master.', + help= + 'The flag will supply the value for GRPC_TEST_INFRA_BRANCH environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the branch of grpc test-infra repo to check out, default to master.', ) argp.add_argument( '--grpc_dotnet_repo', default='grpc/test-infra', type=str, - help='The flag will supply the value for GRPC_DOTNET_REPO environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the source code of grpc dotnet repo to clone from, default to grpc/grpc-dotnet.', + help= + 'The flag will supply the value for GRPC_DOTNET_REPO environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the source code of grpc dotnet repo to clone from, default to grpc/grpc-dotnet.', ) argp.add_argument( '--grpc_dotnet_branch', default='grpc/test-infra', type=str, - help='The flag will supply the value for GRPC_DOTNET_BRANCH environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the branch of grpc dotnet repo to check out, default to master.', + help= + 'The flag will supply the value for GRPC_DOTNET_BRANCH environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the branch of grpc dotnet repo to check out, default to master.', ) argp.add_argument( '--grpc_java_repo', default='grpc/test-infra', type=str, - help='The flag will supply the value for GRPC_JAVA_REPO environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the source code of grpc java repo to clone from, default to grpc/grpc-java.', + help= + 'The flag will supply the value for GRPC_JAVA_REPO environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the source code of grpc java repo to clone from, default to grpc/grpc-java.', ) argp.add_argument( '--grpc_java_branch', default='grpc/test-infra', type=str, - help='The flag will supply the value for GRPC_JAVA_BRANCH environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the branch of grpc java repo to check out, default to master.', + help= + 'The flag will supply the value for GRPC_JAVA_BRANCH environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the branch of grpc java repo to check out, default to master.', ) argp.add_argument( '--grpc_go_repo', default='grpc/test-infra', type=str, - help='The flag will supply the value for GRPC_GO_REPO environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the source code of grpc java repo to clone from, default to grpc/grpc-go.', + help= + 'The flag will supply the value for GRPC_GO_REPO environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the source code of grpc java repo to clone from, default to grpc/grpc-go.', ) argp.add_argument( '--grpc_go_branch', default='grpc/test-infra', type=str, - help='The flag will supply the value for GRPC_GO_BRANCH environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the branch of grpc go repo to check out, default to master.', + help= + 'The flag will supply the value for GRPC_GO_BRANCH environment variables which would be included in tools/internal_ci/linux/grpc_e2e_performance_gke_experiment.sh, specify the branch of grpc go repo to check out, default to master.', ) args = argp.parse_args() env_dict = {} - env_dict ["GRPC_TEST_INFRA_REPO"] = args.grpc_test_infra_repo - env_dict ["GRPC_TEST_INFRA_BRANCH"] = args.grpc_test_infra_branch - env_dict ["GRPC_DOTNET_REPO"] = args.grpc_dotnet_repo - env_dict ["GRPC_DOTNET_INFRA_BRANCH"] = args.grpc_dotnet_branch - env_dict ["GRPC_JAVA_REPO"] = args.grpc_java_repo - env_dict ["GRPC_JAVA_BRANCH"] = args.grpc_java_branch - env_dict ["GRPC_GO_REPO"] = args.grpc_go_repo - env_dict ["GRPC_GO_BRANCH"] = args.grpc_go_branch - + env_dict["GRPC_TEST_INFRA_REPO"] = args.grpc_test_infra_repo + env_dict["GRPC_TEST_INFRA_BRANCH"] = args.grpc_test_infra_branch + env_dict["GRPC_DOTNET_REPO"] = args.grpc_dotnet_repo + env_dict["GRPC_DOTNET_INFRA_BRANCH"] = args.grpc_dotnet_branch + env_dict["GRPC_JAVA_REPO"] = args.grpc_java_repo + env_dict["GRPC_JAVA_BRANCH"] = args.grpc_java_branch + env_dict["GRPC_GO_REPO"] = args.grpc_go_repo + env_dict["GRPC_GO_BRANCH"] = args.grpc_go_branch with open("grpc_e2e_performance_gke_env.sh", 'w') as export_file: export_file.write("#!/bin/bash\n") for k, v in env_dict.items(): export_file.write("{}={}\n".format(k, v)) + if __name__ == '__main__': main()