90 lines
3.2 KiB
Python
90 lines
3.2 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.
|
|
|
|
from unittest import mock
|
|
|
|
from gbpservice.nfp.proxy_agent.modules import proxy_agent
|
|
from neutron_lib import context as ctx
|
|
import unittest2
|
|
|
|
rpc_manager = proxy_agent.RpcHandler
|
|
|
|
|
|
class TestContext(object):
|
|
|
|
def get_context(self):
|
|
try:
|
|
context = ctx.Context(user_id='some_user',
|
|
tenant_id='some_tenant', is_advsvc=True)
|
|
except Exception:
|
|
context = ctx.Context(user_id='admin',
|
|
tenant_id='admin', is_advsvc=True, is_admin=True)
|
|
return context.to_dict()
|
|
|
|
|
|
"Common class for proxy agent test cases"
|
|
|
|
|
|
class ConfigAgentProxyTestCase(unittest2.TestCase):
|
|
|
|
def setUp(self):
|
|
self.manager = rpc_manager('conf', 'sc')
|
|
self.context = TestContext().get_context()
|
|
self.import_lib = 'gbpservice.nfp.lib.transport'
|
|
|
|
def _post(self, conf, context, body,
|
|
method_type, device_config=False,
|
|
network_function_event=False):
|
|
return (200, '')
|
|
|
|
def test_create_network_function_config(self):
|
|
_data = "data"
|
|
import_send = self.import_lib + '.send_request_to_configurator'
|
|
with mock.patch(import_send) as mock_send:
|
|
mock_send.side_effect = self._post
|
|
self.manager.create_network_function_config(self.context, _data)
|
|
|
|
def test_delete_network_function_config(self):
|
|
_data = "data"
|
|
import_send = self.import_lib + '.send_request_to_configurator'
|
|
with mock.patch(import_send) as mock_send:
|
|
mock_send.side_effect = self._post
|
|
self.manager.delete_network_function_config(self.context, _data)
|
|
|
|
def test_create_network_function_device_config(self):
|
|
_data = "data"
|
|
import_send = self.import_lib + '.send_request_to_configurator'
|
|
with mock.patch(import_send) as mock_send:
|
|
mock_send.side_effect = self._post
|
|
self.manager.create_network_function_device_config(
|
|
self.context, _data)
|
|
|
|
def test_delete_network_function_device_config(self):
|
|
_data = "data"
|
|
import_send = self.import_lib + '.send_request_to_configurator'
|
|
with mock.patch(import_send) as mock_send:
|
|
mock_send.side_effect = self._post
|
|
self.manager.delete_network_function_device_config(
|
|
self.context, _data)
|
|
|
|
def test_network_function_event(self):
|
|
_data = "data"
|
|
import_send = self.import_lib + '.send_request_to_configurator'
|
|
with mock.patch(import_send) as mock_send:
|
|
mock_send.side_effect = self._post
|
|
self.manager.network_function_event(
|
|
self.context, _data)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest2.main()
|