File: test_trust_monitor_events.py

package info (click to toggle)
python-duo-client 5.5.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 768 kB
  • sloc: python: 7,105; sh: 6; makefile: 4
file content (50 lines) | stat: -rw-r--r-- 1,502 bytes parent folder | download | duplicates (2)
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
from .. import util
import duo_client.admin
from .base import TestAdmin


MINTIME = 1603399970000
MAXTIME = 1603399973797
LIMIT = 10
NEXT_OFFSET = "99999"
EVENT_TYPE = "auth"


class TestTrustMonitorEvents(TestAdmin):

    def test_get_trust_monitor_events_iterator(self):
        """
        Test to ensure that the correct parameters are supplied when calling
        next on the generator.
        """

        generator = self.client_dtm.get_trust_monitor_events_iterator(
            MINTIME,
            MAXTIME,
            event_type=EVENT_TYPE,
        )
        events = [e for e in generator]
        self.assertEqual(events, [{"foo": "bar"},{"bar": "foo"}])

    def test_get_trust_monitor_events_by_offset(self):
        """
        Test to ensure that the correct parameters are supplied.
        """

        res = self.client_list.get_trust_monitor_events_by_offset(
            MINTIME,
            MAXTIME,
            limit=LIMIT,
            offset=NEXT_OFFSET,
            event_type=EVENT_TYPE,
        )[0]
        uri, qry_str = res["uri"].split("?")
        args = util.params_to_dict(qry_str)

        self.assertEqual(res["method"], "GET")
        self.assertEqual(uri, "/admin/v1/trust_monitor/events")
        self.assertEqual(args["mintime"], ["1603399970000"])
        self.assertEqual(args["maxtime"], ["1603399973797"])
        self.assertEqual(args["offset"], ["99999"])
        self.assertEqual(args["limit"], ["10"])
        self.assertEqual(args["type"], ["auth"])