778ed750a0
The mock third party library was needed for mock support in py2 runtimes. Since we now only support py36 and later, we can use the standard lib unittest.mock module instead. Change-Id: I03172d63f61cba012ec4f85b20106744c9c24a1c
79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
# Copyright (C) 2013,2014 Nippon Telegraph and Telephone Corporation.
|
|
# Copyright (C) 2013,2014 YAMAMOTO Takashi <yamamoto at valinux co jp>
|
|
#
|
|
# 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 sys
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from nose.tools import eq_, raises
|
|
|
|
try:
|
|
# Python 3
|
|
from imp import reload
|
|
except ImportError:
|
|
# Python 2
|
|
pass
|
|
|
|
from os_ken.cmd.manager import main
|
|
|
|
|
|
class Test_Manager(unittest.TestCase):
|
|
"""Test osken-manager command
|
|
"""
|
|
|
|
def __init__(self, methodName):
|
|
super(Test_Manager, self).__init__(methodName)
|
|
|
|
def setUp(self):
|
|
pass
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
@raises(SystemExit)
|
|
@mock.patch('sys.argv', new=['osken-manager', '--version'])
|
|
def test_version(self):
|
|
main()
|
|
|
|
@raises(SystemExit)
|
|
@mock.patch('sys.argv', new=['osken-manager', '--help'])
|
|
def test_help(self):
|
|
main()
|
|
|
|
@staticmethod
|
|
def _reset_globals():
|
|
# hack to reset globals like SERVICE_BRICKS.
|
|
# assumption: this is the only test which actually starts OSKenApp.
|
|
import os_ken.base.app_manager
|
|
import os_ken.ofproto.ofproto_protocol
|
|
|
|
reload(os_ken.base.app_manager)
|
|
reload(os_ken.ofproto.ofproto_protocol)
|
|
|
|
@mock.patch('sys.argv', new=['osken-manager', '--verbose',
|
|
'os_ken.tests.unit.cmd.dummy_app'])
|
|
def test_no_services(self):
|
|
self._reset_globals()
|
|
main()
|
|
self._reset_globals()
|
|
|
|
@mock.patch('sys.argv', new=['osken-manager', '--verbose',
|
|
'os_ken.tests.unit.cmd.dummy_openflow_app'])
|
|
def test_openflow_app(self):
|
|
self._reset_globals()
|
|
main()
|
|
self._reset_globals()
|