diff --git a/players/tests.py b/players/tests.py index df6dae8..1aaac93 100644 --- a/players/tests.py +++ b/players/tests.py @@ -26,6 +26,19 @@ class TestPlayerViews(TestCase): self.assertEqual(response.status_code, 200) def test_player_edit(self): - response = self.client.get(reverse('edit player', args=[2])) + response = self.client.get(reverse('edit player', args=[1])) self.assertEqual(response.status_code, 200) + # create new player + response = self.client.post(reverse('edit player', args=[0]), data={'name': "A new player"}) + self.assertEqual(201, response.status_code) + new_player = Player.objects.get(id=response.context['id']) + self.assertEqual('A new player', new_player.name) + + # modify player + response = self.client.post(reverse('edit player', args=[1]), data={'name': "A different player name"}) + self.assertEqual(200, response.status_code) + self.assertEqual(1, response.context['id']) + modified_player = Player.objects.get(id=response.context['id']) + self.assertEqual('A different player name', modified_player.name) + diff --git a/venues/tests.py b/venues/tests.py index 365d675..8aa4646 100644 --- a/venues/tests.py +++ b/venues/tests.py @@ -21,6 +21,9 @@ class TestVenueViews(TestCase): def test_venue_list(self): response = self.client.get(reverse('venues list')) self.assertEqual(response.status_code, 200) + self.assertIn((1, 'Chesapeake Racetrack and Ballpark'), response.context['items']) + self.assertIn((2, 'Tokyo Fitness Center'), response.context['items']) + self.assertIn((25, 'ILB Historical Preservation Site'), response.context['items']) def test_venue_edit(self): response = self.client.get(reverse('edit venue', args=[1]))