File: test_administrative_units.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 (94 lines) | stat: -rw-r--r-- 3,322 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
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
from .. import util
from .base import TestAdmin


class TestAdminUnits(TestAdmin):
    # Uses underlying paging
    def test_get_administratrive_units(self):
        response = self.client_list.get_administrative_units()
        response = response[0]
        self.assertEqual(response['method'], 'GET')
        (uri, args) = response['uri'].split('?')
        self.assertEqual(uri, '/admin/v1/administrative_units')

    def test_get_administrative_units_with_limit(self):
        response = self.client_list.get_administrative_units(limit=20)
        response = response[0]
        self.assertEqual(response['method'], 'GET')
        (uri, args) = response['uri'].split('?')
        self.assertEqual(uri, '/admin/v1/administrative_units')
        self.assertEqual(
            util.params_to_dict(args),
            {
                'account_id': [self.client.account_id],
                'limit': ['20'],
                'offset': ['0'],
            })

    def test_get_adminstrative_units_with_limit_offset(self):
        response = self.client_list.get_administrative_units(limit=20, offset=2)
        response = response[0]
        self.assertEqual(response['method'], 'GET')
        (uri, args) = response['uri'].split('?')
        self.assertEqual(uri, '/admin/v1/administrative_units')
        self.assertEqual(
            util.params_to_dict(args),
            {
                'account_id': [self.client.account_id],
                'limit': ['20'],
                'offset': ['2'],
            })

    def test_get_administrative_units_with_offset(self):
        response = self.client_list.get_administrative_units(offset=9001)
        response = response[0]
        self.assertEqual(response['method'], 'GET')
        (uri, args) = response['uri'].split('?')
        self.assertEqual(uri, '/admin/v1/administrative_units')
        self.assertEqual(
            util.params_to_dict(args),
            {
                'account_id': [self.client.account_id],
                'limit': ['100'],
                'offset': ['0'],
            })

    def test_get_administrative_units_iterator(self):
        expected_path = '/admin/v1/administrative_units'
        expected_method = 'GET'

        tests = [
            {
                'admin_id': 'aaaa',
                'group_id': '1234',
                'integration_key': 'aaaaaaaaaaaaaaaaaaaa',
            },
            {
                'admin_id': 'aaaa',
                'group_id': '1234',
            },
            {
                'admin_id': 'aaaa',
            },
            {}
        ]

        for test in tests:
            response = (
                self.client_list.get_administrative_units_iterator(**test)
            )
            response = next(response)
            self.assertEqual(response['method'], expected_method)
            (uri, args) = response['uri'].split('?')
            self.assertEqual(uri, expected_path)
            expected_params = {
                key: [value] for (key, value) in test.items()
            }
            expected_params.update(
                {
                    'account_id': [self.client.account_id],
                    'limit': ['100'],
                    'offset': ['0'],
                }
            )
            self.assertEqual(util.params_to_dict(args), expected_params)