def setUp(self):
self.journey = Journey()
self.mock_station = Mock()
def setUp(self):
self.journey = Journey()
self.mock_station = Mock()
class TestJourney(unittest.TestCase):
def setUp(self):
self.journey = Journey()
self.mock_station = Mock()
def test_current_journey_should_have_no_entry_or_exit_before_start(self):
self.assertEqual(self.journey.current_journey, {'entry_station': None, 'exit_station': None})
def test_log_entry_logs_entry_station_to_current_journey(self):
self.journey.log_entry(self.mock_station)
self.assertEqual(self.journey.current_journey, {'entry_station': self.mock_station, 'exit_station': None})
def test_log_exit_logs_exit_station_to_current_journey(self):
self.journey.log_exit(self.mock_station)
self.assertEqual(self.journey.current_journey, {'entry_station': None, 'exit_station': self.mock_station})
def test_isin_journey_is_false_on_initialization(self):
self.assertFalse(self.journey.isin_journey)
def test_isin_journey_is_true_after_log_entry(self):
self.journey.log_entry(self.mock_station)
self.assertTrue(self.journey.isin_journey)
def test_isin_journey_is_false_after_full_journey(self):
self.journey.log_entry(self.mock_station)
self.journey.log_exit(self.mock_station)
self.assertFalse(self.journey.isin_journey)
def test_fare_returns_1_if_journey_is_complete(self):
self.journey.log_entry(self.mock_station)
self.journey.log_exit(self.mock_station)
self.assertEqual(self.journey.fare(), 1)
def test_fare_returns_6_if_journey_is_incomplete(self):
self.journey.log_entry(self.mock_station)
self.assertEqual(self.journey.fare(), 6)
class TestJourney(unittest.TestCase):
def setUp(self):
self.journey = Journey()
self.mock_station = Mock()
def test_current_journey_should_have_no_entry_or_exit_before_start(self):
self.assertEqual(self.journey.current_journey, {
'entry_station': None,
'exit_station': None
})
def test_log_entry_logs_entry_station_to_current_journey(self):
self.journey.log_entry(self.mock_station)
self.assertEqual(self.journey.current_journey, {
'entry_station': self.mock_station,
'exit_station': None
})
def test_log_exit_logs_exit_station_to_current_journey(self):
self.journey.log_exit(self.mock_station)
self.assertEqual(self.journey.current_journey, {
'entry_station': None,
'exit_station': self.mock_station
})
def test_isin_journey_is_false_on_initialization(self):
self.assertFalse(self.journey.isin_journey)
def test_isin_journey_is_true_after_log_entry(self):
self.journey.log_entry(self.mock_station)
self.assertTrue(self.journey.isin_journey)
def test_isin_journey_is_false_after_full_journey(self):
self.journey.log_entry(self.mock_station)
self.journey.log_exit(self.mock_station)
self.assertFalse(self.journey.isin_journey)
def test_fare_returns_1_if_journey_is_complete(self):
self.journey.log_entry(self.mock_station)
self.journey.log_exit(self.mock_station)
self.assertEqual(self.journey.fare(), 1)
def test_fare_returns_6_if_journey_is_incomplete(self):
self.journey.log_entry(self.mock_station)
self.assertEqual(self.journey.fare(), 6)
