tracker/clients/python/tests/test_get_summaries.py at flux-sync · LearningAzure2020/tracker · 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
122
123
124
125
126
127
128
129
130
131
"""Tests for functions that get summary metrics"""
import json

from gql import Client

# Avoid confusion with gql Client
import tracker_client.summary as tc
import tracker_client.queries as queries

# pylint: disable=no-member


def test_get_all_summaries_error(mocker, error_message):
    """Test that get_all_summaries correctly handles an error response"""
    mocker.patch(
        "tracker_client.summary.execute_query",
        auto_spec=True,
        return_value=error_message,
    )

    client = Client()
    result = tc.get_all_summaries(client)

    assert result == json.dumps(
        error_message,
        indent=4,
    )


def test_get_all_summaries(mocker, all_summaries_input, all_summaries_output):
    """Test that get_all_summaries produces correct output"""
    mocker.patch(
        "tracker_client.summary.execute_query",
        auto_spec=True,
        return_value=all_summaries_input,
    )

    client = Client()
    result = tc.get_all_summaries(client)

    tc.execute_query.assert_called_once_with(client, queries.ALL_ORG_SUMMARIES)
    assert result == json.dumps(all_summaries_output, indent=4)


def test_get_summary_by_acronym_error(mocker, error_message):
    """Test that get_summary_by_acronym correctly handles an error response"""
    mocker.patch(
        "tracker_client.summary.execute_query",
        auto_spec=True,
        return_value=error_message,
    )

    client = Client()
    result = tc.get_summary_by_acronym(client, "foo")

    assert result == json.dumps(
        error_message,
        indent=4,
    )


def test_get_summary_by_acronym_key_error(mocker, all_summaries_input):
    """Test that get_summary_by_acronym correctly handles the given acronym not being in the response"""
    mocker.patch(
        "tracker_client.summary.execute_query",
        auto_spec=True,
        return_value=all_summaries_input,
    )

    client = Client()
    result = tc.get_summary_by_acronym(client, "foo")

    tc.execute_query.assert_called_once_with(client, queries.ALL_ORG_SUMMARIES)
    assert result == json.dumps(
        {
            "error": {
                "message": "No organization with the provided acronym could be found."
            }
        },
        indent=4,
    )


def test_get_summary_by_acronym(mocker, all_summaries_input, org_summary_output):
    """Test that get_summary_by_acronym produces correct output"""
    mocker.patch(
        "tracker_client.summary.execute_query",
        auto_spec=True,
        return_value=all_summaries_input,
    )

    client = Client()
    result = tc.get_summary_by_acronym(client, "def")

    tc.execute_query.assert_called_once_with(client, queries.ALL_ORG_SUMMARIES)
    assert result == json.dumps(org_summary_output, indent=4)


def test_get_summary_by_name_error(mocker, error_message):
    """Test that get_summary_by_name correctly handles an error response"""
    mocker.patch(
        "tracker_client.summary.execute_query",
        auto_spec=True,
        return_value=error_message,
    )

    client = Client()
    result = tc.get_summary_by_name(client, "foo bar")

    assert result == json.dumps(
        error_message,
        indent=4,
    )


def test_get_summary_by_name(mocker, name_summary_input, org_summary_output):
    """Test that get_summary_by_acronym produces correct output"""
    mocker.patch(
        "tracker_client.summary.execute_query",
        auto_spec=True,
        return_value=name_summary_input,
    )

    expected_params = {"orgSlug": "foo-bar"}
    client = Client()
    result = tc.get_summary_by_name(client, "foo bar")

    tc.execute_query.assert_called_once_with(
        client, queries.SUMMARY_BY_SLUG, expected_params
    )
    assert result == json.dumps(org_summary_output, indent=4)