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
|
"""Basic tests for flume usage alerts. This module contains unittest classes for testing different functionalities of flume."""
# Standard library imports
import unittest
# Third-party imports
import requests_mock
# Local application/library-specific imports
import pyflume
from .constants import (
CONST_CLIENT_ID,
CONST_CLIENT_SECRET,
CONST_FLUME_TOKEN,
CONST_HTTP_METHOD_POST,
CONST_PASSWORD,
CONST_TOKEN_FILE,
CONST_USER_ID,
CONST_USERNAME,
)
from .utils import load_fixture
class TestFlumeUsageAlerts(unittest.TestCase):
"""Test Flume Usage Alerts Test."""
@requests_mock.Mocker()
def test_usage(self, mock):
"""
Test initialization for Flume Usage Alerts List.
Args:
mock: Requests mock.
"""
mock.register_uri(
CONST_HTTP_METHOD_POST,
pyflume.constants.URL_OAUTH_TOKEN,
text=load_fixture(CONST_TOKEN_FILE),
)
mock.register_uri(
"get",
pyflume.constants.API_USAGE_URL.format(user_id=CONST_USER_ID),
text=load_fixture("usage.json"),
)
flume_auth = pyflume.FlumeAuth(
CONST_USERNAME,
CONST_PASSWORD,
CONST_CLIENT_ID,
CONST_CLIENT_SECRET,
CONST_FLUME_TOKEN,
)
flume_alerts = pyflume.FlumeUsageAlertList(flume_auth)
alerts = flume_alerts.get_usage_alerts()
assert len(alerts) == 50 # noqa: S101, WPS432
assert alerts[0]["device_id"] == "6248148189204194987" # noqa: S101
assert alerts[0]["event_rule_name"] == "High Flow Alert" # noqa: S101
assert flume_alerts.has_next # noqa: S101
mock.register_uri(
"get",
flume_alerts.next_page,
text=load_fixture("usage_next.json"),
)
alerts_next = flume_alerts.get_next_usage_alerts()
assert len(alerts_next) == 50 # noqa: S101, WPS432
assert alerts_next[0]["device_id"] == "6248148189204194987" # noqa: S101
assert alerts_next[0]["event_rule_name"] == "High Flow Alert" # noqa: S101
assert flume_alerts.has_next is False # noqa: S101
mock.register_uri(
"get",
pyflume.constants.API_USAGE_URL.format(user_id=CONST_USER_ID),
text=load_fixture("usage_nopage.json"),
)
alerts_nopage = flume_alerts.get_usage_alerts()
assert len(alerts_nopage) == 50 # noqa: S101, WPS432
assert alerts_nopage[0]["device_id"] == "6248148189204194987" # noqa: S101
assert alerts_nopage[0]["event_rule_name"] == "High Flow Alert" # noqa: S101
assert flume_alerts.has_next is False # noqa: S101
|
