def test_slot_numbers_for_cars_with_colour(self):
# Call test_park_vehicle to re-use code
self.test_park_vehicle()
# Test for white
self.assertEqual(
utils.info_for_vehicles_with_colour("White", "slot_id"),
[1, 2]) # White
self.assertEqual(
utils.info_for_vehicles_with_colour("white", "slot_id"),
[1, 2]) # white
# Unpark 4, 2, and 5
self.assertTrue(utils.unpark_vehicle(4))
self.assertTrue(utils.unpark_vehicle(2))
self.assertTrue(utils.unpark_vehicle(5))
# Park some vehicles
self.assertEqual(utils.park_vehicle('KA-01-HH-7777', 'Red'), 2)
self.assertEqual(utils.park_vehicle('KA-01-HH-1235', 'white'),
4) # white
# Test change in white
self.assertEqual(
utils.info_for_vehicles_with_colour("White", "slot_id"),
[1, 4]) # White
def test_registration_numbers_for_cars_with_colour(self):
# Call test_park_vehicle to re-use code
self.test_park_vehicle()
# Test for white
self.assertEqual(
utils.info_for_vehicles_with_colour("White",
"registration_number"),
["KA-01-HH-1234", "KA-01-HH-9999"]) # White
self.assertEqual(
utils.info_for_vehicles_with_colour("white",
"registration_number"),
["KA-01-HH-1234", "KA-01-HH-9999"]) # white
# Unpark 4, 2, and 5
self.assertTrue(utils.unpark_vehicle(4))
self.assertTrue(utils.unpark_vehicle(2))
self.assertTrue(utils.unpark_vehicle(5))
# Park one vehicle
self.assertEqual(utils.park_vehicle('KA-01-HH-1235', 'white'),
2) # white
# Test change in white
self.assertEqual(
utils.info_for_vehicles_with_colour("White",
"registration_number"),
["KA-01-HH-1234", "KA-01-HH-1235"]) # White
def test_unpark_vehicle(self):
# Call test_park_vehicle to re-use code
self.test_park_vehicle()
# Unpark 4, 2, and 5
self.assertTrue(utils.unpark_vehicle(4))
self.assertTrue(utils.unpark_vehicle(2))
self.assertTrue(utils.unpark_vehicle(5))
# Test the count after unparking
self.assertEqual(
models.Parking.query.filter(
models.Parking.active.is_(True)).count(), 3)
# Unpark the same slot
self.assertTrue(utils.unpark_vehicle(4))
# Test the count after unparking
self.assertEqual(
models.Parking.query.filter(
models.Parking.active.is_(True)).count(), 3)
# Unpark 100
self.assertFalse(utils.unpark_vehicle(100))
# Test the count after unparking
self.assertEqual(
models.Parking.query.filter(
models.Parking.active.is_(True)).count(), 3)
def test_park_vehicle_first_empty(self):
# Call test_park_vehicle to re-use code
self.test_park_vehicle()
# Unpark 4, 2, and 5
self.assertTrue(utils.unpark_vehicle(4))
self.assertTrue(utils.unpark_vehicle(2))
self.assertTrue(utils.unpark_vehicle(5))
# Test the count after unparking
self.assertEqual(
models.Parking.query.filter(
models.Parking.active.is_(True)).count(), 3)
# Park the vehicle
self.assertEqual(utils.park_vehicle('KA-01-HH-1235', 'White'), 2)
# Test the count after parking
self.assertEqual(
models.Parking.query.filter(
models.Parking.active.is_(True)).count(), 4)
# Complete the parking
self.assertEqual(utils.park_vehicle('KA-01-HH-9999', 'White'), 4)
self.assertEqual(utils.park_vehicle('KA-01-HH-7777', 'Red'), 5)
# Test final count
self.assertEqual(
models.Parking.query.filter(
models.Parking.active.is_(True)).count(), self.total_slots)
def test_slot_number_for_registration_number(self):
# Call test_park_vehicle to re-use code
self.test_park_vehicle()
# Test for KA-01-HH-9999
self.assertEqual(
utils.slot_number_for_registration_number("KA-01-HH-9999"), 2)
# Test for KA-01-HH-7777
self.assertEqual(
utils.slot_number_for_registration_number("KA-01-HH-7777"), 4)
# Unpark 4, 2, and 5
self.assertTrue(utils.unpark_vehicle(4))
self.assertTrue(utils.unpark_vehicle(2))
self.assertTrue(utils.unpark_vehicle(5))
# Park one vehicle
self.assertEqual(utils.park_vehicle('KA-01-HH-1235', 'White'), 2)
# Test for KA-01-HH-1235
self.assertEqual(
utils.slot_number_for_registration_number("KA-01-HH-1235"), 2)
def test_parking_lot_status(self):
# Initial empty parking lot
self.assertEqual(utils.parking_lot_status(), [])
# Call test_park_vehicle to re-use code
self.test_park_vehicle()
# Test full parking lot
self.assertEqual(utils.parking_lot_status(), [
{
"slot_id": 1,
"registration_number": "KA-01-HH-1234",
"colour": "White"
},
{
"slot_id": 2,
"registration_number": "KA-01-HH-9999",
"colour": "White"
},
{
"slot_id": 3,
"registration_number": "KA-01-BB-0001",
"colour": "Black"
},
{
"slot_id": 4,
"registration_number": "KA-01-HH-7777",
"colour": "Red"
},
{
"slot_id": 5,
"registration_number": "KA-01-HH-2701",
"colour": "Blue"
},
{
"slot_id": 6,
"registration_number": "KA-01-HH-3141",
"colour": "Black"
},
])
# Unpark 4, 2, and 5
self.assertTrue(utils.unpark_vehicle(4))
self.assertTrue(utils.unpark_vehicle(2))
self.assertTrue(utils.unpark_vehicle(5))
# Test partial parking lot
self.assertEqual(utils.parking_lot_status(), [
{
"slot_id": 1,
"registration_number": "KA-01-HH-1234",
"colour": "White"
},
{
"slot_id": 3,
"registration_number": "KA-01-BB-0001",
"colour": "Black"
},
{
"slot_id": 6,
"registration_number": "KA-01-HH-3141",
"colour": "Black"
},
])
# Park the vehicle
self.assertEqual(utils.park_vehicle('KA-01-HH-1235', 'White'), 2)
# Test partial parking lot with new vehicle
self.assertEqual(utils.parking_lot_status(), [
{
"slot_id": 1,
"registration_number": "KA-01-HH-1234",
"colour": "White"
},
{
"slot_id": 2,
"registration_number": "KA-01-HH-1235",
"colour": "White"
},
{
"slot_id": 3,
"registration_number": "KA-01-BB-0001",
"colour": "Black"
},
{
"slot_id": 6,
"registration_number": "KA-01-HH-3141",
"colour": "Black"
},
])
# Complete the parking
self.assertEqual(utils.park_vehicle('KA-01-HH-9999', 'White'), 4)
self.assertEqual(utils.park_vehicle('KA-01-HH-7777', 'Red'), 5)
# Test full parking lot
self.assertEqual(utils.parking_lot_status(), [
{
"slot_id": 1,
"registration_number": "KA-01-HH-1234",
"colour": "White"
},
{
"slot_id": 2,
"registration_number": "KA-01-HH-1235",
"colour": "White"
},
{
"slot_id": 3,
"registration_number": "KA-01-BB-0001",
"colour": "Black"
},
{
"slot_id": 4,
"registration_number": "KA-01-HH-9999",
"colour": "White"
},
{
"slot_id": 5,
"registration_number": "KA-01-HH-7777",
"colour": "Red"
},
{
"slot_id": 6,
"registration_number": "KA-01-HH-3141",
"colour": "Black"
},
])