def test_vaccate_slot(self):
# Arrange
lh = LotHandler(2)
firstCar = Car('KA-01-HH-1234', 'White')
secondCar = Car('KA-01-BB-0001', 'Red')
lh.FillSlot(firstCar, 1)
lh.FillSlot(secondCar, 2)
slotNumToVaccate = 1
lh.VaccateSlot(slotNumToVaccate)
self.assertTrue(slotNumToVaccate in lh.freeSlots)
self.assertTrue(2 not in lh.freeSlots)
def test_vaccate_slot_not_present(self):
# Arrange
lh = LotHandler(2)
firstCar = Car('KA-01-HH-1234', 'White')
secondCar = Car('KA-01-BB-0001', 'Red')
lh.FillSlot(firstCar, 1)
lh.FillSlot(secondCar, 2)
slotNumToVaccate = 3
with self.assertRaises(ValueError) as ex:
lh.VaccateSlot(slotNumToVaccate)
self.assertEqual(ex.exception,
'No such Slot Number found in Parking lots')
def test_get_nearest_slot(self):
# Arrange
lh = LotHandler(4)
car1 = Car('KA-01-HH-1234', 'White')
car2 = Car('KA-04-HH-1231', 'Blue')
car3 = Car('KA-04-HH-1231', 'Red')
car4 = Car('KA-04-HH-1231', 'Blue')
lh.FillSlot(car1, 1)
lh.FillSlot(car2, 2)
lh.FillSlot(car3, 3)
lh.FillSlot(car4, 4)
lh.VaccateSlot(2)
lh.VaccateSlot(4)
# Act
val = lh.GetNearestSlot()
# Assert
self.assertEqual(val, 2)