"""Test the race views.""" from unittest import mock from django.test import TestCase from django.utils.timezone import now from races.models import Race, Racer, RaceUpdate class RaceViewTest(TestCase): """Test races views.""" def setUp(self): """Do pre-test stuff.""" self.client = self.client_class() def test_race_display(self): """Test the display of race info when it's been somewhat populated.""" race = Race.objects.create(key='test') racer_a = Racer.objects.create(nick='hank', race=race) racer_b = Racer.objects.create(nick='bob', race=race) RaceUpdate.objects.create(race=race, racer=racer_a, update="test 1") RaceUpdate.objects.create(race=race, racer=racer_b, update="test 2") RaceUpdate.objects.create(race=race, racer=racer_a, update="test 3") resp = self.client.get('/races/race/test/') self.assertIn(b'hank — test 1', resp.content) self.assertIn(b'bob — test 2', resp.content) self.assertIn(b'hank — test 3', resp.content)