def get_mission(cls, object_uuid, request): from sb_solutions.neo_models import Solution from sb_questions.neo_models import Question comment = Comment.nodes.get(object_uuid=object_uuid) if comment.parent_type == "solution": return Solution.get_mission(comment.parent_id, request) elif comment.parent_type == "question": return Question.get_mission(comment.parent_id, request) else: return None
def test_get_mission(self): quest = Quest(owner_username=self.pleb.username).save() quest.owner.connect(self.pleb) mission = Mission(owner_username=self.pleb.username).save() quest.missions.connect(mission) mission.associated_with.connect(self.question) res = Question.get_mission(self.question.object_uuid) self.assertEqual(res['id'], mission.object_uuid) mission.delete() quest.delete()