e94ff2681f
This patch limits the number of Amphora build requests handled by the controller worker at a given time. Also, the amphora build requests are assigned priorities based on whether it is a normal loadbalancer create, failover or spares pool loadbalancer create request. Based on the priority and the order in which the requests were made if there is an available build slot the amphora will be built. Co-Authored-By: Lubosz "diltram" Kosnik <lubosz.kosnik@intel.com> Change-Id: I967cf0668f82fb3a63e18dc7a457c58b526b7e66 Closes-Bug: #1571802
130 lines
5.7 KiB
Python
130 lines
5.7 KiB
Python
# Copyright 2016 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# 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 oslo_config import cfg
|
|
from oslo_config import fixture as oslo_fixture
|
|
from oslo_utils import uuidutils
|
|
|
|
from octavia.controller.worker import amphora_rate_limit
|
|
import octavia.tests.unit.base as base
|
|
|
|
AMP_ID = uuidutils.generate_uuid()
|
|
BUILD_PRIORITY = 40
|
|
USED_BUILD_SLOTS = 0
|
|
|
|
|
|
class TestAmphoraBuildRateLimit(base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(TestAmphoraBuildRateLimit, self).setUp()
|
|
self.conf = self.useFixture(oslo_fixture.Config(cfg.CONF))
|
|
|
|
self.rate_limit = amphora_rate_limit.AmphoraBuildRateLimit()
|
|
self.amp_build_slots_repo = mock.MagicMock()
|
|
self.amp_build_req_repo = mock.MagicMock()
|
|
self.conf.config(group='haproxy_amphora', build_rate_limit=1)
|
|
|
|
@mock.patch('octavia.db.api.get_session', mock.MagicMock())
|
|
@mock.patch('octavia.controller.worker.amphora_rate_limit'
|
|
'.AmphoraBuildRateLimit.wait_for_build_slot')
|
|
@mock.patch('octavia.db.repositories.AmphoraBuildReqRepository'
|
|
'.add_to_build_queue')
|
|
def test_add_to_build_request_queue(self,
|
|
mock_add_to_build_queue,
|
|
mock_wait_for_build_slot):
|
|
self.rate_limit.add_to_build_request_queue(AMP_ID, BUILD_PRIORITY)
|
|
|
|
mock_add_to_build_queue.assert_called_once()
|
|
mock_wait_for_build_slot.assert_called_once()
|
|
|
|
@mock.patch('octavia.db.api.get_session', mock.MagicMock())
|
|
@mock.patch('octavia.db.repositories.AmphoraBuildSlotsRepository'
|
|
'.get_used_build_slots_count',
|
|
return_value=USED_BUILD_SLOTS)
|
|
def test_has_build_slot(self, mock_get_used_build_slots_count):
|
|
result = self.rate_limit.has_build_slot()
|
|
|
|
mock_get_used_build_slots_count.assert_called_once()
|
|
self.assertTrue(result)
|
|
|
|
@mock.patch('octavia.db.api.get_session', mock.MagicMock())
|
|
@mock.patch('octavia.db.repositories.AmphoraBuildReqRepository'
|
|
'.get_highest_priority_build_req', return_value=AMP_ID)
|
|
def test_has_highest_priority(self, mock_get_highest_priority_build_req):
|
|
result = self.rate_limit.has_highest_priority(AMP_ID)
|
|
|
|
mock_get_highest_priority_build_req.assert_called_once()
|
|
self.assertTrue(result)
|
|
|
|
@mock.patch('octavia.db.api.get_session', mock.MagicMock())
|
|
@mock.patch('octavia.db.repositories.AmphoraBuildReqRepository'
|
|
'.update_req_status')
|
|
@mock.patch('octavia.db.repositories.AmphoraBuildSlotsRepository'
|
|
'.update_count')
|
|
def test_update_build_status_and_available_build_slots(self,
|
|
mock_update_count,
|
|
mock_update_status):
|
|
self.rate_limit.update_build_status_and_available_build_slots(AMP_ID)
|
|
|
|
mock_update_count.assert_called_once()
|
|
mock_update_status.assert_called_once()
|
|
|
|
@mock.patch('octavia.db.api.get_session', mock.MagicMock())
|
|
@mock.patch('octavia.db.repositories.AmphoraBuildReqRepository.delete')
|
|
@mock.patch('octavia.db.repositories.AmphoraBuildSlotsRepository'
|
|
'.update_count')
|
|
def test_remove_from_build_req_queue(self,
|
|
mock_update_count,
|
|
mock_delete):
|
|
self.rate_limit.remove_from_build_req_queue(AMP_ID)
|
|
|
|
mock_update_count.assert_called_once()
|
|
mock_delete.assert_called_once()
|
|
|
|
@mock.patch('octavia.db.api.get_session', mock.MagicMock())
|
|
@mock.patch('octavia.db.repositories.AmphoraBuildReqRepository'
|
|
'.delete_all')
|
|
@mock.patch('octavia.db.repositories.AmphoraBuildSlotsRepository'
|
|
'.update_count')
|
|
def test_remove_all_from_build_req_queue(self,
|
|
mock_update_count,
|
|
mock_delete_all):
|
|
self.rate_limit.remove_all_from_build_req_queue()
|
|
|
|
mock_update_count.assert_called_once()
|
|
mock_delete_all.assert_called_once()
|
|
|
|
@mock.patch('octavia.controller.worker.amphora_rate_limit'
|
|
'.AmphoraBuildRateLimit.has_build_slot', return_value=True)
|
|
@mock.patch('octavia.controller.worker.amphora_rate_limit'
|
|
'.AmphoraBuildRateLimit.has_highest_priority',
|
|
return_value=True)
|
|
@mock.patch('octavia.controller.worker.amphora_rate_limit'
|
|
'.AmphoraBuildRateLimit.'
|
|
'update_build_status_and_available_build_slots')
|
|
@mock.patch('octavia.controller.worker.amphora_rate_limit'
|
|
'.AmphoraBuildRateLimit.remove_all_from_build_req_queue')
|
|
@mock.patch('time.sleep')
|
|
def test_wait_for_build_slot(self,
|
|
mock_time_sleep,
|
|
mock_remove_all,
|
|
mock_update_status_and_slots_count,
|
|
mock_has_high_priority,
|
|
mock_has_build_slot):
|
|
self.rate_limit.wait_for_build_slot(AMP_ID)
|
|
|
|
self.assertTrue(mock_has_build_slot.called)
|
|
self.assertTrue(mock_has_high_priority.called)
|