File: passkey_plugin.py

package info (click to toggle)
freeipa 4.13.1-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 367,240 kB
  • sloc: javascript: 562,763; python: 310,289; ansic: 49,809; sh: 7,176; makefile: 2,589; xml: 343; sed: 16
file content (126 lines) | stat: -rw-r--r-- 3,860 bytes parent folder | download | duplicates (3)
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
#
# Copyright (C) 2022  FreeIPA Contributors see COPYING for license
#

from ipapython.dn import DN
from ipatests.test_xmlrpc import objectclasses
from ipatests.test_xmlrpc.tracker.base import ConfigurationTracker
from ipatests.test_xmlrpc.xmlrpc_test import fuzzy_string
from ipatests.util import assert_deepequal


class PasskeyconfigTracker(ConfigurationTracker):
    retrieve_keys = {
        'dn',
        'iparequireuserverification',
    }

    retrieve_all_keys = retrieve_keys | {
        'cn',
        'objectclass',
        'aci',
    }

    update_keys = retrieve_keys - {'dn'}
    singlevalue_keys = {'iparequireuserverification'}

    def __init__(self, default_version=None):
        super(PasskeyconfigTracker, self).__init__(
            default_version=default_version)

        self.attrs = {
            'dn': DN(self.api.env.container_passkey, self.api.env.basedn),
            'cn': [self.api.env.container_passkey[0].value],
            'objectclass': objectclasses.passkeyconfig,
            'aci': [fuzzy_string],
            'iparequireuserverif': self.api.Command.passkeyconfig_show(
            )['result']['iparequireuserverification'],
        }

    def make_update_command(self, updates):
        return self.make_command('passkeyconfig_mod', **updates)

    def check_update(self, result, extra_keys=()):
        assert_deepequal(
            dict(
                value=None,
                summary=None,
                result=self.filter_attrs(self.update_keys | set(extra_keys)),
            ),
            result
        )

    def make_retrieve_command(self, all=False, raw=False):
        return self.make_command('passkeyconfig_show', all=all, raw=raw)

    def check_retrieve(self, result, all=False, raw=False):
        if all:
            expected = self.filter_attrs(self.retrieve_all_keys)
        else:
            expected = self.filter_attrs(self.retrieve_keys)
        assert_deepequal(
            dict(
                value=None,
                summary=None,
                result=expected,
            ),
            result
        )


class PasskeyMixin:
    def _make_add_passkey(self):
        raise NotImplementedError("_make_add_passkey method must be "
                                  "implemented in instance.")

    def _make_remove_passkey(self):
        raise NotImplementedError("_make_remove_passkey method must be "
                                  "implemented in instance.")

    def add_passkey(self, **kwargs):
        cmd = self._make_add_passkey()
        result = cmd(**kwargs)
        data = kwargs.get('ipapasskey', [])
        if not isinstance(data, list):
            data = [data]
        self.attrs.setdefault('ipapasskey', []).extend(data)

        expected = dict(
            summary=('Added passkey mappings to user '
                     '"{}"'.format(self.name)),
            value=self.name,
            result=dict(
                uid=(self.name,),
            ),
        )

        if self.attrs['ipapasskey']:
            expected['result']['ipapasskey'] = (
                self.attrs['ipapasskey'])

        assert_deepequal(expected, result)

    def remove_passkey(self, **kwargs):
        cmd = self._make_remove_passkey()

        result = cmd(**kwargs)
        data = kwargs.get('ipapasskey', [])
        if not isinstance(data, list):
            data = [data]

        for key in data:
            self.attrs['ipapasskey'].remove(key)

        expected = dict(
            summary=('Removed passkey mappings from user '
                     '"{}"'.format(self.name)),
            value=self.name,
            result=dict(
                uid=(self.name,),
            ),
        )
        if self.attrs['ipapasskey']:
            expected['result']['ipapasskey'] = (
                self.attrs['ipapasskey'])

        assert_deepequal(expected, result)