Esempio n. 1
0
def create_listener(values, session=None):
    listener = models.Listener()

    listener.update(values)

    try:
        listener.save(session=session)
    except db_exc.DBDuplicateEntry as e:
        raise exc.DBDuplicateEntryException(
            "Duplicate entry for Listener: %s" % e.columns)

    return listener
Esempio n. 2
0
def create_member(values, session=None):
    member = models.Member()

    member.update(values.copy())

    try:
        member.save(session=session)
    except db_exc.DBDuplicateEntry as e:
        raise exc.DBDuplicateEntryException(
            "Duplicate entry for MemberDefinition: %s" % e.columns)

    return member
                          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()