133 lines
4.5 KiB
Python
133 lines
4.5 KiB
Python
# 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 mock
|
|
from oslo_config import cfg
|
|
from stevedore import extension
|
|
|
|
import os_vif
|
|
from os_vif import exception
|
|
from os_vif import objects
|
|
from os_vif import plugin
|
|
from os_vif.tests import base
|
|
|
|
|
|
class DemoPlugin(plugin.PluginBase):
|
|
|
|
CONFIG_OPTS = (
|
|
cfg.BoolOpt("make_it_work",
|
|
default=False,
|
|
help="Make everything work correctly by setting this"),
|
|
cfg.IntOpt("sleep_time",
|
|
default=0,
|
|
help="How long to artifically sleep")
|
|
)
|
|
|
|
def describe(self):
|
|
pass
|
|
|
|
def plug(self, vif, instance_info, config):
|
|
pass
|
|
|
|
def unplug(self, vif, instance_info, config):
|
|
pass
|
|
|
|
|
|
class DemoPluginNoConfig(plugin.PluginBase):
|
|
|
|
def describe(self):
|
|
pass
|
|
|
|
def plug(self, vif, instance_info, config):
|
|
pass
|
|
|
|
def unplug(self, vif, instance_info, config):
|
|
pass
|
|
|
|
|
|
class TestOSVIF(base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(TestOSVIF, self).setUp()
|
|
os_vif._EXT_MANAGER = None
|
|
|
|
@mock.patch('stevedore.extension.ExtensionManager')
|
|
def test_initialize(self, mock_EM):
|
|
self.assertIsNone(os_vif._EXT_MANAGER)
|
|
# Note: the duplicate call for initialize is to validate
|
|
# that the extension manager is only initialized once
|
|
os_vif.initialize()
|
|
os_vif.initialize()
|
|
mock_EM.assert_called_once_with(
|
|
invoke_on_load=False, namespace='os_vif')
|
|
self.assertIsNotNone(os_vif._EXT_MANAGER)
|
|
|
|
def test_load_plugin(self):
|
|
obj = DemoPlugin.load("demo")
|
|
self.assertTrue(hasattr(cfg.CONF, "os_vif_demo"))
|
|
self.assertTrue(hasattr(cfg.CONF.os_vif_demo, "make_it_work"))
|
|
self.assertTrue(hasattr(cfg.CONF.os_vif_demo, "sleep_time"))
|
|
self.assertEqual(cfg.CONF.os_vif_demo.make_it_work, False)
|
|
self.assertEqual(cfg.CONF.os_vif_demo.sleep_time, 0)
|
|
|
|
self.assertEqual(obj.config, cfg.CONF.os_vif_demo)
|
|
|
|
def test_load_plugin_no_config(self):
|
|
obj = DemoPluginNoConfig.load("demonocfg")
|
|
self.assertFalse(hasattr(cfg.CONF, "os_vif_demonocfg"))
|
|
|
|
self.assertIsNone(obj.config)
|
|
|
|
def test_plug_not_initialized(self):
|
|
self.assertRaises(
|
|
exception.LibraryNotInitialized,
|
|
os_vif.plug, None, None)
|
|
|
|
def test_unplug_not_initialized(self):
|
|
self.assertRaises(
|
|
exception.LibraryNotInitialized,
|
|
os_vif.plug, None, None)
|
|
|
|
@mock.patch.object(DemoPlugin, "plug")
|
|
def test_plug(self, mock_plug):
|
|
plg = extension.Extension(name="demo",
|
|
entry_point="os-vif",
|
|
plugin=DemoPlugin,
|
|
obj=None)
|
|
with mock.patch('stevedore.extension.ExtensionManager.names',
|
|
return_value=['foobar']),\
|
|
mock.patch('stevedore.extension.ExtensionManager.__getitem__',
|
|
return_value=plg):
|
|
os_vif.initialize()
|
|
info = objects.instance_info.InstanceInfo()
|
|
vif = objects.vif.VIFBridge(id='uniq',
|
|
plugin='foobar')
|
|
os_vif.plug(vif, info)
|
|
mock_plug.assert_called_once_with(vif, info)
|
|
|
|
@mock.patch.object(DemoPlugin, "unplug")
|
|
def test_unplug(self, mock_unplug):
|
|
plg = extension.Extension(name="demo",
|
|
entry_point="os-vif",
|
|
plugin=DemoPlugin,
|
|
obj=None)
|
|
with mock.patch('stevedore.extension.ExtensionManager.names',
|
|
return_value=['foobar']),\
|
|
mock.patch('stevedore.extension.ExtensionManager.__getitem__',
|
|
return_value=plg):
|
|
os_vif.initialize()
|
|
info = objects.instance_info.InstanceInfo()
|
|
vif = objects.vif.VIFBridge(id='uniq',
|
|
plugin='foobar')
|
|
os_vif.unplug(vif, info)
|
|
mock_unplug.assert_called_once_with(vif, info)
|