class FilmHelperTestCase(TestCase): def initialize(self): self.clean_data() tags = "sciencie-fiction comedy" # set up test user self.user1 = User.objects.create_user("user1", "*****@*****.**", "secret") self.user1.save() # set up films self.film1 = Film() self.film1.title = "Battlefield Earth II" self.film1.type = Object.TYPE_FILM self.film1.permalink = "battlefirld-earth-ii" self.film1.release_year = 2010 self.film1.production_country_list = "USA" self.film1.save() self.film1.save_tags(tags, LANG="pl", saved_by=2) self.film2 = Film() self.film2.title = "Battlefield Earth III" self.film2.type = Object.TYPE_FILM self.film2.permalink = "battlefirld-earth-iii" self.film2.release_year = 2011 self.film2.production_country_list = "USA" self.film2.save() self.film2.save_tags(tags, LANG="pl", saved_by=2) self.film3 = Film() self.film3.title = "Battlefield Earth IV" self.film3.type = Object.TYPE_FILM self.film3.permalink = "battlefirld-earth-iv" self.film3.release_year = 2012 self.film3.production_country_list = "Italy" self.film3.save() self.film3.save_tags(tags, LANG="pl", saved_by=2) self.film4 = Film() self.film4.title = "Battlefield Earth V" self.film4.type = Object.TYPE_FILM self.film4.permalink = "battlefirld-earth-v" self.film4.release_year = 2013 self.film4.production_country_list = "UK" self.film4.save() self.film4.save_tags(tags, LANG="pl", saved_by=2) # set up filmrankings self.filmranking1 = FilmRanking() self.filmranking1.film = self.film1 self.filmranking1.type = Rating.TYPE_FILM self.filmranking1.average_score = Decimal('8.0') self.filmranking1.number_of_votes = 80 self.filmranking1.save() self.filmranking2 = FilmRanking() self.filmranking2.film = self.film2 self.filmranking2.type = Rating.TYPE_FILM self.filmranking2.average_score = Decimal('7.0') self.filmranking2.number_of_votes = 70 self.filmranking2.save() self.filmranking3 = FilmRanking() self.filmranking3.film = self.film3 self.filmranking3.type = Rating.TYPE_FILM self.filmranking3.average_score = Decimal('6.0') self.filmranking3.number_of_votes = 60 self.filmranking3.save() self.filmranking4 = FilmRanking() self.filmranking4.film = self.film4 self.filmranking4.type = Rating.TYPE_FILM self.filmranking4.average_score = Decimal('5.0') self.filmranking4.number_of_votes = 2 self.filmranking4.save() # save compared objects in database self.compared_film1 = FilmComparator() self.compared_film1.main_film = self.film1 self.compared_film1.compared_film = self.film1 self.compared_film1.score = 10 self.compared_film1.save() self.compared_film2 = FilmComparator() self.compared_film2.main_film = self.film1 self.compared_film2.compared_film = self.film2 self.compared_film2.score = 9 self.compared_film2.save() self.compared_film3 = FilmComparator() self.compared_film3.main_film = self.film1 self.compared_film3.compared_film = self.film3 self.compared_film3.score = 8 self.compared_film3.save() self.compared_film4 = FilmComparator() self.compared_film4.main_film = self.film1 self.compared_film4.compared_film = self.film4 self.compared_film4.score = 7 self.compared_film4.save() # set up test user ratings self.user1rating1 = Rating() self.user1rating1.type = Rating.TYPE_FILM self.user1rating1.user = self.user1 self.user1rating1.parent = self.film1 self.user1rating1.rating = 10 self.user1rating1.save() self.user1rating2 = Rating() self.user1rating2.type = Rating.TYPE_FILM self.user1rating2.user = self.user1 self.user1rating2.parent = self.film2 self.user1rating2.rating = 9 self.user1rating2.save() self.user1rating3 = Rating() self.user1rating3.type = Rating.TYPE_FILM self.user1rating3.user = self.user1 self.user1rating3.parent = self.film3 self.user1rating3.rating = 8 self.user1rating3.save() self.user1rating4 = Rating() self.user1rating4.type = Rating.TYPE_FILM self.user1rating4.user = self.user1 self.user1rating4.parent = self.film4 self.user1rating4.rating = 7 self.user1rating4.save() def clean_data(self): Film.objects.all().delete() FilmComparator.objects.all().delete() FilmRanking.objects.all().delete() def test_get_users_best_films(self): """ Test get_users_best_films method """ self.initialize() helper = FilmHelper() best_films = helper.get_users_best_films(self.user1, 3) self.assertEquals(len(best_films), 3) self.assertEquals(self.film4 in best_films, False) self.assertEquals(self.film1 in best_films, True) def test_get_related_localized_objects(self): """ Test get_related_localized_objects() method """ self.initialize() helper = FilmHelper() related = helper.get_related_localized_objects(self.film1, 3) self.assertEquals(len(related), 3) def test_get_popular_films(self): """ Test get_popular_films method """ self.initialize() helper = FilmHelper() films_english = set(helper.get_popular_films(exclude_nonEnglish=True)) self.assertEquals(self.film1 in films_english, True) self.assertEquals(self.film3 in films_english, False) self.assertEquals(self.film4 in films_english, False) films_english = set(helper.get_popular_films()) self.assertEquals(self.film1 in films_english, True) self.assertEquals(self.film3 in films_english, True)
class FilmHelperTestCase(TestCase): def initialize(self): self.clean_data() tags = "sciencie-fiction comedy" # set up test user self.user1 = User.objects.create_user("user1", "*****@*****.**", "secret") self.user1.save() # set up films self.film1 = Film() self.film1.title = "Battlefield Earth II" self.film1.type = Object.TYPE_FILM self.film1.permalink = "battlefirld-earth-ii" self.film1.release_year = 2010 self.film1.production_country_list = "USA" self.film1.save() self.film1.save_tags(tags, LANG="pl", saved_by=2) self.film2 = Film() self.film2.title = "Battlefield Earth III" self.film2.type = Object.TYPE_FILM self.film2.permalink = "battlefirld-earth-iii" self.film2.release_year = 2011 self.film2.production_country_list = "USA" self.film2.save() self.film2.save_tags(tags, LANG="pl", saved_by=2) self.film3 = Film() self.film3.title = "Battlefield Earth IV" self.film3.type = Object.TYPE_FILM self.film3.permalink = "battlefirld-earth-iv" self.film3.release_year = 2012 self.film3.production_country_list = "Italy" self.film3.save() self.film3.save_tags(tags, LANG="pl", saved_by=2) self.film4 = Film() self.film4.title = "Battlefield Earth V" self.film4.type = Object.TYPE_FILM self.film4.permalink = "battlefirld-earth-v" self.film4.release_year = 2013 self.film4.production_country_list = "UK" self.film4.save() self.film4.save_tags(tags, LANG="pl", saved_by=2) # set up filmrankings self.filmranking1 = FilmRanking() self.filmranking1.film = self.film1 self.filmranking1.type = Rating.TYPE_FILM self.filmranking1.average_score = Decimal('8.0') self.filmranking1.number_of_votes = 80 self.filmranking1.save() self.filmranking2 = FilmRanking() self.filmranking2.film = self.film2 self.filmranking2.type = Rating.TYPE_FILM self.filmranking2.average_score = Decimal('7.0') self.filmranking2.number_of_votes = 70 self.filmranking2.save() self.filmranking3 = FilmRanking() self.filmranking3.film = self.film3 self.filmranking3.type = Rating.TYPE_FILM self.filmranking3.average_score = Decimal('6.0') self.filmranking3.number_of_votes = 60 self.filmranking3.save() self.filmranking4 = FilmRanking() self.filmranking4.film = self.film4 self.filmranking4.type = Rating.TYPE_FILM self.filmranking4.average_score = Decimal('5.0') self.filmranking4.number_of_votes = 2 self.filmranking4.save() # save compared objects in database self.compared_film1 = FilmComparator() self.compared_film1.main_film = self.film1 self.compared_film1.compared_film = self.film1 self.compared_film1.score = 10 self.compared_film1.save() self.compared_film2 = FilmComparator() self.compared_film2.main_film = self.film1 self.compared_film2.compared_film = self.film2 self.compared_film2.score = 9 self.compared_film2.save() self.compared_film3 = FilmComparator() self.compared_film3.main_film = self.film1 self.compared_film3.compared_film = self.film3 self.compared_film3.score = 8 self.compared_film3.save() self.compared_film4 = FilmComparator() self.compared_film4.main_film = self.film1 self.compared_film4.compared_film = self.film4 self.compared_film4.score = 7 self.compared_film4.save() # set up test user ratings self.user1rating1 = Rating() self.user1rating1.type = Rating.TYPE_FILM self.user1rating1.user = self.user1 self.user1rating1.film = self.film1 self.user1rating1.parent = self.film1 self.user1rating1.rating = 10 self.user1rating1.save() self.user1rating2 = Rating() self.user1rating2.type = Rating.TYPE_FILM self.user1rating2.user = self.user1 self.user1rating2.film = self.film2 self.user1rating2.parent = self.film2 self.user1rating2.rating = 9 self.user1rating2.save() self.user1rating3 = Rating() self.user1rating3.type = Rating.TYPE_FILM self.user1rating3.user = self.user1 self.user1rating3.film = self.film3 self.user1rating3.parent = self.film3 self.user1rating3.rating = 8 self.user1rating3.save() self.user1rating4 = Rating() self.user1rating4.type = Rating.TYPE_FILM self.user1rating4.user = self.user1 self.user1rating4.film = self.film4 self.user1rating4.parent = self.film4 self.user1rating4.rating = 7 self.user1rating4.save() def clean_data(self): Film.objects.all().delete() FilmComparator.objects.all().delete() FilmRanking.objects.all().delete() def test_get_users_best_films(self): """ Test get_users_best_films method """ self.initialize() helper = FilmHelper() best_films = helper.get_users_best_films(self.user1, 3) self.assertEquals(len(best_films), 3) self.assertEquals(self.film4 in best_films, False) self.assertEquals(self.film1 in best_films, True) def test_get_related_localized_objects(self): """ Test get_related_localized_objects() method """ self.initialize() helper = FilmHelper() related = helper.get_related_localized_objects(self.film1, 3) self.assertEquals(len(related), 3) def test_get_popular_films(self): """ Test get_popular_films method """ self.initialize() helper = FilmHelper() films_english = set(helper.get_popular_films(exclude_nonEnglish=True)) self.assertEquals(self.film1 in films_english, True) self.assertEquals(self.film3 in films_english, False) self.assertEquals(self.film4 in films_english, False) films_english = set(helper.get_popular_films()) self.assertEquals(self.film1 in films_english, True) self.assertEquals(self.film3 in films_english, True)
class ShortReviewTestCase(TestCase): def initialize(self): self.clean_data() # set up users self.u1= User.objects.create_user('michuk', '*****@*****.**', 'secret') self.u1.save() # set up film self.film = Film() self.film.title = "Battlefield Earth II" self.film.type = Object.TYPE_FILM self.film.permalink = "battlefirld-earth-ii" self.film.release_year = 2010 self.film.save() # set up rating self.rating = Rating() self.rating.parent = self.film self.rating.user = self.u1 self.rating.film = self.film self.rating.type = Rating.TYPE_FILM self.rating.rating = 8 self.rating.normalized = 8 self.rating.save() def clean_data(self): User.objects.all().delete() Post.objects.all().delete() ShortReview.objects.all().delete() def test_saving_single_shortreview(self): """ Test saving single shortreview """ self.initialize() # set up Shortreview obj = Object.objects.get(id=self.film.id) shortreview = ShortReview() shortreview.user = self.u1 shortreview.review_text = "sialala bumcyk cyk" shortreview.status = Object.PUBLIC_STATUS shortreview.type = Object.TYPE_SHORT_REVIEW shortreview.kind = ShortReview.REVIEW shortreview.object = obj shortreview.rating = self.rating shortreview.save() # testing if shortreview was saved sr = ShortReview.objects.get(object=obj) self.failUnlessEqual(sr.user, shortreview.user) self.failUnlessEqual(sr.review_text, "sialala bumcyk cyk") self.failUnlessEqual(sr.status, Object.PUBLIC_STATUS) self.failUnlessEqual(sr.type, Object.TYPE_SHORT_REVIEW) self.failUnlessEqual(sr.kind, ShortReview.REVIEW) self.failUnlessEqual(sr.rating.rating, 8) self.failUnlessEqual(sr.object, obj) def test_delete_single_shortreview(self): """ Test deleting single shortreview """ self.initialize() # set up Shortreview obj = Object.objects.get(id=self.film.id) shortreview = ShortReview() shortreview.user = self.u1 shortreview.review_text = "sialala bumcyk cyk" shortreview.status = Object.PUBLIC_STATUS shortreview.type = Object.TYPE_SHORT_REVIEW shortreview.kind = ShortReview.REVIEW shortreview.object = obj shortreview.save() # there should be one short review self.failUnlessEqual(ShortReview.objects.all().count(), 1) sr = ShortReview.objects.get(object=obj) # deleting shortreview sr.delete() # there should be zero short reviews self.failUnlessEqual(ShortReview.objects.all().count(), 0) def test_edit_single_shortreview(self): """ Test editing single shortreview """ self.initialize() # set up Shortreview obj = Object.objects.get(id=self.film.id) shortreview = ShortReview() shortreview.user = self.u1 shortreview.review_text = "sialala bumcyk cyk" shortreview.status = Object.PUBLIC_STATUS shortreview.type = Object.TYPE_SHORT_REVIEW shortreview.kind = ShortReview.REVIEW shortreview.object = obj shortreview.save() shortreview.rating.rating = 10 shortreview.rating.save() # testing if shortreview was saved sr = ShortReview.objects.get(object=obj) self.failUnlessEqual(sr.review_text, "sialala bumcyk cyk") sr.review_text = "Lorem ipsum" sr.save() # testing if short review was edited successfully sr = ShortReview.objects.get(object=obj) self.failUnlessEqual(sr.review_text, "Lorem ipsum") self.failUnlessEqual(sr.rating.rating, 10) def test_saving_double_shortreviews(self): """ Test saving two shortreviews about the same movie by the same user """ self.initialize() # set up Shortreviews obj = Object.objects.get(id=self.film.id) shortreview1 = ShortReview() shortreview1.user = self.u1 shortreview1.review_text = "sialala bumcyk cyk" shortreview1.status = Object.PUBLIC_STATUS shortreview1.type = Object.TYPE_SHORT_REVIEW shortreview1.kind = ShortReview.REVIEW shortreview1.object = obj shortreview1.rating = self.rating shortreview1.save() obj = Object.objects.get(id=self.film.id) shortreview2 = ShortReview() shortreview2.user = self.u1 shortreview2.review_text = "Lorem ipsum lorem ipsum" shortreview2.status = Object.PUBLIC_STATUS shortreview2.type = Object.TYPE_SHORT_REVIEW shortreview2.kind = ShortReview.REVIEW shortreview2.object = obj shortreview2.rating = self.rating shortreview2.save() # testing if shortreview was saved sr = ShortReview.objects.filter(object=obj).order_by("created_at") self.failUnlessEqual(sr[0].user, shortreview1.user) self.failUnlessEqual(sr[0].review_text, "sialala bumcyk cyk") self.failUnlessEqual(sr[0].status, Object.PUBLIC_STATUS) self.failUnlessEqual(sr[0].type, Object.TYPE_SHORT_REVIEW) self.failUnlessEqual(sr[0].kind, ShortReview.REVIEW) self.failUnlessEqual(sr[0].rating.rating, 8) self.failUnlessEqual(sr[0].object, obj) self.failUnlessEqual(sr[1].user, shortreview1.user) self.failUnlessEqual(sr[1].review_text, "Lorem ipsum lorem ipsum") self.failUnlessEqual(sr[1].status, Object.PUBLIC_STATUS) self.failUnlessEqual(sr[1].type, Object.TYPE_SHORT_REVIEW) self.failUnlessEqual(sr[1].kind, ShortReview.REVIEW) self.failUnlessEqual(sr[1].rating.rating, 8) self.failUnlessEqual(sr[1].object, obj) def test_saving_wall_post(self): """ Test saving wall post """ self.initialize() wallpost = ShortReview() wallpost.user = self.u1 wallpost.review_text = "sialala bumcyk cyk" wallpost.status = Object.PUBLIC_STATUS wallpost.kind = ShortReview.WALLPOST wallpost.type = Object.TYPE_SHORT_REVIEW wallpost.save() permalink = 'http://%s.%s/%s/%s/' % (self.u1.username, settings.DOMAIN, urls['WALL'], wallpost.id) wp = ShortReview.objects.all()[0] self.failUnlessEqual(wp.get_absolute_url(), permalink) self.failUnlessEqual(wp.user, wallpost.user) self.failUnlessEqual(wp.review_text, "sialala bumcyk cyk") self.failUnlessEqual(wp.status, Object.PUBLIC_STATUS) self.failUnlessEqual(wp.type, Object.TYPE_SHORT_REVIEW)
class UserActivityTestCase(TestCase): def setUp(self): Film.objects.filter(imdb_code__lt=1000).delete() def initialize(self): self.clean_data() # set up users self.u1= User.objects.create_user('michuk', '*****@*****.**', 'secret') self.u1.save() self.u2 = User(username='******', email='*****@*****.**') self.u2.save() self.u3 = User(username='******', email='*****@*****.**') self.u3.save() self.u4 = User(username='******', email='*****@*****.**') self.u4.save() # set up film self.film = Film() self.film.title = "Battlefield Earth II" self.film.type = Object.TYPE_FILM self.film.permalink = "battlefirld-earth-ii" self.film.release_year = 2010 self.film.save() self.film.save_tags('comedy') self.film1 = Film() self.film1.title = "Battlefield Earth III" self.film1.type = Object.TYPE_FILM self.film1.permalink = "battlefirld-earth-iii" self.film1.release_year = 2010 self.film1.save() self.film1.save_tags('sci-fi') self.film2 = Film() self.film2.title = "Battlefield Earth IV" self.film2.type = Object.TYPE_FILM self.film2.permalink = "battlefirld-earth-iv" self.film2.release_year = 2010 self.film2.save() self.film2.save_tags('horror') self.rating = Rating() self.rating.parent = self.film self.rating.user = self.u1 self.rating.film = self.film self.rating.type = Rating.TYPE_FILM self.rating.rating = 8 self.rating.normalized = 8 self.rating.save() self.rating1 = Rating() self.rating1.parent = self.film1 self.rating1.user = self.u1 self.rating1.film = self.film1 self.rating1.type = Rating.TYPE_FILM self.rating1.rating = 8 self.rating1.normalized = 8 self.rating1.save() self.rating2 = Rating() self.rating2.parent = self.film2 self.rating2.user = self.u1 self.rating2.film = self.film2 self.rating2.type = Rating.TYPE_FILM self.rating2.rating = 8 self.rating2.normalized = 8 self.rating2.save() def films_for_comparator(self): # films for comnparator self.f1 = Film(type=1, permalink='przypadek', imdb_code=111, status=1, version=1, release_year=1999, title='Przypadek', popularity=20, popularity_month=10) self.f1.save() self.f2 = Film(type=1, permalink='wrestler', imdb_code=112, status=1, version=1, release_year=2008, title='Wrestler', popularity=10, popularity_month=1) self.f2.save() self.f3 = Film(type=1, permalink='american-history-x', imdb_code=113, status=1, version=1, release_year=1998, title='American History X', popularity=1, popularity_month=1) self.f3.save() self.f4 = Film(type=1, permalink='the-big-lebowski', imdb_code=114, status=1, version=1, release_year=1998, title='The Big Lebowski', popularity=1, popularity_month=1) self.f4.save() self.f5 = Film(type=1, permalink='the-lord-of-the-rings-the-fellowship-of-the-ring', imdb_code=115, status=1, version=1, release_year=2001, title='The Lord of the Rings: The Fellowship of the Ring', popularity=1, popularity_month=1) self.f5.save() self.f6 = Film(type=1, permalink='raiders-of-the-lost-ark', imdb_code=116, status=1, version=1, release_year=1981, title='Raiders of the Lost Ark', popularity=1, popularity_month=1) self.f6.save() self.f7 = Film(type=1, permalink='the-alien', imdb_code=117, status=1, version=1, release_year=1979, title='The Alien', popularity=1, popularity_month=1) self.f7.save() self.f8 = Film(type=1, permalink='terminator-2-judgment-day', imdb_code=118, status=1, version=1, release_year=1991, title='Terminator 2: Judgment Day', popularity=1, popularity_month=1) self.f8.save() self.f9 = Film(type=1, permalink='przypadek', imdb_code=119, status=1, version=1, release_year=1999, title='Przypadek', popularity=20, popularity_month=10) self.f9.save() self.f10 = Film(type=1, permalink='wrestler', imdb_code=120, status=1, version=1, release_year=2008, title='Wrestler', popularity=10, popularity_month=1) self.f10.save() self.f11 = Film(type=1, permalink='american-history-x', imdb_code=121, status=1, version=1, release_year=1998, title='American History X', popularity=1, popularity_month=1) self.f11.save() self.f12 = Film(type=1, permalink='the-big-lebowski', imdb_code=122, status=1, version=1, release_year=1998, title='The Big Lebowski', popularity=1, popularity_month=1) self.f12.save() self.f13 = Film(type=1, permalink='the-lord-of-the-rings-the-fellowship-of-the-ring', imdb_code=123, status=1, version=1, release_year=2001, title='The Lord of the Rings: The Fellowship of the Ring', popularity=1, popularity_month=1) self.f13.save() self.f14 = Film(type=1, permalink='raiders-of-the-lost-ark', imdb_code=124, status=1, version=1, release_year=1981, title='Raiders of the Lost Ark', popularity=1, popularity_month=1) self.f14.save() self.f15 = Film(type=1, permalink='the-alien', imdb_code=125, status=1, version=1, release_year=1979, title='The Alien', popularity=1, popularity_month=1) self.f15.save() self.f16 = Film(type=1, permalink='przypadek', imdb_code=126, status=1, version=1, release_year=1999, title='Przypadek', popularity=20, popularity_month=10) self.f16.save() self.f17 = Film(type=1, permalink='wrestler', imdb_code=127, status=1, version=1, release_year=2008, title='Wrestler', popularity=10, popularity_month=1) self.f17.save() self.f18 = Film(type=1, permalink='american-history-x', imdb_code=128, status=1, version=1, release_year=1998, title='American History X', popularity=1, popularity_month=1) self.f18.save() self.f19 = Film(type=1, permalink='the-big-lebowski', imdb_code=129, status=1, version=1, release_year=1998, title='The Big Lebowski', popularity=1, popularity_month=1) self.f19.save() self.f20 = Film(type=1, permalink='the-lord-of-the-rings-the-fellowship-of-the-ring', imdb_code=130, status=1, version=1, release_year=2001, title='The Lord of the Rings: The Fellowship of the Ring', popularity=1, popularity_month=1) self.f20.save() self.f21 = Film(type=1, permalink='raiders-of-the-lost-ark', imdb_code=131, status=1, version=1, release_year=1981, title='Raiders of the Lost Ark', popularity=1, popularity_month=1) self.f21.save() self.f22 = Film(type=1, permalink='the-alien', imdb_code=132, status=1, version=1, release_year=1979, title='The Alien', popularity=1, popularity_month=1) self.f22.save() self.f23 = Film(type=1, permalink='terminator-2-judgment-day', imdb_code=133, status=1, version=1, release_year=1991, title='Terminator 2: Judgment Day', popularity=1, popularity_month=1) self.f23.save() self.f24 = Film(type=1, permalink='przypadek', imdb_code=134, status=1, version=1, release_year=1999, title='Przypadek', popularity=20, popularity_month=10) self.f24.save() self.f25 = Film(type=1, permalink='wrestler', imdb_code=135, status=1, version=1, release_year=2008, title='Wrestler', popularity=10, popularity_month=1) self.f25.save() self.f26 = Film(type=1, permalink='american-history-x', imdb_code=136, status=1, version=1, release_year=1998, title='American History X', popularity=1, popularity_month=1) self.f26.save() self.f27 = Film(type=1, permalink='the-big-lebowski', imdb_code=137, status=1, version=1, release_year=1998, title='The Big Lebowski', popularity=1, popularity_month=1) self.f27.save() self.f28 = Film(type=1, permalink='the-lord-of-the-rings-the-fellowship-of-the-ring', imdb_code=138, status=1, version=1, release_year=2001, title='The Lord of the Rings: The Fellowship of the Ring', popularity=1, popularity_month=1) self.f28.save() self.f29 = Film(type=1, permalink='raiders-of-the-lost-ark', imdb_code=139, status=1, version=1, release_year=1981, title='Raiders of the Lost Ark', popularity=1, popularity_month=1) self.f29.save() self.f30 = Film(type=1, permalink='the-alien', imdb_code=140, status=1, version=1, release_year=1979, title='The Alien', popularity=1, popularity_month=1) self.f30.save() self.f1.save_tags('comedy drama horror animation mystery') self.f2.save_tags('comedy sci-fi thriller series') self.f3.save_tags('comedy') self.f4.save_tags('sci-fi') self.f5.save_tags('horror') self.f6.save_tags('thriller') self.f7.save_tags('drama') self.f8.save_tags('series') self.f9.save_tags('sci-fi') self.f10.save_tags('fantasy') self.f11.save_tags('mystery') self.f12.save_tags('action') self.f13.save_tags('adventure') self.f14.save_tags('animation') self.f15.save_tags('family') self.f16.save_tags('thriller') self.f17.save_tags('drama') self.f18.save_tags('series') self.f19.save_tags('sci-fi') self.f20.save_tags('fantasy') self.f21.save_tags('comedy') self.f22.save_tags('animation') self.f23.save_tags('comedy') self.f24.save_tags('sci-fi') self.f25.save_tags('horror') self.f26.save_tags('thriller') self.f27.save_tags('drama') self.f28.save_tags('series') self.f29.save_tags('sci-fi') self.f30.save_tags('fantasy') #First user ratings save_rating(self.f1, self.u1, 1) save_rating(self.f2, self.u1, 2) save_rating(self.f3, self.u1, 3) save_rating(self.f4, self.u1, 4) save_rating(self.f5, self.u1, 5) save_rating(self.f6, self.u1, 6) save_rating(self.f7, self.u1, 7) save_rating(self.f8, self.u1, 8) save_rating(self.f9, self.u1, 9) save_rating(self.f10, self.u1, 10) save_rating(self.f11, self.u1, 1) save_rating(self.f12, self.u1, 2) save_rating(self.f13, self.u1, 3) save_rating(self.f14, self.u1, 4) save_rating(self.f15, self.u1, 5) save_rating(self.f16, self.u1, 6) save_rating(self.f17, self.u1, 7) save_rating(self.f18, self.u1, 8) save_rating(self.f19, self.u1, 9) save_rating(self.f20, self.u1, 10) save_rating(self.f21, self.u1, 1) save_rating(self.f22, self.u1, 2) save_rating(self.f23, self.u1, 3) save_rating(self.f24, self.u1, 4) save_rating(self.f25, self.u1, 5) save_rating(self.f26, self.u1, 6) save_rating(self.f27, self.u1, 7) save_rating(self.f28, self.u1, 8) save_rating(self.f29, self.u1, 9) save_rating(self.f30, self.u1, 10) #Second user ratings - more than minimal common films requirement (15) save_rating(self.f1, self.u2, 1) save_rating(self.f2, self.u2, 2) save_rating(self.f3, self.u2, 3) save_rating(self.f4, self.u2, 4) save_rating(self.f5, self.u2, 5) save_rating(self.f6, self.u2, 6) save_rating(self.f7, self.u2, 7) save_rating(self.f8, self.u2, 8) save_rating(self.f9, self.u2, 9) save_rating(self.f10, self.u2, 10) save_rating(self.f11, self.u2, 1) #save_rating(self.f12, self.u2, 2) save_rating(self.f13, self.u2, 3) #save_rating(self.f14, self.u2, 4) save_rating(self.f15, self.u2, 5) #save_rating(self.f16, self.u2, 6) save_rating(self.f17, self.u2, 7) #save_rating(self.f18, self.u2, 8) save_rating(self.f19, self.u2, 9) #save_rating(self.f20, self.u2, 10) save_rating(self.f21, self.u2, 1) #save_rating(self.f22, self.u2, 2) save_rating(self.f23, self.u2, 3) #save_rating(self.f24, self.u2, 4) save_rating(self.f25, self.u2, 5) #save_rating(self.f26, self.u2, 6) save_rating(self.f27, self.u2, 7) #save_rating(self.f28, self.u2, 8) save_rating(self.f29, self.u2, 9) #save_rating(self.f30, self.u2, 10) # Third user ratings - equal with minimal common films requirement (15) #save_rating(self.f1, self.u3, 1) #save_rating(self.f2, self.u3, 2) #save_rating(self.f3, self.u3, 3) #save_rating(self.f4, self.u3, 4) #save_rating(self.f5, self.u3, 5) #save_rating(self.f6, self.u3, 6) #save_rating(self.f7, self.u3, 7) #save_rating(self.f8, self.u3, 8) #save_rating(self.f9, self.u3, 9) #save_rating(self.f10, self.u3, 10) #save_rating(self.f11, self.u3, 1) #save_rating(self.f12, self.u3, 2) #save_rating(self.f13, self.u3, 3) #save_rating(self.f14, self.u3, 4) #save_rating(self.f15, self.u3, 5) save_rating(self.f16, self.u3, 6) save_rating(self.f17, self.u3, 7) save_rating(self.f18, self.u3, 8) save_rating(self.f19, self.u3, 9) save_rating(self.f20, self.u3, 10) save_rating(self.f21, self.u3, 1) save_rating(self.f22, self.u3, 2) save_rating(self.f23, self.u3, 3) save_rating(self.f24, self.u3, 4) save_rating(self.f25, self.u3, 5) save_rating(self.f26, self.u3, 6) save_rating(self.f27, self.u3, 7) save_rating(self.f28, self.u3, 8) save_rating(self.f29, self.u3, 9) save_rating(self.f30, self.u3, 10) # Fourth user ratings - less than minimal films requirement #save_rating(self.f1, self.u4, 1) #save_rating(self.f2, self.u4, 2) #save_rating(self.f3, self.u4, 3) #save_rating(self.f4, self.u4, 4) #save_rating(self.f5, self.u4, 5) #save_rating(self.f6, self.u4, 6) #save_rating(self.f7, self.u4, 7) #save_rating(self.f8, self.u4, 8) #save_rating(self.f9, self.u4, 9) #save_rating(self.f10, self.u4, 10) #save_rating(self.f11, self.u4, 1) #save_rating(self.f12, self.u4, 2) #save_rating(self.f13, self.u4, 3) #save_rating(self.f14, self.u4, 4) #save_rating(self.f15, self.u4, 5) #save_rating(self.f16, self.u4, 6) #save_rating(self.f17, self.u4, 7) #save_rating(self.f18, self.u4, 8) #save_rating(self.f19, self.u4, 9) #save_rating(self.f20, self.u4, 10) save_rating(self.f21, self.u4, 1) save_rating(self.f22, self.u4, 2) save_rating(self.f23, self.u4, 3) save_rating(self.f24, self.u4, 4) save_rating(self.f25, self.u4, 5) save_rating(self.f26, self.u4, 6) save_rating(self.f27, self.u4, 7) save_rating(self.f28, self.u4, 8) save_rating(self.f29, self.u4, 9) save_rating(self.f30, self.u4, 10) def clean_data(self): User.objects.all().delete() Post.objects.all().delete() Rating.objects.all().delete() UserActivity.objects.all().delete() def test_saving_post_activity(self): """ Test saving post activity """ self.initialize() # set up post post = Post() post.title = "Lorem ipsum" post.permalink = "lorem-ipsum" post.body = "siala lala tralala" post.user = self.u1 post.status = Object.PUBLIC_STATUS post.type = Object.TYPE_POST post.save() post.related_film.add(self.film) # this method will call save_activity method and in this case create new activity post.save() activity = UserActivity.objects.get(post=post) # testing if activity was saved properly self.failUnlessEqual(activity.title, post.title) self.failUnlessEqual(activity.content, post.body) self.failUnlessEqual(activity.permalink, post.get_absolute_url()) self.failUnlessEqual(activity.status, post.status) self.failUnlessEqual(activity.activity_type, UserActivity.TYPE_POST) self.failUnlessEqual(activity.username, self.u1.username) self.failUnlessEqual(activity.film_title, self.film.title) self.failUnlessEqual(activity.film_permalink, self.film.permalink) # TODO: add a get_absolute_url(), get_slug() and get_subdomain() test def test_updating_post_activity(self): """ Test updating post activity """ self.initialize() # set up post post = Post() post.title = "Lorem ipsum" post.permalink = "lorem-ipsum" post.body = "siala lala tralala" post.user = self.u1 post.status = Object.PUBLIC_STATUS post.type = Object.TYPE_POST post.save() post.related_film.add(self.film) post.save() post.title = "Bum bum bum" post.permalink = "bum-bum-bum" post.body = "Lubuduubu lubu dubu niech zyje nam prezes naszego klubu!" post.status = Object.DRAFT_STATUS post.save() activity = UserActivity.objects.get(post=post) # testing if activity was updated properly self.failUnlessEqual(activity.title, "Bum bum bum") self.failUnlessEqual(activity.content, "Lubuduubu lubu dubu niech zyje nam prezes naszego klubu!") self.failUnlessEqual(activity.permalink, post.get_absolute_url()) self.failUnlessEqual(activity.status, UserActivity.DRAFT_STATUS) self.failUnlessEqual(activity.activity_type, UserActivity.TYPE_POST) self.failUnlessEqual(activity.username, self.u1.username) self.failUnlessEqual(activity.film_title, self.film.title) self.failUnlessEqual(activity.film_permalink, self.film.permalink) def test_saving_shortreview_activity(self): """ Test saving shortreview activity """ self.initialize() # set up Shortreview obj = Object.objects.get(id=self.film.id) shortreview = ShortReview() shortreview.user = self.u1 shortreview.review_text = "sialala bumcyk cyk" shortreview.status = Object.PUBLIC_STATUS shortreview.type = Object.TYPE_SHORT_REVIEW shortreview.object = obj shortreview.kind = ShortReview.REVIEW shortreview.save() activity = UserActivity.objects.get(short_review=shortreview) # testing if activity was saved properly self.failUnlessEqual(activity.title, shortreview.get_title()) self.failUnlessEqual(activity.content, shortreview.review_text) self.failUnlessEqual(activity.permalink, shortreview.get_absolute_url()) self.failUnlessEqual(activity.status, shortreview.status) self.failUnlessEqual(activity.activity_type, UserActivity.TYPE_SHORT_REVIEW) self.failUnlessEqual(activity.username, self.u1.username) self.failUnlessEqual(activity.film_title, self.film.title) self.failUnlessEqual(activity.film_permalink, self.film.permalink) def test_updating_shortreview_activity(self): """ Test updating shortreview activity """ self.initialize() # set up Shortreview obj = Object.objects.get(id=self.film.id) shortreview = ShortReview() shortreview.user = self.u1 shortreview.review_text = "sialala bumcyk cyk" shortreview.status = Object.PUBLIC_STATUS shortreview.type = Object.TYPE_SHORT_REVIEW shortreview.object = obj shortreview.kind = ShortReview.REVIEW shortreview.save() shortreview = ShortReview.objects.get(id=shortreview.id) shortreview.review_text = "Lorem ipsum" shortreview.status = Object.DELETED_STATUS shortreview.save() activity = UserActivity.objects.get(short_review=shortreview) # testing if activity was updated properly self.failUnlessEqual(activity.title, shortreview.get_title()) self.failUnlessEqual(activity.content, "Lorem ipsum") self.failUnlessEqual(activity.permalink, shortreview.get_absolute_url()) self.failUnlessEqual(activity.status, UserActivity.DELETED_STATUS) self.failUnlessEqual(activity.activity_type, UserActivity.TYPE_SHORT_REVIEW) self.failUnlessEqual(activity.username, self.u1.username) self.failUnlessEqual(activity.film_title, self.film.title) self.failUnlessEqual(activity.film_permalink, self.film.permalink) def test_saving_externallink_activity(self): """ Test saving externallink activity """ self.initialize() # set up Externallink ext = ExternalLink() ext.title = "Link title!" ext.url = "http://filmaster.pl" ext.url_kind = ExternalLink.REVIEW ext.excerpt = "Lorem ipsum" ext.film = self.film ext.user = self.u1 ext.status = Object.PUBLIC_STATUS ext.type = Object.TYPE_LINK ext.save() self.assertRaises(UserActivity.DoesNotExist, UserActivity.objects.get, link=ext) ext.moderation_status = ExternalLink.STATUS_ACCEPTED ext.save() activity = UserActivity.objects.get(link=ext) # testing if activity was saved properly self.failUnlessEqual(activity.title, ext.title) self.failUnlessEqual(activity.content, ext.excerpt) self.failUnlessEqual(activity.get_absolute_url(), ext.get_absolute_url()) self.failUnlessEqual(activity.status, ext.status) self.failUnlessEqual(activity.activity_type, UserActivity.TYPE_LINK) self.failUnlessEqual(activity.username, self.u1.username) self.failUnlessEqual(activity.film_title, self.film.title) self.failUnlessEqual(activity.film_permalink, self.film.permalink) def test_updating_externallink_activity(self): """ Test updating externallink activity """ self.initialize() # set up Externallink ext = ExternalLink() ext.title = "Link title!" ext.url = "http://filmaster.pl" ext.url_kind = ExternalLink.REVIEW ext.excerpt = "Lorem ipsum" ext.film = self.film ext.user = self.u1 ext.status = Object.PUBLIC_STATUS ext.type = Object.TYPE_LINK ext.save() ext.url = "http://osnews.pl" ext.excerpt = "Lorem lorem" ext.save() activity = UserActivity.objects.get(link=ext) # testing if activity was saved properly self.failUnlessEqual(activity.title, ext.title) self.failUnlessEqual(activity.content, "Lorem lorem") self.failUnlessEqual(activity.get_absolute_url(), ext.get_absolute_url()) self.failUnlessEqual(activity.status, ext.status) self.failUnlessEqual(activity.url, "http://osnews.pl") self.failUnlessEqual(activity.activity_type, UserActivity.TYPE_LINK) self.failUnlessEqual(activity.username, self.u1.username) self.failUnlessEqual(activity.film_title, self.film.title) self.failUnlessEqual(activity.film_permalink, self.film.permalink) def test_saving_comment_activity(self): """ Test saving comment activity """ self.initialize() # set up post for commenting post = Post() post.title = "Lorem ipsum" post.permalink = "lorem-ipsum" post.body = "siala lala tralala" post.user = self.u1 post.status = Object.PUBLIC_STATUS post.type = Object.TYPE_POST post.save() post.related_film.add(self.film) post.save() comment = ThreadedComment() comment.user = self.u1 comment.comment = "Lorem ipsum" comment.content_object = post comment.status = Object.PUBLIC_STATUS comment.type=Object.TYPE_COMMENT comment.permalink="COMMENT" comment.save() activity = UserActivity.objects.get(comment=comment) # testing if activity was saved properly self.failUnlessEqual(activity.title, comment.content_object.get_comment_title()) self.failUnlessEqual(activity.content, comment.comment) if settings.ENSURE_OLD_STYLE_PERMALINKS: self.failUnlessEqual(activity.permalink, comment.content_object.get_absolute_url_old_style()+"#"+ str(comment.id)) else: self.failUnlessEqual(activity.permalink, comment.content_object.get_absolute_url()+"#"+ str(comment.id)) self.failUnlessEqual(activity.status, comment.status) self.failUnlessEqual(activity.activity_type, UserActivity.TYPE_COMMENT) self.failUnlessEqual(activity.username, self.u1.username) #planet_tag_helper without tag = 1 comment pth = PlanetTagHelper() comments = pth.planet_tag_comments() self.assertEquals(len(comments), 1) #planet_tag_helper with tag 'comedy' = 0 comment #no assigned films to comment pth_tags = PlanetTagHelper(tag='comedy') comments = pth_tags.planet_tag_comments() self.assertEquals(len(comments), 0) def test_updating_comment_activity(self): """ Test updating comment activity """ self.initialize() # set up post for commenting post = Post() post.title = "Lorem ipsum" post.permalink = "lorem-ipsum" post.body = "siala lala tralala" post.user = self.u1 post.status = Object.PUBLIC_STATUS post.type = Object.TYPE_POST post.save() post.related_film.add(self.film) post.save() comment = ThreadedComment() comment.user = self.u1 comment.comment = "Lorem ipsum" comment.content_object = post comment.status = Object.PUBLIC_STATUS comment.type=Object.TYPE_COMMENT comment.permalink="COMMENT" comment.save() comment.comment = "siala lala tralala" comment.status = Object.DELETED_STATUS comment.save() activity = UserActivity.objects.get(comment=comment) # testing if activity was updated properly self.failUnlessEqual(activity.title, comment.content_object.get_comment_title()) self.failUnlessEqual(activity.content, "siala lala tralala") if settings.ENSURE_OLD_STYLE_PERMALINKS: self.failUnlessEqual(activity.permalink, comment.content_object.get_absolute_url_old_style()+"#"+ str(comment.id)) else: self.failUnlessEqual(activity.permalink, comment.content_object.get_absolute_url()+"#"+ str(comment.id)) self.failUnlessEqual(activity.status, UserActivity.DELETED_STATUS) self.failUnlessEqual(activity.activity_type, UserActivity.TYPE_COMMENT) self.failUnlessEqual(activity.username, self.u1.username) def test_saving_draft_post_activity(self): """ Test saving draft activity Make sure that activity is not created for draft posts """ self.initialize() post = Post() post.title = "Lorem ipsum" post.permalink = "lorem-ipsum" post.body = "siala lala tralala" post.user = self.u1 post.status = Object.DRAFT_STATUS post.type = Object.TYPE_POST post.save() post.related_film.add(self.film) post.save() self.failUnlessEqual(UserActivity.objects.all().count(), 1) # now we want to create an activity each time act = UserActivity.objects.get(post=post) self.failUnlessEqual(act.status, UserActivity.DRAFT_STATUS) post.status = Post.PUBLIC_STATUS post.save() self.failUnlessEqual(UserActivity.objects.all().count(), 1) act = UserActivity.objects.get(post=post) self.failUnlessEqual(act.status, UserActivity.PUBLIC_STATUS) post.status = Post.DRAFT_STATUS post.save() self.failUnlessEqual(UserActivity.objects.all().count(), 1) act = UserActivity.objects.get(post=post) self.failUnlessEqual(act.status, UserActivity.DRAFT_STATUS) def test_deleting_post_activity(self): """ Test deleting post activity """ self.initialize() post = Post() post.title = "Lorem ipsum" post.permalink = "lorem-ipsum" post.body = "siala lala tralala" post.user = self.u1 post.status = Object.PUBLIC_STATUS post.type = Object.TYPE_POST post.save() post.related_film.add(self.film) post.save() act = UserActivity.objects.get(post=post) self.failUnlessEqual(act.status, UserActivity.PUBLIC_STATUS) post.status = Object.DELETED_STATUS post.save() act = UserActivity.objects.get(post=post) self.failUnlessEqual(act.status, UserActivity.DELETED_STATUS) def test_updating_relatedfilm_post_activity(self): """ Test updating film fields in post activity when related films are removed from post """ self.initialize() self.client.login(username='******', password='******') # set up post post = Post() post.title = "Lorem ipsum" post.permalink = "lorem-ipsum" post.body = "siala lala tralala" post.user = self.u1 post.status = Object.PUBLIC_STATUS post.type = Object.TYPE_POST post.save() post.related_film.add(self.film) post.save() activity = UserActivity.objects.get(post=post) # testing if activity was saved properly self.failUnlessEqual(activity.title, "Lorem ipsum") self.failUnlessEqual(activity.content, "siala lala tralala") self.failUnlessEqual(activity.permalink, post.get_absolute_url()) self.failUnlessEqual(activity.status, UserActivity.PUBLIC_STATUS) self.failUnlessEqual(activity.activity_type, UserActivity.TYPE_POST) self.failUnlessEqual(activity.username, self.u1.username) self.failUnlessEqual(activity.film is None, False) self.failUnlessEqual(activity.film_permalink, self.film.permalink) self.failUnlessEqual(activity.film_title, self.film.title) # update post post.title = "New title" post.body = "Lorem ipsum! Lorem ipsum! Lorem ipsum!" post.related_film = [] post.save() activity = UserActivity.objects.get(post=post) # testing if activity was updated properly self.failUnlessEqual(activity.title, "New title") self.failUnlessEqual(activity.content, "Lorem ipsum! Lorem ipsum! Lorem ipsum!") self.failUnlessEqual(activity.permalink, post.get_absolute_url()) self.failUnlessEqual(activity.status, UserActivity.PUBLIC_STATUS) self.failUnlessEqual(activity.activity_type, UserActivity.TYPE_POST) self.failUnlessEqual(activity.username, self.u1.username) self.failUnlessEqual(activity.film is None, True) self.failUnlessEqual(activity.film_permalink, None) self.failUnlessEqual(activity.film_title, None) def test_rating_activity(self): """ Test creating rating activity """ self.initialize() ratings = Rating.objects.filter(user=self.u1) userratingtime = UserRatingTimeRange() userratingtime.save_activity(ratings, 3, self.u1) ua = UserActivity.objects.get(activity_type=UserActivity.TYPE_RATING, user=self.u1) self.failUnlessEqual(ua is None, False) self.failUnlessEqual(ua.user, self.u1) self.failUnlessEqual(ua.content, u'2') def test_planet_tag(self): # --- Initialize --- self.initialize() helper_without_tags = PlanetTagHelper() helper_with_tags = PlanetTagHelper(tag='comedy') recom_helper = RecomHelper() user_activity_helper = UserActivityHelper() # --- setup followers --- self.u1.followers.follow(self.u2) self.u1.followers.follow(self.u3) self.u1.followers.follow(self.u4) # similar users from film20.recommendations.bot_helper import do_create_comparators self.films_for_comparator() do_create_comparators() friends_list = recom_helper.get_following_ids_as_list(self.u1) friends_without_activities = helper_without_tags.planet_tag_friends(friends_list) self.assertEquals(len(friends_without_activities), 0) similar_users_list = user_activity_helper.get_similar_users_list(request) similar_users_without_tags = helper_without_tags.planet_tag_similar_users(similar_users_list) self.assertEquals(len(similar_users_without_tags), 0) notes_without_tags = helper_without_tags.planet_tag_notes() self.assertEquals(len(notes_without_tags), 0) all_planet_activities = helper_without_tags.planet_tag_all() self.assertEquals(len(all_planet_activities), 0) # --- setup User activities --- post = Post() post.title = "Lorem ipsum" post.permalink = "lorem-ipsum" post.body = "siala lala tralala" post.user = self.u1 post.status = Object.PUBLIC_STATUS post.type = Object.TYPE_POST post.save() post.related_film.add(self.film) post.save() all_activities = helper_without_tags.planet_tag_all() self.assertEquals(len(all_activities), 1) post = Post() post.title = "Lorem ipsum" post.permalink = "lorem-ipsum" post.body = "siala lala tralala" post.user = self.u1 post.status = Object.PUBLIC_STATUS post.type = Object.TYPE_POST post.save() post.related_film.add(self.film1) post.save() all_activities = helper_without_tags.planet_tag_all() self.assertEquals(len(all_activities), 2) obj = Object.objects.get(id=self.film2.id) shortreview = ShortReview() shortreview.user = self.u1 shortreview.review_text = "sialala bumcyk cyk" shortreview.status = Object.PUBLIC_STATUS shortreview.type = Object.TYPE_SHORT_REVIEW shortreview.object = obj shortreview.kind = ShortReview.REVIEW shortreview.save() all_activities = helper_without_tags.planet_tag_all() self.assertEquals(len(all_activities), 3) # Activities with 'comedy' tag all_activities_with_tags = helper_with_tags.planet_tag_all() self.assertEquals(len(all_activities_with_tags), 1) # --- setup followers activities --- post = Post() post.title = "#1 Lorem ipsum" post.permalink = "lorem-ipsum" post.body = "siala lala tralala" post.user = self.u2 post.status = Object.PUBLIC_STATUS post.type = Object.TYPE_POST post.save() post.related_film.add(self.film) post.save() all_activities = helper_without_tags.planet_tag_all() self.assertEquals(len(all_activities), 4) post = Post() post.title = "#2 Lorem ipsum" post.permalink = "lorem-ipsum" post.body = "siala lala tralala" post.user = self.u3 post.status = Object.PUBLIC_STATUS post.type = Object.TYPE_POST post.save() post.related_film.add(self.film1) post.save() all_activities = helper_without_tags.planet_tag_all() self.assertEquals(len(all_activities), 5) obj = Object.objects.get(id=self.film2.id) shortreview = ShortReview() shortreview.user = self.u4 shortreview.review_text = "sialala bumcyk cyk" shortreview.status = Object.PUBLIC_STATUS shortreview.type = Object.TYPE_SHORT_REVIEW shortreview.object = obj shortreview.kind = ShortReview.REVIEW shortreview.save() all_activities = helper_without_tags.planet_tag_all() self.assertEquals(len(all_activities), 6) activities_with_tags = helper_with_tags.planet_tag_all() self.assertEquals(len(activities_with_tags), 2) all_friends_activities = helper_without_tags.planet_tag_friends(friends_list) self.assertEquals(len(all_friends_activities), 3) friends_activities_with_tag = helper_with_tags.planet_tag_friends(friends_list) self.assertEquals(len(friends_activities_with_tag), 1) # --- Check notes --- all_planet_notes = helper_without_tags.planet_tag_notes() self.assertEquals(len(all_planet_notes), 4) # Post notes with 'comedy' tag notes_activities_with_tag = helper_with_tags.planet_tag_notes() self.assertEquals(len(notes_activities_with_tag), 2) similar_users_activities = helper_without_tags.planet_tag_similar_users(similar_users_list) self.assertEquals(len(similar_users_activities), 2) users_with_tags = helper_with_tags.planet_tag_similar_users(similar_users_list) self.assertEquals(len(users_with_tags), 1) activities = UserActivity.objects.all()