summaryrefslogtreecommitdiff
path: root/tests/unit/objects/test_keys.py
blob: 187a309e37b23850a98e6a93f7c0cc6ed2bb620a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
"""
GitLab API: https://docs.gitlab.com/ce/api/keys.html
"""
import pytest
import responses

from gitlab.v4.objects import Key

key_content = {"id": 1, "title": "title", "key": "ssh-keytype AAAAC3Nza/key comment"}


@pytest.fixture
def resp_get_key_by_id():
    with responses.RequestsMock() as rsps:
        rsps.add(
            method=responses.GET,
            url="http://localhost/api/v4/keys/1",
            json=key_content,
            content_type="application/json",
            status=200,
        )
        yield rsps


@pytest.fixture
def resp_get_key_by_fingerprint():
    with responses.RequestsMock() as rsps:
        rsps.add(
            method=responses.GET,
            url="http://localhost/api/v4/keys?fingerprint=foo",
            json=key_content,
            content_type="application/json",
            status=200,
        )
        yield rsps


def test_get_key_by_id(gl, resp_get_key_by_id):
    key = gl.keys.get(1)
    assert isinstance(key, Key)
    assert key.id == 1
    assert key.title == "title"


def test_get_key_by_fingerprint(gl, resp_get_key_by_fingerprint):
    key = gl.keys.get(fingerprint="foo")
    assert isinstance(key, Key)
    assert key.id == 1
    assert key.title == "title"


def test_get_key_missing_attrs(gl):
    with pytest.raises(AttributeError):
        gl.keys.get()