67 lines
2.5 KiB
Python
67 lines
2.5 KiB
Python
# Copyright 2013 Cloudbase Solutions Srl
|
|
#
|
|
# 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 nova import test
|
|
from nova.virt.hyperv import networkutilsv2
|
|
from nova.virt.hyperv import vmutils
|
|
|
|
|
|
class NetworkUtilsV2TestCase(test.NoDBTestCase):
|
|
"""Unit tests for the Hyper-V NetworkUtilsV2 class."""
|
|
|
|
_FAKE_VSWITCH_NAME = "fake_vswitch_name"
|
|
_FAKE_VSWITCH_PATH = "fake_vswitch_path"
|
|
|
|
def setUp(self):
|
|
self._networkutils = networkutilsv2.NetworkUtilsV2()
|
|
self._networkutils._conn = mock.MagicMock()
|
|
|
|
super(NetworkUtilsV2TestCase, self).setUp()
|
|
|
|
def test_get_external_vswitch(self):
|
|
mock_vswitch = mock.MagicMock()
|
|
mock_vswitch.path_.return_value = self._FAKE_VSWITCH_PATH
|
|
self._networkutils._conn.Msvm_VirtualEthernetSwitch.return_value = [
|
|
mock_vswitch]
|
|
|
|
switch_path = self._networkutils.get_external_vswitch(
|
|
self._FAKE_VSWITCH_NAME)
|
|
|
|
self.assertEqual(self._FAKE_VSWITCH_PATH, switch_path)
|
|
|
|
def test_get_external_vswitch_not_found(self):
|
|
mock_vswitch = mock.MagicMock()
|
|
mock_vswitch.path_.return_value = self._FAKE_VSWITCH_PATH
|
|
self._networkutils._conn.Msvm_VirtualEthernetSwitch.return_value = []
|
|
|
|
self.assertRaises(vmutils.HyperVException,
|
|
self._networkutils.get_external_vswitch,
|
|
self._FAKE_VSWITCH_NAME)
|
|
|
|
def test_get_external_vswitch_no_name(self):
|
|
mock_vswitch = mock.MagicMock()
|
|
mock_vswitch.path_.return_value = self._FAKE_VSWITCH_PATH
|
|
|
|
mock_ext_port = self._networkutils._conn.Msvm_ExternalEthernetPort()[0]
|
|
mock_lep = mock_ext_port.associators()[0]
|
|
mock_lep1 = mock_lep.associators()[0]
|
|
mock_esw = mock_lep1.associators()[0]
|
|
mock_esw.associators.return_value = [mock_vswitch]
|
|
|
|
switch_path = self._networkutils.get_external_vswitch(None)
|
|
|
|
self.assertEqual(self._FAKE_VSWITCH_PATH, switch_path)
|