def test_initialization(self): """Testing init method.""" completed_activities = (user_domain.CompletedActivities( 'user_id0', ['exp_id0'], ['collect_id0'])) self.assertEqual('user_id0', completed_activities.id) self.assertListEqual(completed_activities.exploration_ids, ['exp_id0']) self.assertListEqual(completed_activities.collection_ids, ['collect_id0'])
def test_remove_exploration_id(self): """Testing remove_exploration_id.""" completed_activities = (user_domain.CompletedActivities( 'user_id0', ['exp_id0'], ['collect_id0'])) self.assertListEqual(completed_activities.exploration_ids, ['exp_id0']) completed_activities.remove_exploration_id('exp_id0') self.assertListEqual(completed_activities.exploration_ids, [])
def test_add_collection_id(self): """Testing add_collection_id.""" completed_activities = (user_domain.CompletedActivities( 'user_id0', ['exp_id0'], ['collect_id0'])) self.assertListEqual(completed_activities.collection_ids, ['collect_id0']) completed_activities.add_collection_id('collect_id1') self.assertListEqual(completed_activities.collection_ids, ['collect_id0', 'collect_id1'])
def test_remove_learnt_topic_id(self): """Testing remove_learnt_topic_id.""" completed_activities = (user_domain.CompletedActivities( 'user_id0', ['exp_id0'], ['collect_id0'], ['story_id0'], ['topic_id0'])) self.assertListEqual(completed_activities.learnt_topic_ids, ['topic_id0']) completed_activities.remove_learnt_topic_id('topic_id0') self.assertListEqual(completed_activities.learnt_topic_ids, [])
def _get_completed_activities_from_model(completed_activities_model): """Returns an activities completed domain object given a activities completed model loaded from the datastore. Args: completed_activities_model: CompletedActivitiesModel. The activities completed model loaded from the datastore. Returns: CompletedActivities. The domain object corresponding to the given model. """ return user_domain.CompletedActivities( completed_activities_model.id, completed_activities_model.exploration_ids, completed_activities_model.collection_ids)