summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJordan Petridis <jordan@centricular.com>2018-10-23 22:10:16 +0300
committerJordan Petridis <jordan@centricular.com>2018-11-05 11:24:12 +0200
commit61c5d7b6cf943a708bab3400c8947925f63fae46 (patch)
treea282b85b6e1605cb0a093627cd6b15e8fc3d44f8
parent6772fa76aaa6920ac76b983050efed125280216e (diff)
build_manifest.py: split the request function in two
Make it so it does not depend on env variables so it will be easier to test.
-rwxr-xr-xgitlab/build_manifest.py12
1 files changed, 9 insertions, 3 deletions
diff --git a/gitlab/build_manifest.py b/gitlab/build_manifest.py
index a94ce31..dda8266 100755
--- a/gitlab/build_manifest.py
+++ b/gitlab/build_manifest.py
@@ -33,13 +33,19 @@ MANIFEST_TEMPLATE: str = """<?xml version="1.0" encoding="UTF-8"?>
</manifest>"""
-def request(path: str) -> Dict[str, str]:
- gitlab_header: Dict[str, str] = {'JOB_TOKEN': os.environ["CI_JOB_TOKEN"]}
- base_url: str = get_hostname(os.environ['CI_PROJECT_URL'])
+def request_raw(path: str, token: str, project_url: str) -> Dict[str, str]:
+ gitlab_header: Dict[str, str] = {'JOB_TOKEN': token }
+ base_url: str = get_hostname(project_url)
return requests.get(f"https://{base_url}/api/v4/" + path, headers=gitlab_header).json()
+def request(path: str) -> Dict[str, str]:
+ token = os.environ["CI_JOB_TOKEN"]
+ project_url = os.environ['CI_PROJECT_URL']
+ return request_raw(path, token, project_url)
+
+
def get_hostname(url: str) -> str:
return urlparse(url).hostname