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()