def test_create_user_func_key(self, send_bus_event, UserCreateFuncKeyEvent):
new_event = UserCreateFuncKeyEvent.return_value = Mock()
func_key = UserFuncKey(id=1, user_id=2)
notifier.created(func_key)
UserCreateFuncKeyEvent.assert_called_once_with(func_key.id,
func_key.user_id)
send_bus_event.assert_called_once_with(new_event, new_event.routing_key)
def test_create_bsfilter_func_key(self, send_bus_event, BSFilterCreateFuncKeyEvent):
new_event = BSFilterCreateFuncKeyEvent.return_value = Mock()
func_key = BSFilterFuncKey(id=1, filter_id=2, secretary_id=3)
notifier.created(func_key)
BSFilterCreateFuncKeyEvent.assert_called_once_with(func_key.id,
func_key.filter_id,
func_key.secretary_id)
send_bus_event.assert_called_once_with(new_event, new_event.routing_key)
def create_user_destination(user):
func_key = UserFuncKey(user_id=user.id)
created_func_key = dao.create(func_key)
notifier.created(created_func_key)
return created_func_key