def test_do_park(self):
'''Ensures correct messages are printed when do_park is called with correct input'''
pls = ParkingLotShell()
pls.onecmd("create_parking_lot 1")
print_out = get_print_results(pls.onecmd, "park AK-01-HH-1234 White")
self.assertEqual(print_out, "Allocated slot number: 1")
print_out = get_print_results(pls.onecmd, "park AK-01-HH-1234 White")
self.assertEqual(print_out, "Sorry, parking lot is full")
def test_do_park_errors(self):
'''Ensures correct messages are printed when do_park is called with incorrect input'''
pls = ParkingLotShell()
print_out = get_print_results(pls.onecmd, "park AK-01-HH-1234 White")
self.assertEqual(print_out[-15:], ERROR_MSG_1)
pls.onecmd("create_parking_lot 1")
print_out = get_print_results(pls.onecmd, "park")
self.assertEqual(print_out[-15:], ERROR_MSG_2)
print_out = get_print_results(pls.onecmd,
"park AK-01-HH-1234 White White")
self.assertEqual(print_out[-15:], ERROR_MSG_2)
def test_do_leave_errors(self):
'''Ensures correct messages are printed when do_leave is called with incorrect input'''
pls = ParkingLotShell()
print_out = get_print_results(pls.onecmd, "leave 1")
self.assertEqual(print_out[-15:], ERROR_MSG_1)
pls.onecmd("create_parking_lot 1")
print_out = get_print_results(pls.onecmd, "leave")
self.assertEqual(print_out[-15:], ERROR_MSG_2)
print_out = get_print_results(pls.onecmd, "leave fgh")
self.assertEqual(print_out[-15:], ERROR_MSG_2)
print_out = get_print_results(pls.onecmd, "leave 6 6")
self.assertEqual(print_out[-15:], ERROR_MSG_2)
def test_do_leave(self):
'''Ensures correct messages are printed when do_leave is called with correct input'''
pls = ParkingLotShell()
pls.onecmd("create_parking_lot 1")
print_out = get_print_results(pls.onecmd, "leave 1")
self.assertEqual(print_out, "Slot number 1 is free")
print_out = get_print_results(pls.onecmd, "leave 2")
self.assertEqual(print_out,
"That slot doesn't exist in this parking lot")
print_out = get_print_results(pls.onecmd, "leave 0")
self.assertEqual(print_out,
"That slot doesn't exist in this parking lot")
def test_do_registration_numbers_for_cars_with_colour_errors(self):
'''Ensures correct messages are printed when do_registration_numbers_for_cars_with_colour
is called with incorrect input'''
pls = ParkingLotShell()
print_out = get_print_results(
pls.onecmd, "registration_numbers_for_cars_with_colour White")
self.assertEqual(print_out[-15:], ERROR_MSG_1)
def test_do_create_parking_lot_errors(self):
'''Ensures correct messages are printed when do_create_parking_lot is called with incorrect input'''
pls = ParkingLotShell()
print_out = get_print_results(pls.onecmd, "create_parking_lot sdfsdf")
self.assertEqual(print_out[-15:], ERROR_MSG_2)
print_out = get_print_results(pls.onecmd, "create_parking_lot")
self.assertEqual(print_out[-15:], ERROR_MSG_2)
print_out = get_print_results(pls.onecmd, "create_parking_lot 6 6 6")
self.assertEqual(print_out[-15:], ERROR_MSG_2)
def test_do_status(self):
'''Ensures correct messages are printed when do_status is called'''
pls = ParkingLotShell()
pls.onecmd("create_parking_lot 1")
pls.onecmd("park AK-01-HH-1234 White")
print_out = get_print_results(pls.onecmd, "status")
expected_results = "Slot No.\nRegistration No\nColour\n1\nAK-01-HH-1234\nWhite"
self.assertEqual(print_out, expected_results)
def test_do_registration_numbers_for_cars_with_colour(self):
'''Ensures correct messages are printed when do_registration_numbers_for_cars_with_colour
is called with correct input'''
pls = ParkingLotShell()
pls.onecmd("create_parking_lot 2")
pls.onecmd("park AK-01-HH-1234 White")
print_out = get_print_results(
pls.onecmd, "registration_numbers_for_cars_with_colour White")
self.assertEqual(print_out, "AK-01-HH-1234")
pls.onecmd("park AK-01-HH-1234 White")
print_out = get_print_results(
pls.onecmd, "registration_numbers_for_cars_with_colour White")
self.assertEqual(print_out, "AK-01-HH-1234, AK-01-HH-1234")
print_out = get_print_results(
pls.onecmd, "registration_numbers_for_cars_with_colour Blackish")
self.assertEqual(print_out, "Not found")
def test_do_slot_number_for_registration_number(self):
'''Ensures correct messages are printed when do_slot_number_for_registration_number
is called with correct input'''
pls = ParkingLotShell()
pls.onecmd("create_parking_lot 2")
pls.onecmd("park AK-01-HH-1234 White")
print_out = get_print_results(
pls.onecmd, "slot_number_for_registration_number AK-01-HH-1234")
self.assertEqual(print_out, "1")
pls.onecmd("park AK-01-HH-1234 White")
print_out = get_print_results(
pls.onecmd, "slot_number_for_registration_number AK-01-HH-1234")
self.assertEqual(print_out, "1, 2")
print_out = get_print_results(
pls.onecmd, "slot_number_for_registration_number AK-01-HH-12345")
self.assertEqual(print_out, "Not found")
def test_do_create_parking_lot(self):
'''Ensures correct messages are printed when do_create_parking_lot is called with correct input'''
pls = ParkingLotShell()
print_out = get_print_results(pls.onecmd, "create_parking_lot 6")
self.assertEqual(print_out, "Created a parking lot with 6 slots")