charm-odl-controller/unit_tests/test_odl_controller_utils.py

121 lines
4.2 KiB
Python

# Copyright 2016 Canonical Ltd
#
# 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 mock import patch, call, ANY
from test_utils import CharmTestCase
import odl_controller_utils as utils
import odl_outputs
TO_PATCH = [
'subprocess',
'render',
'config',
'retry_on_exception',
]
class ODLControllerUtilsTests(CharmTestCase):
def setUp(self):
super(ODLControllerUtilsTests, self).setUp(utils, TO_PATCH)
self.config.side_effect = self.test_config.get
self.test_config.set('http-proxy', 'http://proxy.int:3128')
def test_mvn_proxy_ctx(self):
expect = {
'http_noproxy': [],
'http_proxy': True,
'http_proxy_host': 'proxy.int',
'http_proxy_port': 3128
}
self.assertEqual(utils.mvn_proxy_ctx('http'), expect)
def test_mvn_ctx(self):
self.test_config.set('http-proxy', 'http://proxy.int:3128')
expect = {
'http_noproxy': [],
'http_proxy': True,
'http_proxy_host': 'proxy.int',
'http_proxy_port': 3128
}
self.assertEqual(utils.mvn_ctx(), expect)
def test_mvn_ctx_unset(self):
self.test_config.set('http-proxy', '')
self.assertEqual(utils.mvn_ctx(), {})
def test_write_mvn_config(self):
self.test_config.set('http-proxy', '')
self.test_config.set('https-proxy', '')
utils.write_mvn_config()
self.render.assert_called_with(
"settings.xml", "/home/opendaylight/.m2/settings.xml", {},
"opendaylight", "opendaylight", 0400
)
def test_run_odl(self):
utils.run_odl(["feature:list"])
self.subprocess.check_output.assert_called_with(
["/opt/opendaylight-karaf/bin/client", "-r", '20', "-h",
'localhost', "-a", '8101', "-u", "karaf", 'feature:list']
)
def test_installed_features(self):
self.subprocess.check_output.return_value = \
odl_outputs.ODL_023_FEATURE_LIST
installed = utils.installed_features()
for feature in utils.PROFILES["openvswitch-odl"]["feature:install"]:
self.assertTrue(feature in installed)
self.assertFalse('odl-l2switch-hosttracker' in installed)
def test_filter_installed(self):
self.subprocess.check_output.return_value = \
odl_outputs.ODL_023_FEATURE_LIST
self.assertEqual(
utils.filter_installed(['odl-l2switch-hosttracker']),
['odl-l2switch-hosttracker']
)
self.assertEqual(utils.filter_installed(['odl-config-api']), [])
@patch.object(utils, 'run_odl')
@patch.object(utils, 'filter_installed')
def test_process_odl_cmds(self, mock_filter_installed, mock_run_odl):
test_profile = {
"feature:install": ["odl-l2switch-all"],
"log:set": {
"TRACE": ["cosc-cvpn-ovs-rest"],
},
"port": 1181
}
mock_filter_installed.return_value = ["odl-l2switch-all"]
utils.process_odl_cmds(test_profile)
mock_run_odl.assert_has_calls([
call(["feature:install", "odl-l2switch-all"]),
call(['log:set', 'TRACE', 'cosc-cvpn-ovs-rest'])
])
@patch.object(utils, 'service_running')
@patch.object(utils, 'status_set')
def test_assess_status(self, status_set, service_running):
service_running.return_value = False
utils.assess_status()
service_running.assert_called_with('odl-controller')
status_set.assert_called_with('blocked', ANY)
service_running.return_value = True
utils.assess_status()
service_running.assert_called_with('odl-controller')
status_set.assert_called_with('active', ANY)