def test_get_hotel(self):
# --<Incorrect Inputs>--
self.assertRaises(TypeError, Hotel.get_hotels_in_city, ["Delhi"])
self.assertRaises(ValueError, Hotel.get_hotels_in_city, "Cairo" * 11)
self.assertRaises(TypeError, Hotel.get_hotel, "Delhi", 17)
self.assertRaises(ValueError, Hotel.get_hotel, "Madagascan",
"Banana House" * 9)
# --<Returned Values>--
self.assertIs(Hotel.get_hotel(self.yolo.city, self.yolo.name),
self.yolo,
msg="YOLO isn't what we thought!")
self.assertEqual(Hotel.get_hotel("Hesitance", self.yolo.name), {},
msg="Hesitance and YOLO don't go together!")
self.assertEqual(Hotel.get_hotels_in_city("Mars"), {},
msg="Elon Musk hasn't made it to Mars yet!")
self.assertIsInstance(Hotel.get_hotels_in_city(self.tcu.city),
dict,
msg="The hotels aren't stored in a "
"dictionary!")
def test_delete_hotel(self):
# --<Returned Values>--
self.assertIsNone(self.yolo.delete_hotel(),
msg="Deleting a hotel shouldn't return anything!")
# --<Incorrect Inputs>--
self.assertRaises(ValueError, self.yolo.delete_hotel)
self.assertRaises(ValueError, self.tcu_.delete_hotel)
# --<Effect on the Database>--
self.assertNotIn(self.yolo.name,
Hotel.get_hotels_in_city(self.yolo.city),
msg="YOLO was deleted so it "
"shouldn't exist!")
self.assertIn(self.tcu.name,
Hotel.get_hotels_in_city(self.tcu.city),
msg="Where did \"The Crack up\" go?!")
self.assertIs(Hotel.get_hotel(self.tcu.city, self.tcu.name),
self.tcu,
msg="Who changed The Crack up?!")