def test_should_add_event(self):
# Arrange & Act
state = FootballSimulationGameState() \
.event(Event('my event 1')) \
.event(Event('second event'))
# Assert
self.assertEqual(state.get_events()[0].get_description(), 'my event 1')
self.assertEqual(state.get_events()[1].get_description(), 'second event')
def test_should_only_occur_once_a_quarter(self):
# Arrange
state = FootballSimulationGameState() \
.team1(FootballTeam('teama')) \
.team2(FootballTeam('teamb')) \
.quarter(2) \
.time(0)
# Act
state = quarter_banner(state)
state = quarter_banner(state)
# Assert
types = [type(t) for t in state.get_events()]
self.assertEqual(1, types.count(QuarterStartBannerEvent))
def test_get_events_should_be_zero_length_at_constructor(self):
# Arrange
state = FootballSimulationGameState()
# Assert
self.assertEqual(len(state.get_events()), 0)
