def test_process_qpx(self):
#got to figure our how to load initial data
departure_date = datetime.strptime(trip_dates.get('departure_date'), '%Y-%m-%d')
dates = Dates.objects.get(departure_date=departure_date)
self.assertEqual(Slice.objects.count(), 0)
self.assertEqual(HistoricSlice.objects.count(), 0)
process_qpx(flight_data, dates)
self.assertEqual(Slice.objects.count(), 20)
self.assertEqual(HistoricSlice.objects.count(), 0)
process_qpx(flight_data, dates)
self.assertEqual(Slice.objects.count(), 20)
self.assertEqual(HistoricSlice.objects.count(), 20)
self.assertEqual(Slice.objects.count(), HistoricSlice.objects.count())
def test_process_qpx_no_data(self):
departure_date = datetime.strptime(trip_dates.get('departure_date'), '%Y-%m-%d')
dates = Dates.objects.get(departure_date=departure_date)
process_qpx(no_data, dates)
self.assertEqual(Slice.objects.count(), HistoricSlice.objects.count())
def create(self):
departure_date = datetime.strptime(self.data.get('dates').get('departure_date'), '%Y-%m-%d')
return_date = datetime.strptime(self.data.get('dates').get('return_date'), '%Y-%m-%d')
dates = Dates.objects.get(departure_date=departure_date, return_date=return_date)
return process_qpx(self.data.get('qpx_data'), dates)