summaryrefslogtreecommitdiff
path: root/designate/tests/unit/test_pool.py
blob: c8e6ebe2a1165af752a3dd1ec5a3482e47e3b311 (plain)
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
# Copyright 2015 Hewlett-Packard Development Company, L.P.
#
# Author: Federico Ceratto <federico.ceratto@hpe.com>
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

from oslo_log import log as logging
import oslotest.base

from designate import objects

LOG = logging.getLogger(__name__)


def create_test_pool():
    return objects.Pool(
        name='pool1',
        description='desc',
    )


class RoObject(dict):
    def __setitem__(self, *a):
        raise NotImplementedError

    def __setattr__(self, *a):
        raise NotImplementedError

    def __getattr__(self, k):
        return self[k]


mock_conf = RoObject(**{
    'host': 'foohost',
    'pool:769ca3fc-5924-4a44-8c1f-7efbe52fbd59': RoObject(
        targets=['1588652b-50e7-46b9-b688-a9bad40a873e',
                 '2588652b-50e7-46b9-b688-a9bad40a873e'],
        nameservers=['169ca3fc-5924-4a44-8c1f-7efbe52fbd59',
                     '269ca3fc-5924-4a44-8c1f-7efbe52fbd59'],
        also_notifies=['1.0.0.0:1', '2.0.0.0:2']
    ),
    'pool_nameserver:169ca3fc-5924-4a44-8c1f-7efbe52fbd59': RoObject(
        host='pool_host_1.example.',
        port=123
    ),
    'pool_nameserver:269ca3fc-5924-4a44-8c1f-7efbe52fbd59': RoObject(
        host='pool_host_2.example.',
        port=456
    ),
    'pool_target:1588652b-50e7-46b9-b688-a9bad40a873e': RoObject(
        type='t1',
        masters=[],
        options=dict(a='1', b='2'),
    ),
    'pool_target:2588652b-50e7-46b9-b688-a9bad40a873e': RoObject(
        type='t2',
        masters=['1.1.1.1:11'],
        options={},
    ),
})


def deep_sort_lists(obj):
    """Sort lists nested in dictionaries
    """
    if isinstance(obj, dict):
        return sorted((k, deep_sort_lists(obj[k])) for k in obj)

    if isinstance(obj, list):
        return sorted(deep_sort_lists(v) for v in obj)

    return obj


class poolTest(oslotest.base.BaseTestCase):

    def test_init_from_config(self):
        pool = objects.Pool.from_config(mock_conf,
                                        '769ca3fc-5924-4a44-8c1f-7efbe52fbd59')
        expected = [('also_notifies', [[('host', '1.0.0.0'), ('port', 1)],
                                      [('host', '2.0.0.0'), ('port', 2)]]),
                    ('description', 'Pool built from configuration on foohost'),  # noqa
                    ('id', '769ca3fc-5924-4a44-8c1f-7efbe52fbd59'),
                    ('nameservers', [[('host', 'pool_host_1.example.'),
                                     ('id', '169ca3fc-5924-4a44-8c1f-7efbe52fbd59'),  # noqa
                                     ('port', 123)],
                                    [('host', 'pool_host_2.example.'),
                                     ('id', '269ca3fc-5924-4a44-8c1f-7efbe52fbd59'),  # noqa
                                     ('port', 456)]]),
                    ('targets', [[('id', '1588652b-50e7-46b9-b688-a9bad40a873e'),  # noqa
                                 ('masters', []),
                                 ('options', [[('key', 'a'), ('value', '1')],
                                              [('key', 'b'), ('value', '2')]]),
                                  ('type', 't1')],
                                [('id', '2588652b-50e7-46b9-b688-a9bad40a873e'),  # noqa
                                 ('masters', [[('host', '1.1.1.1'),
                                               ('port', 11)]]),
                                 ('options', []),
                                 ('type', 't2')]])]

        actual = deep_sort_lists(pool.to_dict())
        self.assertEqual(actual, expected)