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)
