class PlacesTest(TestCase):
def setUp(self):
self.route = Route(name="Route 101")
self.place1 = Place(name="Villa Venezuela")
self.place2 = Place(name="Villa Libertad")
self.route.save()
self.place1.save()
self.place2.save()
self.route.places.add(self.place1, self.place2)
self.route.save()
def test_places_can_get_route(self):
"""docstring for test_places_can_get_route"""
place1 = Place.objects.get(name="Villa Venezuela")
place2 = Place.objects.get(name="Villa Libertad")
self.assertEqual(place1.routes.all()[0], self.route)
self.assertEqual(place2.routes.all()[0], self.route)
def test_add_route_from_place(self):
self.place1.add_route("Ruta 112")
self.place1.save()
self.assertEqual(self.place1.routes.count(), 2)
def tearDown(self):
Place.objects.all().delete()
Route.objects.all().delete()
def create_route(request):
if request.method == 'POST':
route_id = str(uuid.uuid4())
route = Route(route_id=route_id, creation_date=datetime.date.today())
route.save()
return JsonResponse({'route_id': route.route_id}, status=201)
def add_route(self, name):
"""Adds a route from the model"""
from route.models import Route
if not Route.objects.filter(name=name).exists():
route = Route(name=name)
route.save()
self.routes.add(route)
self.save()
def test_get_route_length_successful(self):
route = Route(route_id='d214623e-94da-4789-9c4d-20d1f90aa908',
creation_date=datetime.date.today(),
length=Decimal(20.25))
route.save()
response = self.client.get(
'/route/d214623e-94da-4789-9c4d-20d1f90aa908/length/')
response_details = response.json()
assert 200 == response.status_code
assert 20.25 == response_details['km']
def test_add_way_point_successful(self):
route = Route(route_id='d214623e-94da-4789-9c4d-20d1f90aa907',
creation_date=datetime.date.today())
route.save()
response = self.client.post(
'/route/d214623e-94da-4789-9c4d-20d1f90aa907/way_point/',
json.dumps({
'lat': 25.25,
'lon': 23.23
}),
content_type="application/json")
assert 201 == response.status_code
class RoutesTest(TestCase):
def setUp(self):
self.route = Route(name="Route 101")
self.place1 = Place(name="Villa Venezuela")
self.place2 = Place(name="Villa Libertad")
self.route.save()
self.place1.save()
self.place2.save()
self.route.places.add(self.place1, self.place2)
self.route.save()
def test_routes_adds_places(self):
"""
Creating a route and adding a place to it
"""
# Getting the object from the database
route = Route.objects.get(name="Route 101")
self.assertEqual(route.places.count(), 2)
def test_add_place_from_route(self):
"""
Test the method add_place
"""
self.route.add_place("14 De Septiembre")
self.route.save()
self.assertEqual(self.route.places.count(), 3)
def tearDown(self):
Place.objects.all().delete()
Route.objects.all().delete()
