sdk-python/tests/unit_tests/long_term/test_long_term.py at 19ec59db3030526c1810d8b9e2e63481dd0b7e7f · verda-cloud/sdk-python · 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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
import pytest
import responses

from verda.exceptions import APIException
from verda.long_term import LongTermPeriod, LongTermService

INVALID_REQUEST = 'invalid_request'
INVALID_REQUEST_MESSAGE = 'Bad request'

PERIOD_1 = {
    'code': '3_MONTHS',
    'name': '3 months',
    'is_enabled': True,
    'unit_name': 'month',
    'unit_value': 3.0,
    'discount_percentage': 14.0,
}

PERIOD_2 = {
    'code': '6_MONTHS',
    'name': '6 months',
    'is_enabled': True,
    'unit_name': 'month',
    'unit_value': 6.0,
    'discount_percentage': 20.0,
}

PAYLOAD = [PERIOD_1, PERIOD_2]


class TestLongTermService:
    @pytest.fixture
    def long_term_service(self, http_client):
        return LongTermService(http_client)

    @pytest.fixture
    def endpoint(self, http_client):
        return http_client._base_url + '/long-term/periods'

    def test_get_cluster_periods(self, long_term_service, endpoint):
        # arrange
        responses.add(responses.GET, endpoint + '/clusters', json=PAYLOAD, status=200)

        # act
        periods = long_term_service.get_cluster_periods()

        # assert
        assert isinstance(periods, list)
        assert len(periods) == 2
        assert isinstance(periods[0], LongTermPeriod)
        assert periods[0].code == '3_MONTHS'
        assert periods[0].name == '3 months'
        assert periods[0].is_enabled is True
        assert periods[0].unit_name == 'month'
        assert periods[0].unit_value == 3.0
        assert periods[0].discount_percentage == 14.0
        assert periods[1].code == '6_MONTHS'
        assert responses.assert_call_count(endpoint + '/clusters', 1) is True

    def test_get_cluster_periods_failed(self, long_term_service, endpoint):
        # arrange
        url = endpoint + '/clusters'
        responses.add(
            responses.GET,
            url,
            json={'code': INVALID_REQUEST, 'message': INVALID_REQUEST_MESSAGE},
            status=400,
        )

        # act + assert
        with pytest.raises(APIException) as excinfo:
            long_term_service.get_cluster_periods()

        assert excinfo.value.code == INVALID_REQUEST
        assert excinfo.value.message == INVALID_REQUEST_MESSAGE
        assert responses.assert_call_count(url, 1) is True

    def test_get_instance_periods(self, long_term_service, endpoint):
        # arrange
        responses.add(responses.GET, endpoint + '/instances', json=PAYLOAD, status=200)

        # act
        periods = long_term_service.get_instance_periods()

        # assert
        assert isinstance(periods, list)
        assert len(periods) == 2
        assert isinstance(periods[0], LongTermPeriod)
        assert periods[0].code == '3_MONTHS'
        assert periods[0].discount_percentage == 14.0
        assert periods[1].unit_value == 6.0
        assert responses.assert_call_count(endpoint + '/instances', 1) is True

    def test_get_instance_periods_failed(self, long_term_service, endpoint):
        # arrange
        url = endpoint + '/instances'
        responses.add(
            responses.GET,
            url,
            json={'code': INVALID_REQUEST, 'message': INVALID_REQUEST_MESSAGE},
            status=400,
        )

        # act + assert
        with pytest.raises(APIException) as excinfo:
            long_term_service.get_instance_periods()

        assert excinfo.value.code == INVALID_REQUEST
        assert excinfo.value.message == INVALID_REQUEST_MESSAGE
        assert responses.assert_call_count(url, 1) is True

    def test_get_cluster_periods_empty_list(self, long_term_service, endpoint):
        # arrange
        responses.add(responses.GET, endpoint + '/clusters', json=[], status=200)

        # act
        periods = long_term_service.get_cluster_periods()

        # assert
        assert isinstance(periods, list)
        assert len(periods) == 0