def test_initialization(self): """Testing init method.""" user_contribution_scoring = (user_domain.UserContributionScoring( 'user_id0', 'category0', 5, True)) self.assertEqual(user_contribution_scoring.user_id, 'user_id0') self.assertEqual(user_contribution_scoring.score_category, 'category0') self.assertEqual(user_contribution_scoring.score, 5) self.assertEqual(user_contribution_scoring.has_email_been_sent, True)
def get_user_contribution_scoring_from_model(userContributionScoringModel): """Returns the UserContributionScoring domain object corresponding to the UserContributionScoringModel Args: userContributionScoringModel: UserContributionScoringModel. The model instance. Returns: UserContributionScoring. The corresponding domain object. """ return user_domain.UserContributionScoring( userContributionScoringModel.user_id, userContributionScoringModel.score_category, userContributionScoringModel.score)
def get_user_scoring_from_model(user_scoring_model): """Converts the given UserContributionScoringModel to a UserContributionScoring domain object. Args: user_scoring_model: UserContributionScoringModel. UserContributionScoringModel to be converted to UserContributionScoring domain object. Returns: UserContributionScoring. The corresponding UserContributionScoring domain object. """ return user_domain.UserContributionScoring( user_scoring_model.user_id, user_scoring_model.score_category, user_scoring_model.score, user_scoring_model.onboarding_email_sent)
def _get_user_scoring(user_id, score_category): """Gets the user scoring model from storage and creates the corresponding user scoring domain object if the model exists. If the model does not exist a user scoring domain object with the given user_id and score category is created with the initial score and email values. Args: user_id: str. The id of the user. score_category: str. The category of the suggestion. Returns: UserContributionScoring. The user scoring object. """ user_scoring_model = user_models.UserContributionScoringModel.get( user_id, score_category) if user_scoring_model is not None: return get_user_scoring_from_model(user_scoring_model) return user_domain.UserContributionScoring(user_id, score_category, 0, False)
def setUp(self): super(UserContributionScoringTests, self).setUp() self.user_scoring = user_domain.UserContributionScoring( 'user_id0', 'category0', 0, False)