def get_listener(name): listener = _get_listener(name) if not listener: raise exc.NotFoundException("Listener not found [name=%s]" % name) return listener
def delete_listener(name, session=None): listener = _get_listener(name) if not listener: raise exc.NotFoundException("Listener not found [name=%s]" % name) session.delete(listener)
def delete_member(name, session=None): member = _get_member(name) if not member: raise exc.NotFoundException("Member not found [member_name=%s]" % name) session.delete(member)
def get_member(name): member = _get_member(name) if not member: raise exc.NotFoundException("Member not found [member_name=%s]" % name) return member
def update_listener(name, values, session=None): listener = _get_listener(name) if not listener: raise exc.NotFoundException("Listener not found [name=%s]" % name) listener.update(values) return listener
def update_member(name, values, session=None): member = _get_member(name) if not member: raise exc.NotFoundException("Member not found [member_name=%s]" % name) member.update(values.copy()) return member
protocol_port=LISTENER['protocol_port'], algorithm=LISTENER['algorithm'], created_at=datetime.datetime(1970, 1, 1), updated_at=datetime.datetime(1970, 1, 1)) LISTENER_DB_DICT = LISTENER_DB.to_dict() FOR_UPDATED_LISTENER = copy.deepcopy(LISTENER_FOR_UPDATE) UPDATED_LISTENER = copy.deepcopy(LISTENER) UPDATED_LISTENER_DB = db.Listener(**LISTENER_DB_DICT) MOCK_LISTENER = mock.MagicMock(return_value=LISTENER_DB) MOCK_LISTENERS = mock.MagicMock(return_value=[LISTENER_DB]) MOCK_UPDATED_LISTENER = mock.MagicMock(return_value=UPDATED_LISTENER_DB) MOCK_EMPTY = mock.MagicMock(return_value=[]) MOCK_NOT_FOUND = mock.MagicMock(side_effect=exc.NotFoundException()) MOCK_DUPLICATE = mock.MagicMock(side_effect=exc.DBDuplicateEntryException()) MOCK_DELETE = mock.MagicMock(return_value=None) class TestListenerController(base.FunctionalTest): def setUp(self): super(TestListenerController, self).setUp() self.driver_origin = driver.LB_DRIVER driver.LB_DRIVER = mock.Mock() def tearDown(self): driver.LB_DRIVER = self.driver_origin super(TestListenerController, self).tearDown()