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
|
# Copyright 2014 Rackspace, Inc.
# All Rights Reserved
#
# 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.
import inspect
from unittest import mock
import stevedore
from ironic.common import dhcp_factory
from ironic.common import exception
from ironic.dhcp import base as base_class
from ironic.dhcp import neutron
from ironic.dhcp import none
from ironic.tests import base
class TestDHCPFactory(base.TestCase):
def setUp(self):
super(TestDHCPFactory, self).setUp()
self.config(endpoint_override='test-url',
timeout=30,
group='neutron')
dhcp_factory.DHCPFactory._dhcp_provider = None
self.addCleanup(setattr, dhcp_factory.DHCPFactory,
'_dhcp_provider', None)
def test_default_dhcp(self):
# dhcp provider should default to neutron
api = dhcp_factory.DHCPFactory()
self.assertIsInstance(api.provider, neutron.NeutronDHCPApi)
def test_set_none_dhcp(self):
self.config(dhcp_provider='none',
group='dhcp')
api = dhcp_factory.DHCPFactory()
self.assertIsInstance(api.provider, none.NoneDHCPApi)
def test_set_neutron_dhcp(self):
self.config(dhcp_provider='neutron',
group='dhcp')
api = dhcp_factory.DHCPFactory()
self.assertIsInstance(api.provider, neutron.NeutronDHCPApi)
def test_only_one_dhcp(self):
self.config(dhcp_provider='none',
group='dhcp')
dhcp_factory.DHCPFactory()
with mock.patch.object(dhcp_factory.DHCPFactory, '_set_dhcp_provider',
autospec=True) as mock_set_dhcp:
# There is already a dhcp_provider, so this shouldn't call
# _set_dhcp_provider again.
dhcp_factory.DHCPFactory()
self.assertEqual(0, mock_set_dhcp.call_count)
def test_set_bad_dhcp(self):
self.config(dhcp_provider='bad_dhcp',
group='dhcp')
self.assertRaises(exception.DHCPLoadError, dhcp_factory.DHCPFactory)
@mock.patch.object(stevedore.driver, 'DriverManager', autospec=True)
def test_dhcp_some_error(self, mock_drv_mgr):
mock_drv_mgr.side_effect = Exception('No module mymod found.')
self.assertRaises(exception.DHCPLoadError, dhcp_factory.DHCPFactory)
class CompareBasetoModules(base.TestCase):
def test_drivers_match_dhcp_base(self):
signature_method = inspect.signature
def _get_public_apis(inst):
methods = {}
for (name, value) in inspect.getmembers(inst, inspect.ismethod):
if name.startswith("_"):
continue
methods[name] = value
return methods
def _compare_classes(baseclass, driverclass):
basemethods = _get_public_apis(baseclass)
implmethods = _get_public_apis(driverclass)
for name in basemethods:
baseargs = signature_method(basemethods[name])
implargs = signature_method(implmethods[name])
self.assertEqual(
baseargs,
implargs,
"%s args of %s don't match base %s" % (
name,
driverclass,
baseclass)
)
_compare_classes(base_class.BaseDHCP, none.NoneDHCPApi)
_compare_classes(base_class.BaseDHCP, neutron.NeutronDHCPApi)
|
