cloudscale-cli/tests/test_image.py at e501d59f0a21cc3a4a908bcd8544e9358c204314 · cloudscale-ch/cloudscale-cli · GitHub

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
55
56
57
58
59
60
61
62
import responses
from click.testing import CliRunner
from cloudscale import CLOUDSCALE_API_URL
from cloudscale_cli.cli import cli

IMAGE_RESP = {
    "slug": "arch-18.06",
    "name": "Arch 18.06",
    "operating_system": "Arch",
    "default_username": "arch",
    "zones": [{"slug": "rma1"}, {"slug": "lpg1"}],
}


@responses.activate
def test_image_get_all():
    responses.add(
        responses.GET,
        CLOUDSCALE_API_URL + "/images",
        json=[IMAGE_RESP],
        status=200,
    )
    responses.add(
        responses.GET,
        CLOUDSCALE_API_URL + "/images",
        json={},
        status=500,
    )

    runner = CliRunner()
    result = runner.invoke(
        cli,
        [
            "-a",
            "token",
            "image",
            "list",
        ],
    )
    assert result.exit_code == 0
    result = runner.invoke(
        cli,
        [
            "-a",
            "token",
            "image",
            "list",
        ],
    )
    assert result.exit_code > 0


def test_image_missing_api_key():
    runner = CliRunner()
    result = runner.invoke(
        cli,
        [
            "image",
            "list",
        ],
    )
    assert result.exit_code == 1