def test_doesnt_release_if_bike_is_broken(self):
d_station = DockingStation()
with mock.patch('bike.Bike') as mocked_bike:
mocked_bike = False
d_station.dock(mocked_bike)
with self.assertRaises(Exception) as context:
d_station.release(mocked_bike)
self.assertFalse("Bike is broken." in str(context.exception))
self.assertEqual(d_station.storage, [mocked_bike])
def test_storage(self):
d_station = DockingStation()
self.assertEqual(d_station.storage, [])
def test_feature_docking_with_real_bike(self):
d_station = DockingStation()
bike = Bike()
self.assertEqual(d_station.dock(bike), "Bike docked.")
self.assertEqual(d_station.storage, [bike])
def test_raises_error_if_no_bike_is_in_docking_station(self):
d_station = DockingStation()
with mock.patch('bike.Bike') as mocked_bike:
with self.assertRaises(Exception) as context:
d_station.release(mocked_bike)
self.assertTrue("No bike to release." in str(context.exception))
def test_be_able_to_take_a_bike(self):
d_station = DockingStation()
with mock.patch('bike.Bike') as mocked_bike:
# print(mocked_bike)
self.assertEqual(d_station.dock(mocked_bike), "Bike docked.")
self.assertEqual(d_station.storage, [mocked_bike])
def test_dock_then_release_a_bike(self):
d_station = DockingStation()
with mock.patch('bike.Bike') as mocked_bike:
d_station.dock(mocked_bike)
self.assertEqual(d_station.release(mocked_bike), "Bike released.")
self.assertEqual(d_station.storage, [])
def test_rack(self):
self.assertIsInstance(DockingStation().rack, list)
def test_dock_bike(self):
bike = 'bike'
station = DockingStation()
station.dock_bike(bike)
self.assertTrue(station.rack[0] == bike)
def test_no_bike_released_if_rack_empty(self):
station = DockingStation()
with self.assertRaises(Exception) as e:
station.release_bike()
self.assertEqual(e.exception.message, 'Bike rack is empty!')
def test_release_bike(self):
station = DockingStation()
station.dock_bike('Bike')
self.assertEqual(station.release_bike(), 'Bike')
def test_setting_chosen_capacity(self):
station = DockingStation(15)
self.assertEqual(station.capacity, 15)
def test_default_capacity_set(self):
self.assertEqual(DockingStation().capacity, 20)
def test_default_capacity_constant(self):
self.assertEqual(DockingStation().DEFAULT_CAPACITY, 20)