def test_find_all_by_line_id(self, find_all_by_line_id): user_line = Mock(UserLine, line_id=1) find_all_by_line_id.return_value = [user_line] result = user_line_services.find_all_by_line_id(1) assert_that(result, contains(user_line))
def _delete_line_associations(channel, line_id): from xivo_dao.data_handler.line import services as line_services from xivo_dao.data_handler.line_extension import services as line_extension_services from xivo_dao.data_handler.user_line import services as user_line_services line = line_services.get(line_id) line.device_id = None line.device_slot = 1 line_services.edit(line) line_extension = line_extension_services.find_by_line_id(line_id) if line_extension: line_extension_services.dissociate(line_extension) user_lines = user_line_services.find_all_by_line_id(line_id) secondary_associations = [ul for ul in user_lines if not ul.main_user] main_associations = [ul for ul in user_lines if ul.main_user] for user_line in secondary_associations + main_associations: user_line_services.dissociate(user_line)