def test_get_primary_contributors(self):
# create website site item and set as current site
web_site = Site(domain="web.address.com")
web_site.save()
settings.SITE_ID = web_site.id
# create a track with some credits
track = Track(title="title")
track.save()
contributor1 = TrackContributor(title="title", state="published")
contributor1.save()
contributor1.sites.add(web_site)
contributor2 = TrackContributor(title="title", state="published")
contributor2.save()
contributor2.sites.add(web_site)
contributor3 = TrackContributor(title="title", state="published")
contributor3.save()
contributor3.sites.add(web_site)
contributor4 = TrackContributor(title="title", state="published")
contributor4.save()
contributor4.sites.add(web_site)
unpublished_contributor = TrackContributor(title="title")
unpublished_contributor.save()
Credit(track=track, contributor=contributor1, role=2).save()
Credit(track=track, contributor=contributor2, role=10).save()
Credit(track=track, contributor=contributor3, role=2).save()
Credit(track=track, contributor=unpublished_contributor, role=2).save()
Credit(track=track, contributor=contributor4).save()
# result should only contain contributors with highest role.
# can contain multiples.
# highest role not neccessarily 1.
# result should not include non permitted contributors.
# result should not include contributors with None credit role
primary_contributors = track.get_primary_contributors()
self.failUnless(contributor1 in primary_contributors)
self.failUnless(contributor3 in primary_contributors)
self.failIf(contributor2 in primary_contributors)
self.failIf(unpublished_contributor in primary_contributors)
self.failIf(contributor4 in primary_contributors)
