def test_price_threshold(self):
f1 = FlightSearchBuilder().price_threshold(100)
f2 = FlightSearch.price_threshold(100)
f3 = FlightSearchBuilder(price_threshold=100)
self.assertEqual(f1._price_threshold, f2._price_threshold)
self.assertEqual(f2._price_threshold, f3._price_threshold)
def test_start_from(self):
f1 = FlightSearchBuilder().start_from(datetime(2020,1,1))
f2 = FlightSearch.start_from(datetime(2020,1,1))
f3 = FlightSearchBuilder(start_from=datetime(2020,1,1))
self.assertEqual(f1._start_from, f2._start_from)
self.assertEqual(f2._start_from, f3._start_from)
def test_from_place(self):
f1 = FlightSearchBuilder().from_place("SFO")
f2 = FlightSearch.from_place("SFO")
f3 = FlightSearchBuilder(from_location="SFO")
self.assertEqual(f1._from_location, f2._from_location)
self.assertEqual(f2._from_location, f3._from_location)
def test_return_arrival_time(self):
f1 = FlightSearchBuilder().return_arrival_time((time(0,0,1), time(0,0,2)))
f2 = FlightSearch.return_arrival_time((time(0,0,1), time(0,0,2)))
f3 = FlightSearchBuilder(return_arrival_time=(time(0,0,1), time(0,0,2)))
self.assertEqual(f1._departure_date, f2._departure_date)
self.assertEqual(f2._departure_date, f3._departure_date)
def test_return_departure_date(self):
f1 = FlightSearchBuilder().return_departure_date(datetime(2020,1,1))
f2 = FlightSearch.return_departure_date(datetime(2020,1,1))
f3 = FlightSearchBuilder(return_departure_date=datetime(2020,1,1))
self.assertEqual(f1._departure_date, f2._departure_date)
self.assertEqual(f2._departure_date, f3._departure_date)
def test_allow_layovers(self):
f1 = FlightSearchBuilder().allow_layovers(True)
f2 = FlightSearch.allow_layovers(True)
f3 = FlightSearchBuilder(allow_layovers=True)
self.assertEqual(f1._allow_layovers, f2._allow_layovers)
self.assertEqual(f2._allow_layovers, f3._allow_layovers)
def test_weekend(self):
f1 = FlightSearchBuilder().weekend()
f2 = FlightSearch.weekend()
f3 = FlightSearchBuilder(departure_time=(time(hour=18, minute=0), time(hour=23, minute=59)), return_arrival_time=(time(hour=0, minute=0), time(hour=23, minute=0)), departure_day=Day.FRIDAY, return_arrival_day=Day.SUNDAY)
self.assertEqual(f1._departure_time, f2._departure_time)
self.assertEqual(f2._departure_time, f3._departure_time)
self.assertEqual(f1._return_arrival_time, f2._return_arrival_time)
self.assertEqual(f2._return_arrival_time, f3._return_arrival_time)
self.assertEqual(f1._departure_day, f2._departure_day)
self.assertEqual(f2._departure_day, f3._departure_day)
self.assertEqual(f1._return_arrival_day, f2._return_arrival_day)
self.assertEqual(f2._return_arrival_day, f3._return_arrival_day)