Example #1
0
  def test_order_by_start_date(self):
    with self._app.app_context():
      # Setup.
      trips = []
      # Should be excluded because of start date.
      _add_trip(trips=trips, trip_id=1, title='Tahoe', activity_name='Hike',
                start=datetime.date(2016, 1, 1), end=datetime.date(2016, 1, 10))
      # Should be excluded because of end date.
      _add_trip(trips=trips, trip_id=2, title='Tahoe', activity_name='Hike',
                start=datetime.date(2016, 2, 11), end=datetime.date(2016, 2, 22))
      # Should be included and be second in the results because of start date.
      _add_trip(trips=trips, trip_id=3, title='Tahoe', activity_name='Hike',
                start=datetime.date(2016, 2, 14), end=datetime.date(2016, 2, 16))
      # Should be included and be first in the results because of start date.
      _add_trip(trips=trips, trip_id=4, title='Tahoe', activity_name='Hike',
                start=datetime.date(2016, 2, 12), end=datetime.date(2016, 2, 18))
      _insert_trips(trips)

      # Exercise.
      (count, trips) = trips_models.search(from_date=datetime.date(2016, 2, 10),
                                           to_date=datetime.date(2016, 2, 20))

      # Verify.
      self.assertEqual(2, count)
      self.assertEqual(2, len(trips))
      self.assertEqual(4, trips[0].trip_id)
      self.assertEqual(3, trips[1].trip_id)
Example #2
0
  def test_find_by_search_text(self):
    with self._app.app_context():
      # Setup.
      trips = []
      _add_trip(trips=trips, trip_id=1, title='Tahoe', activity_name='Hike',
                start=datetime.date(2016, 1, 1), end=datetime.date(2016, 1, 10))
      _insert_trips(trips)

      # Exercise.
      (count, trips) = trips_models.search(search_text='Tahoe')

      # Verify.
      self.assertEqual(1, count)
      self.assertEqual(1, len(trips))
Example #3
0
def search_trips():
  '''Gets a list of trips with the given parameters.'''
  title = request.args.get('title', '')
  from_date = _parse_date(request.args.get('fromDate', ''))
  to_date = _parse_date(request.args.get('toDate', ''))
  page_str = request.args.get('page', '')
  offset = _get_offset(page_str)
  min_price = _get_int(request.args.get('minPrice', ''))
  max_price = _get_int(request.args.get('maxPrice', ''))
  # TODO(Janet): change it to ngrams.corrections(title) when suggestions are better.
  suggested_text = ''
  (count, trips) = trips_models.search(search_text=title, from_date=from_date, to_date=to_date,
                                       min_price=min_price, max_price=max_price,
                                       offset=offset, limit=PAGE_SIZE)
  count_pages = int(math.ceil(count / float(PAGE_SIZE)))


  list_trips = [trip.as_dict() for trip in trips]

  return jsonify(trips=list_trips, count_pages=count_pages, count=count, suggested=suggested_text)
Example #4
0
  def test_find_with_limit(self):
    with self._app.app_context():
      # Setup.
      trips = []
      _add_trip(trips=trips, trip_id=1, title='Tahoe 1', activity_name='Hike',
                start=datetime.date(2016, 1, 1), end=datetime.date(2016, 1, 10))
      _add_trip(trips=trips, trip_id=2, title='Tahoe 2', activity_name='Hike',
                start=datetime.date(2016, 1, 1), end=datetime.date(2016, 1, 10))
      _add_trip(trips=trips, trip_id=3, title='Tahoe 3', activity_name='Hike',
                start=datetime.date(2016, 1, 1), end=datetime.date(2016, 1, 10))
      _add_trip(trips=trips, trip_id=4, title='Tahoe 4', activity_name='Hike',
                start=datetime.date(2016, 1, 1), end=datetime.date(2016, 1, 10))
      _insert_trips(trips)

      # Exercise.
      (count, trips) = trips_models.search(search_text='Tahoe', offset=1, limit=2)

      # Verify.
      self.assertEqual(4, count)
      self.assertEqual(2, len(trips))
      self.assertEqual(2, trips[0].trip_id)
Example #5
0
  def test_find_by_from_to_dates(self):
    with self._app.app_context():
      # Setup.
      trips = []
      _add_trip(trips=trips, trip_id=1, title='Tahoe', activity_name='Hike',
                start=datetime.date(2016, 1, 1), end=datetime.date(2016, 1, 10))

      _add_trip(trips=trips, trip_id=2, title='Tahoe', activity_name='Hike',
                start=datetime.date(2016, 2, 20), end=datetime.date(2016, 3, 29))

      _add_trip(trips=trips, trip_id=3, title='Tahoe', activity_name='Hike',
                start=datetime.date(2016, 2, 21), end=datetime.date(2016, 4, 10))
      _insert_trips(trips)

      # Exercise.
      (count, trips) = trips_models.search(from_date=datetime.date(2016, 2, 10),
                                           to_date=datetime.date(2016, 3, 30))

      # Verify.
      self.assertEqual(1, count)
      self.assertEqual(1, len(trips))
      self.assertEqual(2, trips[0].trip_id)
Example #6
0
  def test_search_by_price(self):
    with self._app.app_context():
      # Setup.
      trips = []
      _add_trip(trips=trips, trip_id=1, title='Tahoe', activity_name='Hike',
                start=datetime.date(2016, 1, 1), end=datetime.date(2016, 1, 10),
                from_price=100)

      _add_trip(trips=trips, trip_id=2, title='Yosemitee', activity_name='Hike',
                start=datetime.date(2016, 2, 20), end=datetime.date(2016, 3, 29),
                from_price=200)

      _add_trip(trips=trips, trip_id=3, title='Kings Canyon', activity_name='Hike',
                start=datetime.date(2016, 2, 21), end=datetime.date(2016, 4, 10),
                from_price=300)
      _insert_trips(trips)

      # Exercise.
      (count, trips) = trips_models.search(min_price=150, max_price=250)

      # Verify.
      self.assertEqual(1, count)
      self.assertEqual(1, len(trips))
      self.assertEqual(2, trips[0].trip_id)