def test_initialization(self): """Testing init method.""" learner_playlist = (user_domain.LearnerPlaylist( 'user_id0', ['exp_id0'], ['collect_id0'])) self.assertEqual(learner_playlist.id, 'user_id0') self.assertListEqual(learner_playlist.exploration_ids, ['exp_id0']) self.assertListEqual(learner_playlist.collection_ids, ['collect_id0'])
def test_remove_collection_id(self): """Testing remove_collection_id.""" learner_playlist = (user_domain.LearnerPlaylist( 'user_id0', ['exp_id0'], ['collect_id0'])) self.assertListEqual(learner_playlist.collection_ids, ['collect_id0']) learner_playlist.remove_collection_id('collect_id0') self.assertListEqual(learner_playlist.collection_ids, [])
def test_add_collection_id_list(self): """Testing add_collection_id.""" learner_playlist = (user_domain.LearnerPlaylist( 'user_id0', ['exp_id0'], ['collect_id0'])) self.assertListEqual(learner_playlist.collection_ids, ['collect_id0']) learner_playlist.add_collection_id_to_list('collect_id1') self.assertListEqual(learner_playlist.collection_ids, ['collect_id0', 'collect_id1'])
def test_insert_exploration_id_at_given_position(self): """Testing inserting the given exploration id at the given position.""" learner_playlist = (user_domain.LearnerPlaylist( 'user_id0', ['exp_id0'], ['collect_id0'])) self.assertListEqual(learner_playlist.exploration_ids, ['exp_id0']) learner_playlist.insert_exploration_id_at_given_position('exp_id1', 1) learner_playlist.insert_exploration_id_at_given_position('exp_id2', 1) self.assertListEqual(learner_playlist.exploration_ids, ['exp_id0', 'exp_id2', 'exp_id1'])
def get_learner_playlist_from_model(learner_playlist_model): """Returns the learner playlist domain object given the learner playlist model loaded from the datastore. Args: learner_playlist_model: LearnerPlaylistModel. The learner playlist model from the datastore. Returns: LearnerPlaylist. The learner playlist domain object corresponding to the given model. """ return user_domain.LearnerPlaylist(learner_playlist_model.id, learner_playlist_model.exploration_ids, learner_playlist_model.collection_ids)