def setUp(self):
self.browser = webdriver.Firefox()
self.browser.implicitly_wait(3)
strawhat = factories.Crew()
alvida_crew = factories.Crew(name='Alvida Pirates')
kuja = factories.Crew(name='Kuja Pirates')
self.luffy = factories.Pirate(crew=strawhat)
self.alvida = factories.Pirate(name='Iron Mace Alvida',
crew=alvida_crew)
factories.Pirate(name='God Usopp', bounty=200000000, crew=strawhat)
factories.Pirate(name='Chopper', bounty=100, crew=strawhat)
factories.Pirate(name='Roronoa Zoro', bounty=320000000, crew=strawhat)
factories.Pirate(name='Sanji',
bounty=177000000,
crew=strawhat,
wanted_status=Pirate.ONLY_ALIVE)
factories.Pirate(name='Nico Robin', bounty=130000000, crew=strawhat)
factories.Pirate(name='Franky', bounty=94000000, crew=strawhat)
factories.Pirate(name='Nami', bounty=66000000, crew=strawhat)
factories.Pirate(name='Soul King', bounty=83000000, crew=strawhat)
self.hancock = factories.Pirate(name='Boa Hancock',
bounty=None,
crew=kuja,
wanted_status=None)
self.sandersonia = factories.Pirate(name='Boa Sandersonia',
bounty=None,
crew=kuja,
wanted_status=None)
self.marigold = factories.Pirate(name='Boa Marigold',
bounty=None,
crew=kuja,
wanted_status=None)
def test_should_return_wanted_status_string(self):
crew = factories.Crew()
pirate = factories.Pirate(crew=crew, wanted_status=Pirate.ONLY_ALIVE)
result = Pirate.objects.first()
status_choices_dict = dict(Pirate.STATUS_CHOICES)
self.assertEqual(pirate.get_wanted_status_display(),
status_choices_dict.get(Pirate.ONLY_ALIVE))
def test_shows_crew_of_matching_pirates(self):
crew = factories.Crew(name='Kuja Pirates')
hancock = factories.Pirate(name='Boa Hancock', bounty=None, crew=crew)
marigold = factories.Pirate(name='Boa Marigold',
bounty=None,
crew=crew)
response = self.client.get(reverse('bounties:index'),
data={'pirate-search-field': 'Boa'})
self.assertContains(response, crew.name.upper())
def test_shows_wanted_status_of_matching_pirates(self):
crew = factories.Crew()
luffy = factories.Pirate(name='Test Luffy',
wanted_status=Pirate.DEAD_OR_ALIVE)
sanji = factories.Pirate(name='Test Sanji',
bounty=177000000,
crew=crew,
wanted_status=Pirate.ONLY_ALIVE)
response = self.client.get(reverse('bounties:index'),
data={'pirate-search-field': 'Test'})
self.assertContains(response,
luffy.get_wanted_status_display().upper())
self.assertContains(response,
sanji.get_wanted_status_display().upper())
def test_save_and_retrieve_wanted_status(self):
crew = factories.Crew()
pirate = factories.Pirate(crew=crew,
wanted_status=Pirate.DEAD_OR_ALIVE)
result = Pirate.objects.first()
self.assertEqual(pirate.wanted_status, result.wanted_status)
def test_save_and_retrieve_pirate_with_crew(self):
crew = factories.Crew()
pirate = factories.Pirate(crew=crew)
result = Pirate.objects.first()
self.assertEqual(result.crew, crew)
def test_model_save_and_retrieve_crew(self):
crew = factories.Crew()
result = Crew.objects.first()
self.assertEqual(result, crew)