Skip to content

Commit 3d49155

Browse files
committed
upload_assets: use options instead of env vars for endpoint URL and bucket
1 parent a16f14f commit 3d49155

File tree

2 files changed

+24
-15
lines changed

2 files changed

+24
-15
lines changed

.github/workflows/benchmarks.yaml

+3-3
Original file line numberDiff line numberDiff line change
@@ -31,12 +31,12 @@ jobs:
3131
--html report/report.html --self-contained-html \
3232
--track-metrics-report=report/metrics.json \
3333
--basetemp=tmp-results \
34-
--upload-assets-runid="gh-$GITHUB_RUN_ID"
34+
--upload-assets-run-id="gh-$GITHUB_RUN_ID" \
35+
--upload-assets-endpoint-url="https://s3.waw3-1.cloudferro.com" \
36+
--upload-assets-bucket="APEx-benchmarks"
3537
env:
3638
OPENEO_AUTH_METHOD: client_credentials
3739
OPENEO_AUTH_CLIENT_CREDENTIALS_CDSEFED: ${{ secrets.OPENEO_AUTH_CLIENT_CREDENTIALS_CDSEFED }}
38-
UPLOAD_ASSETS_ENDPOINT_URL: "https://s3.waw3-1.cloudferro.com"
39-
UPLOAD_ASSETS_BUCKET: "APEx-benchmarks"
4040
UPLOAD_ASSETS_ACCESS_KEY_ID: ${{ secrets.UPLOAD_ASSETS_ACCESS_KEY_ID }}
4141
UPLOAD_ASSETS_SECRET_ACCESS_KEY: ${{ secrets.UPLOAD_ASSETS_SECRET_ACCESS_KEY }}
4242
- name: List local reports

qa/tools/apex_algorithm_qa_tools/pytest_upload_assets.py

+21-12
Original file line numberDiff line numberDiff line change
@@ -19,36 +19,45 @@
1919

2020

2121
def pytest_addoption(parser):
22-
# TODO: options for S3 bucket, credentials, ...
2322
# TODO: option to always upload (also on success).
2423
parser.addoption(
25-
"--upload-assets-runid",
24+
"--upload-assets-run-id",
2625
metavar="ID",
2726
action="store",
2827
help="The run ID to use for building the S3 key.",
2928
)
29+
parser.addoption(
30+
"--upload-assets-endpoint-url",
31+
metavar="URL",
32+
action="store",
33+
help="The S3 endpoint URL to upload to.",
34+
)
35+
parser.addoption(
36+
"--upload-assets-bucket",
37+
metavar="BUCKET",
38+
action="store",
39+
help="The S3 bucket to upload to.",
40+
)
3041

3142

3243
def pytest_configure(config: pytest.Config):
44+
run_id = config.getoption("upload_assets_run_id")
45+
endpoint_url = config.getoption("upload_assets_endpoint_url")
46+
bucket = config.getoption("upload_assets_bucket")
3347
if (
34-
# TODO only register if enough config is available for setup
48+
endpoint_url
49+
and bucket
3550
# Don't register on xdist worker nodes
36-
not hasattr(config, "workerinput")
51+
and not hasattr(config, "workerinput")
3752
):
3853
s3_client = boto3.client(
3954
service_name="s3",
4055
aws_access_key_id=os.environ.get("UPLOAD_ASSETS_ACCESS_KEY_ID"),
4156
aws_secret_access_key=os.environ.get("UPLOAD_ASSETS_SECRET_ACCESS_KEY"),
42-
# TODO Option for endpoint url
43-
endpoint_url=os.environ.get("UPLOAD_ASSETS_ENDPOINT_URL"),
57+
endpoint_url=endpoint_url,
4458
)
45-
bucket = os.environ.get("UPLOAD_ASSETS_BUCKET")
4659
config.pluginmanager.register(
47-
S3UploadPlugin(
48-
run_id=config.getoption("upload_assets_runid"),
49-
s3_client=s3_client,
50-
bucket=bucket,
51-
),
60+
S3UploadPlugin(run_id=run_id, s3_client=s3_client, bucket=bucket),
5261
name=_PLUGIN_NAME,
5362
)
5463

0 commit comments

Comments
 (0)