360 lines
15 KiB
Python
360 lines
15 KiB
Python
# Copyright 2017 Red Hat, Inc.
|
|
#
|
|
# 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
|
|
|
|
try:
|
|
from BaseHTTPServer import BaseHTTPRequestHandler
|
|
except ImportError:
|
|
from http.server import BaseHTTPRequestHandler
|
|
|
|
from oslo_serialization import jsonutils
|
|
|
|
from kuryr_kubernetes.controller.managers import pool as m_pool
|
|
from kuryr_kubernetes.tests import base as test_base
|
|
|
|
|
|
class TestRequestHandler(test_base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(TestRequestHandler, self).setUp()
|
|
client_address = 'localhost'
|
|
server = '/tmp/server.lock'
|
|
req = mock.MagicMock()
|
|
with mock.patch.object(BaseHTTPRequestHandler, '__init__') as m_http:
|
|
m_http.return_value = None
|
|
self._req_handler = m_pool.RequestHandler(req, client_address,
|
|
server)
|
|
self._req_handler.rfile = mock.Mock()
|
|
self._req_handler.wfile = mock.Mock()
|
|
|
|
def _do_POST_helper(self, method, path, headers, body, expected_resp,
|
|
trigger_exception, trunk_ips, num_ports=None):
|
|
self._req_handler.headers = headers
|
|
self._req_handler.path = path
|
|
|
|
with mock.patch.object(self._req_handler.rfile, 'read') as m_read,\
|
|
mock.patch.object(self._req_handler,
|
|
'_create_subports') as m_create,\
|
|
mock.patch.object(self._req_handler,
|
|
'_delete_subports') as m_delete:
|
|
m_read.return_value = body
|
|
if trigger_exception:
|
|
m_create.side_effect = Exception
|
|
m_delete.side_effect = Exception
|
|
|
|
with mock.patch.object(self._req_handler,
|
|
'send_header') as m_send_header,\
|
|
mock.patch.object(self._req_handler,
|
|
'end_headers') as m_end_headers,\
|
|
mock.patch.object(self._req_handler.wfile,
|
|
'write') as m_write:
|
|
self._req_handler.do_POST()
|
|
|
|
if method == 'create':
|
|
if trunk_ips:
|
|
m_create.assert_called_once_with(num_ports, trunk_ips)
|
|
else:
|
|
m_create.assert_not_called()
|
|
if method == 'delete':
|
|
m_delete.assert_called_once_with(trunk_ips)
|
|
|
|
m_send_header.assert_called_once_with('Content-Length',
|
|
len(expected_resp))
|
|
m_end_headers.assert_called_once()
|
|
m_write.assert_called_once_with(expected_resp)
|
|
|
|
def _do_GET_helper(self, method, method_resp, path, headers, body,
|
|
expected_resp, trigger_exception, pool_key=None):
|
|
self._req_handler.headers = headers
|
|
self._req_handler.path = path
|
|
|
|
with mock.patch.object(self._req_handler.rfile, 'read') as m_read,\
|
|
mock.patch.object(self._req_handler,
|
|
'_list_pools') as m_list,\
|
|
mock.patch.object(self._req_handler,
|
|
'_show_pool') as m_show:
|
|
m_read.return_value = body
|
|
if trigger_exception:
|
|
m_list.side_effect = Exception
|
|
m_show.side_effect = Exception
|
|
else:
|
|
m_list.return_value = method_resp
|
|
m_show.return_value = method_resp
|
|
|
|
with mock.patch.object(self._req_handler,
|
|
'send_header') as m_send_header,\
|
|
mock.patch.object(self._req_handler,
|
|
'end_headers') as m_end_headers,\
|
|
mock.patch.object(self._req_handler.wfile,
|
|
'write') as m_write:
|
|
self._req_handler.do_GET()
|
|
|
|
if method == 'list':
|
|
m_list.assert_called_once()
|
|
if method == 'show':
|
|
if pool_key and len(pool_key) == 3:
|
|
m_show.assert_called_once_with(
|
|
(pool_key[0], pool_key[1],
|
|
tuple(sorted(pool_key[2]))))
|
|
else:
|
|
m_show.assert_not_called()
|
|
|
|
m_send_header.assert_called_once_with('Content-Length',
|
|
len(expected_resp))
|
|
m_end_headers.assert_called_once()
|
|
m_write.assert_called_once_with(expected_resp)
|
|
|
|
def test_do_POST_populate(self):
|
|
method = 'create'
|
|
path = "http://localhost/populatePool"
|
|
trunk_ips = [u"10.0.0.6"]
|
|
num_ports = 3
|
|
body = jsonutils.dumps({"trunks": trunk_ips,
|
|
"num_ports": num_ports})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = False
|
|
|
|
expected_resp = ('Ports pool at {} was populated with 3 ports.'
|
|
.format(trunk_ips)).encode()
|
|
|
|
self._do_POST_helper(method, path, headers, body, expected_resp,
|
|
trigger_exception, trunk_ips, num_ports)
|
|
|
|
def test_do_POST_populate_exception(self):
|
|
method = 'create'
|
|
path = "http://localhost/populatePool"
|
|
trunk_ips = [u"10.0.0.6"]
|
|
num_ports = 3
|
|
body = jsonutils.dumps({"trunks": trunk_ips,
|
|
"num_ports": num_ports})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = True
|
|
|
|
expected_resp = ('Error while populating pool {0} with {1} ports.'
|
|
.format(trunk_ips, num_ports)).encode()
|
|
|
|
self._do_POST_helper(method, path, headers, body, expected_resp,
|
|
trigger_exception, trunk_ips, num_ports)
|
|
|
|
def test_do_POST_populate_no_trunks(self):
|
|
method = 'create'
|
|
path = "http://localhost/populatePool"
|
|
trunk_ips = []
|
|
num_ports = 3
|
|
body = jsonutils.dumps({"trunks": trunk_ips,
|
|
"num_ports": num_ports})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = False
|
|
|
|
expected_resp = ('Trunk port IP(s) missing.'
|
|
.format(trunk_ips, num_ports)).encode()
|
|
|
|
self._do_POST_helper(method, path, headers, body, expected_resp,
|
|
trigger_exception, trunk_ips, num_ports)
|
|
|
|
def test_do_POST_free(self):
|
|
method = 'delete'
|
|
path = "http://localhost/freePool"
|
|
trunk_ips = [u"10.0.0.6"]
|
|
body = jsonutils.dumps({"trunks": trunk_ips})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = False
|
|
|
|
expected_resp = ('Ports pool belonging to {0} was freed.'
|
|
.format(trunk_ips)).encode()
|
|
|
|
self._do_POST_helper(method, path, headers, body, expected_resp,
|
|
trigger_exception, trunk_ips)
|
|
|
|
def test_do_POST_free_exception(self):
|
|
method = 'delete'
|
|
path = "http://localhost/freePool"
|
|
trunk_ips = [u"10.0.0.6"]
|
|
body = jsonutils.dumps({"trunks": trunk_ips})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = True
|
|
|
|
expected_resp = ('Error freeing ports pool: {0}.'
|
|
.format(trunk_ips)).encode()
|
|
|
|
self._do_POST_helper(method, path, headers, body, expected_resp,
|
|
trigger_exception, trunk_ips)
|
|
|
|
def test_do_POST_free_no_trunks(self):
|
|
method = 'delete'
|
|
path = "http://localhost/freePool"
|
|
trunk_ips = []
|
|
body = jsonutils.dumps({"trunks": trunk_ips})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = False
|
|
|
|
expected_resp = ('Ports pool belonging to all was freed.').encode()
|
|
|
|
self._do_POST_helper(method, path, headers, body, expected_resp,
|
|
trigger_exception, trunk_ips)
|
|
|
|
def test_do_POST_wrong_action(self):
|
|
method = 'fake'
|
|
path = "http://localhost/fakeMethod"
|
|
trunk_ips = [u"10.0.0.6"]
|
|
body = jsonutils.dumps({"trunks": trunk_ips})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = False
|
|
|
|
expected_resp = ('Method not allowed.').encode()
|
|
|
|
self._do_POST_helper(method, path, headers, body, expected_resp,
|
|
trigger_exception, trunk_ips)
|
|
|
|
def test_do_GET_list(self):
|
|
method = 'list'
|
|
method_resp = ('["10.0.0.6", "9d2b45c4efaa478481c30340b49fd4d2", '
|
|
'["00efc78c-f11c-414a-bfcd-a82e16dc07d1", '
|
|
'"fd6b13dc-7230-4cbe-9237-36b4614bc6b5"]] '
|
|
'has 5 ports')
|
|
|
|
path = "http://localhost/listPools"
|
|
body = jsonutils.dumps({})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = False
|
|
|
|
expected_resp = ('Pools:\n{0}'.format(method_resp)).encode()
|
|
|
|
self._do_GET_helper(method, method_resp, path, headers, body,
|
|
expected_resp, trigger_exception)
|
|
|
|
def test_do_GET_list_exception(self):
|
|
method = 'list'
|
|
method_resp = ('["10.0.0.6", "9d2b45c4efaa478481c30340b49fd4d2", '
|
|
'["00efc78c-f11c-414a-bfcd-a82e16dc07d1", '
|
|
'"fd6b13dc-7230-4cbe-9237-36b4614bc6b5"]] '
|
|
'has 5 ports')
|
|
|
|
path = "http://localhost/listPools"
|
|
body = jsonutils.dumps({})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = True
|
|
|
|
expected_resp = ('Error listing the pools.').encode()
|
|
|
|
self._do_GET_helper(method, method_resp, path, headers, body,
|
|
expected_resp, trigger_exception)
|
|
|
|
def test_do_GET_list_empty(self):
|
|
method = 'list'
|
|
method_resp = 'There are no pools'
|
|
|
|
path = "http://localhost/listPools"
|
|
body = jsonutils.dumps({})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = False
|
|
|
|
expected_resp = ('Pools:\n{0}'.format(method_resp)).encode()
|
|
|
|
self._do_GET_helper(method, method_resp, path, headers, body,
|
|
expected_resp, trigger_exception)
|
|
|
|
def test_do_GET_show(self):
|
|
method = 'show'
|
|
method_resp = "251f748d-2a0d-4143-bce8-2e616f7a6a4a"
|
|
path = "http://localhost/showPool"
|
|
pool_key = [u"10.0.0.6", u"9d2b45c4efaa478481c30340b49fd4d2",
|
|
[u"00efc78c-f11c-414a-bfcd-a82e16dc07d1",
|
|
u"fd6b13dc-7230-4cbe-9237-36b4614bc6b5"]]
|
|
body = jsonutils.dumps({"pool_key": pool_key})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = False
|
|
formated_key = (pool_key[0], pool_key[1], tuple(sorted(pool_key[2])))
|
|
expected_resp = ('Pool {0} ports are:\n{1}'
|
|
.format(formated_key, method_resp)).encode()
|
|
|
|
self._do_GET_helper(method, method_resp, path, headers, body,
|
|
expected_resp, trigger_exception, pool_key)
|
|
|
|
def test_do_GET_show_exception(self):
|
|
method = 'show'
|
|
method_resp = "251f748d-2a0d-4143-bce8-2e616f7a6a4a"
|
|
path = "http://localhost/showPool"
|
|
pool_key = [u"10.0.0.6", u"9d2b45c4efaa478481c30340b49fd4d2",
|
|
[u"00efc78c-f11c-414a-bfcd-a82e16dc07d1",
|
|
u"fd6b13dc-7230-4cbe-9237-36b4614bc6b5"]]
|
|
body = jsonutils.dumps({"pool_key": pool_key})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = True
|
|
formated_key = (pool_key[0], pool_key[1], tuple(sorted(pool_key[2])))
|
|
expected_resp = ('Error showing pool: {0}.'
|
|
.format(formated_key)).encode()
|
|
|
|
self._do_GET_helper(method, method_resp, path, headers, body,
|
|
expected_resp, trigger_exception, pool_key)
|
|
|
|
def test_do_GET_show_empty(self):
|
|
method = 'show'
|
|
method_resp = "Empty pool"
|
|
path = "http://localhost/showPool"
|
|
pool_key = [u"10.0.0.6", u"9d2b45c4efaa478481c30340b49fd4d2",
|
|
[u"00efc78c-f11c-414a-bfcd-a82e16dc07d1",
|
|
u"fd6b13dc-7230-4cbe-9237-36b4614bc6b5"]]
|
|
body = jsonutils.dumps({"pool_key": pool_key})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = False
|
|
formated_key = (pool_key[0], pool_key[1], tuple(sorted(pool_key[2])))
|
|
expected_resp = ('Pool {0} ports are:\n{1}'
|
|
.format(formated_key, method_resp)).encode()
|
|
|
|
self._do_GET_helper(method, method_resp, path, headers, body,
|
|
expected_resp, trigger_exception, pool_key)
|
|
|
|
def test_do_GET_show_wrong_key(self):
|
|
method = 'show'
|
|
method_resp = ""
|
|
path = "http://localhost/showPool"
|
|
pool_key = [u"10.0.0.6", u"9d2b45c4efaa478481c30340b49fd4d2"]
|
|
body = jsonutils.dumps({"pool_key": pool_key})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = False
|
|
expected_resp = ('Invalid pool key. Proper format is:\n[trunk_ip, '
|
|
'project_id, [security_groups]]\n').encode()
|
|
|
|
self._do_GET_helper(method, method_resp, path, headers, body,
|
|
expected_resp, trigger_exception, pool_key)
|
|
|
|
def test_do_GET_wrong_action(self):
|
|
method = 'fake'
|
|
method_resp = ""
|
|
path = "http://localhost/fakeMethod"
|
|
body = jsonutils.dumps({})
|
|
headers = {'Content-Type': 'application/json', 'Connection': 'close'}
|
|
headers['Content-Length'] = len(body)
|
|
trigger_exception = False
|
|
|
|
expected_resp = ('Method not allowed.').encode()
|
|
|
|
self._do_GET_helper(method, method_resp, path, headers, body,
|
|
expected_resp, trigger_exception)
|