tracker/clients/python/tests/test_domain.py at setup-cluster-network-policies · 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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
"""Tests for methods on the Domain class"""
import json

import pytest

import tracker_client.queries as queries


# pylint: disable=no-member
def test_domain_get_status(mock_domain, domain_status_input, domain_status_output):
    """Test that Domain.get_status produces correct output"""
    test_domain = mock_domain(domain_status_input)
    result = test_domain.get_status()

    test_domain.client.execute_query.assert_called_once_with(
        queries.DOMAIN_STATUS, {"domain": "foo.bar"}
    )
    assert result == json.dumps(domain_status_output, indent=4)


def test_domain_get_status_error(mock_domain, error_message):
    """Test that Domain.get_status correctly handles error response"""
    test_domain = mock_domain(error_message)
    assert test_domain.get_status() == json.dumps(error_message, indent=4)


def test_domain_get_monthly_dmarc_summary(
    mock_domain, monthly_dmarc_input, monthly_dmarc_output
):
    """Test that Domain.get_monthly_dmarc_summary produces correct output"""
    test_domain = mock_domain(monthly_dmarc_input)
    result = test_domain.get_monthly_dmarc_summary("september", 2020)

    test_domain.client.execute_query.assert_called_once_with(
        queries.DMARC_SUMMARY,
        {"domain": "foo.bar", "month": "SEPTEMBER", "year": "2020"},
    )
    assert result == json.dumps(monthly_dmarc_output, indent=4)


def test_domain_get_monthly_dmarc_summary_error(mock_domain, error_message):
    """Test that Domain.get_monthly_dmarc_summary correctly handles error response"""
    test_domain = mock_domain(error_message)
    assert test_domain.get_monthly_dmarc_summary("september", 2020) == json.dumps(
        error_message, indent=4
    )


def test_domain_get_yearly_dmarc_summaries(
    mock_domain, yearly_dmarc_input, yearly_dmarc_output
):
    """Test that Domain.get_yearly_dmarc_summaries produces correct output"""
    test_domain = mock_domain(yearly_dmarc_input)
    result = test_domain.get_yearly_dmarc_summaries()

    test_domain.client.execute_query.assert_called_once_with(
        queries.DMARC_YEARLY_SUMMARIES, {"domain": "foo.bar"}
    )
    assert result == json.dumps(yearly_dmarc_output, indent=4)


def test_domain_get_yearly_dmarc_summaries_error(mock_domain, error_message):
    """Test that Domain.get_yearly_dmarc_summaries correctly handles error response"""
    test_domain = mock_domain(error_message)
    assert test_domain.get_yearly_dmarc_summaries() == json.dumps(
        error_message, indent=4
    )


def test_domain_get_all_results(mock_domain, all_results_input, all_results_output):
    """Test that Domain.get_all_results produces correct output"""
    test_domain = mock_domain(all_results_input)
    result = test_domain.get_all_results()

    test_domain.client.execute_query.assert_called_once_with(
        queries.ALL_RESULTS, {"domain": "foo.bar", "first": 1}
    )
    assert result == json.dumps(all_results_output, indent=4)


def test_domain_get_all_results_error(mock_domain, error_message):
    """Test that Domain.get_all_results correctly handles error response"""
    test_domain = mock_domain(error_message)
    assert test_domain.get_all_results() == json.dumps(error_message, indent=4)


def test_domain_get_web_results(mock_domain, web_results_input, web_results_output):
    """Test that Domain.get_web_results produces correct output"""
    test_domain = mock_domain(web_results_input)
    result = test_domain.get_web_results()

    test_domain.client.execute_query.assert_called_once_with(
        queries.WEB_RESULTS, {"domain": "foo.bar", "first": 1}
    )
    assert result == json.dumps(web_results_output, indent=4)


def test_domain_get_web_results_error(mock_domain, error_message):
    """Test that Domain.get_web_results correctly handles error response"""
    test_domain = mock_domain(error_message)
    assert test_domain.get_web_results() == json.dumps(error_message, indent=4)


def test_domain_get_email_results(
    mock_domain, email_results_input, email_results_output
):
    """Test that Domain.get_email_results produces correct output"""
    test_domain = mock_domain(email_results_input)
    result = test_domain.get_email_results()

    test_domain.client.execute_query.assert_called_once_with(
        queries.EMAIL_RESULTS, {"domain": "foo.bar", "first": 1}
    )
    assert result == json.dumps(email_results_output, indent=4)


def test_domain_get_email_results_error(mock_domain, error_message):
    """Test that Domain.get_email_results correctly handles error response"""
    test_domain = mock_domain(error_message)
    assert test_domain.get_email_results() == json.dumps(error_message, indent=4)


def test_domain_get_owners(mock_domain, domain_get_owners_input):
    """Test that Domain.get_owners produces correct output"""
    test_domain = mock_domain(domain_get_owners_input)
    org_list = test_domain.get_owners()

    test_domain.client.execute_query.assert_called_once_with(
        queries.GET_DOMAIN_OWNERS, {"domain": "foo.bar"}
    )
    assert org_list[0].acronym == "FOO"
    assert org_list[1].name == "Fizz Bang"
    assert org_list[0].domain_count == 10
    assert org_list[1].verified


def test_domain_get_owners_error(mock_domain, error_message, capsys):
    """Test that Domain.get_owners correctly handles error response"""
    test_domain = mock_domain(error_message)

    with pytest.raises(ValueError, match=r"Unable to get owners for foo.bar"):
        test_domain.get_owners()

    captured = capsys.readouterr()
    assert "Server error:" in captured.out