def __init__(self, user_id): line = line_dao.get_by_user_id(user_id) self.context = line.context self.protocol = line.protocol.upper() self.name = line.name self.number = user_line_dao.get_main_exten_by_user_id(user_id)
def test_get_main_exten_by_user_id(self): self.add_user_line_with_exten(exten='445') self.add_user_line_with_exten(exten='221') user_line = self.add_user_line_with_exten(exten=LINE_NUMBER) number = user_line_dao.get_main_exten_by_user_id(user_line.user_id) self.assertEqual(number, LINE_NUMBER)