python-gitlab/tests/functional/api/test_keys.py at main · python-gitlab/python-gitlab · GitHub

Repository navigation

History

51 lines (38 loc) · 1.46 KB

Raw

"""

GitLab API:

https://docs.gitlab.com/ce/api/keys.html

"""

import base64

import hashlib

def key_fingerprint(key: str) -> str:

key_part = key.split()[1]

decoded = base64.b64decode(key_part.encode("ascii"))

digest = hashlib.sha256(decoded).digest()

return f"SHA256:{base64.b64encode(digest).rstrip(b'=').decode('utf-8')}"

def test_keys_ssh(gl, user, SSH_KEY):

key = user.keys.create({"title": "foo@bar", "key": SSH_KEY})

# Get key by ID (admin only).

key_by_id = gl.keys.get(key.id)

assert key_by_id.title == key.title

assert key_by_id.key == key.key

fingerprint = key_fingerprint(SSH_KEY)

# Get key by fingerprint (admin only).

key_by_fingerprint = gl.keys.get(fingerprint=fingerprint)

assert key_by_fingerprint.title == key.title

assert key_by_fingerprint.key == key.key

key.delete()

def test_keys_deploy(gl, project, DEPLOY_KEY):

key = project.keys.create({"title": "foo@bar", "key": DEPLOY_KEY})

fingerprint = key_fingerprint(DEPLOY_KEY)

key_by_fingerprint = gl.keys.get(fingerprint=fingerprint)

assert key_by_fingerprint.title == key.title

assert key_by_fingerprint.key == key.key

if not any(

key_project.get("project_id") == project.id

for key_project in key_by_fingerprint.deploy_keys_projects

raise AssertionError(

f"Project {project} not found in 'deploy_keys_projects' "

f"{key_by_fingerprint.pformat()}"

)

key.delete()

python-gitlab/tests/functional/api/test_keys.py at main · python-gitlab/python-gitlab · GitHub

Repository navigation

Expand file tree

Breadcrumbs

test_keys.py

Latest commit

History

Breadcrumbs

test_keys.py

File metadata and controls