def test_entity_index(self):
author = testing.create_user()
jack_white = Entity.create_for_testing()
seven_nations_army = Song.create_for_testing(author)
self.add_contribution(seven_nations_army, jack_white, True)
jolene = Song.create_for_testing(author)
self.add_contribution(jolene, jack_white, True)
# Approve only Jolene.
seven_nations_army.reviewed = False
seven_nations_army.save()
jolene.reviewed = True
jolene.save()
# General public should see only Jolene.
response = testing.get_public_client().get(
jack_white.get_absolute_url())
self.assertEqual(200, response.status_code)
self.assertEqual(1, len(response.context['songs']))
# The author should see both.
response = testing.get_user_client(author).get(
jack_white.get_absolute_url())
self.assertEqual(200, response.status_code)
self.assertEqual(2, len(response.context['songs']))
# Any logged-in user should see both, too.
response = testing.get_user_client().get(jack_white.get_absolute_url())
self.assertEqual(200, response.status_code)
self.assertEqual(2, len(response.context['songs']))
def setUp(self):
author = testing.create_user()
song = Song.create_for_testing(author)
entity = Entity.create_for_testing()
contribution = EntityContribution()
contribution.song = song
contribution.entity = entity
contribution.texted = True
contribution.save()
song.old_slug = "some-old-slug"
song.reviewed = True
song.save()
self.song = song
self.entity = entity
