def test_get_watched_categories_with_no_smart_category(self): """ categories of media watched are returned from most recent watched to less recent watched :return: """ member = Member.objects.get(pk='56eb6d04b37b3379b531e011') watched = get_watched(member) watched_categories = get_watched_categories(watched) expected_categories = [Category.objects.get(pk=pk) for pk in ('56eb6d04b37b3379b531e092', '56eb6d04b37b3379b531e091', '56eb6d04b37b3379b531e093')] self.assertListEqual(watched_categories, expected_categories)
def test_get_watched_categories_with_one_smart_category(self): """ smart categories should not be taken into account in get_watched_categories :return: """ category = Category.objects.get(pk='56eb6d04b37b3379b531e091') category.smart = True category.save() member = Member.objects.get(pk='56eb6d04b37b3379b531e011') watched = get_watched(member) watched_categories = [category.slug for category in get_watched_categories(watched)] expected_categories = [Category.objects.get(pk=pk).slug for pk in ('56eb6d04b37b3379b531e092', '56eb6d04b37b3379b531e093')] self.assertListEqual(watched_categories, expected_categories)
def test_get_watched_categories_with_no_smart_category(self): """ categories of media watched are returned from most recent watched to less recent watched :return: """ member = Member.objects.get(pk='56eb6d04b37b3379b531e011') watched = get_watched(member) watched_categories = get_watched_categories(watched) expected_categories = [ Category.objects.get(pk=pk) for pk in ('56eb6d04b37b3379b531e092', '56eb6d04b37b3379b531e091', '56eb6d04b37b3379b531e093') ] self.assertListEqual(watched_categories, expected_categories)
def test_get_watched_categories_with_one_smart_category(self): """ smart categories should not be taken into account in get_watched_categories :return: """ category = Category.objects.get(pk='56eb6d04b37b3379b531e091') category.smart = True category.save() member = Member.objects.get(pk='56eb6d04b37b3379b531e011') watched = get_watched(member) watched_categories = [ category.slug for category in get_watched_categories(watched) ] expected_categories = [ Category.objects.get(pk=pk).slug for pk in ('56eb6d04b37b3379b531e092', '56eb6d04b37b3379b531e093') ] self.assertListEqual(watched_categories, expected_categories)