def setUp(self):
damaged_bike = Bike()
damaged_bike.damage()
self.bike_container = BikeContainer(20)
self.other_container = BikeContainer(20)
self.bike = Bike()
self.damaged_bike = damaged_bike
class BikeTestCase(unittest.TestCase):
def setUp(self):
self.bike = Bike()
def test_default_bike_broken_state(self):
self.assertEqual(self.bike.is_damaged, False)
def test_can_be_damaged(self):
self.bike.damage()
self.assertEqual(self.bike.is_damaged, True)
def test_can_be_fixed(self):
self.bike.damage()
self.bike.fix()
self.assertEqual(self.bike.is_damaged, False)
def add_bikes_to(holder, count, damaged_bikes = True):
for num in range(0, count):
bike = Bike()
if damaged_bikes == False:
bike.damage()
holder.dock(bike)
