def test_get_areas(self):
update_area(self.area1)
areas = get_areas(self.waypoint1, 'en')
self.assertEqual(len(areas), 1)
self.assertEqual(
areas[0].document_id, self.area1.document_id)
def test_get_areas_forwarded(self):
"""Tests that forwarded areas are not included.
"""
# add an association with wp1
self.session.add(
AreaAssociation(document_id=self.waypoint1.document_id,
area_id=self.area2.document_id))
self.session.flush()
areas = get_areas(self.waypoint1, 'en')
self.assertEqual(len(areas), 0)
def test_get_areas_forwarded(self):
"""Tests that forwarded areas are not included.
"""
# add an association with wp1
self.session.add(AreaAssociation(
document_id=self.waypoint1.document_id,
area_id=self.area2.document_id))
self.session.flush()
areas = get_areas(self.waypoint1, 'en')
self.assertEqual(len(areas), 0)
def _set_areas(self, document, lang):
areas = get_areas(document, lang)
document.areas = [to_json_dict(m, schema_listing_area) for m in areas]
def test_get_areas(self):
update_area(self.area1)
areas = get_areas(self.waypoint1, 'en')
self.assertEqual(len(areas), 1)
self.assertEqual(areas[0].document_id, self.area1.document_id)
def _set_areas(self, document, lang):
areas = get_areas(document, lang)
document.areas = [
to_json_dict(m, schema_listing_area) for m in areas
]
