def setUp(self): """Setup.""" self.client = APIClient() self.profile = Profile() self.user = User.objects.create_user(username='******', password='******', email='*****@*****.**') self.user.save() self.client.force_authenticate(user=self.user) self.profile.user = self.user self.profile.carbrand = 'Audi' self.profile.carseat = 1 self.profile.petsallowed = True self.profile.save() self.route = Route() self.route.user = self.profile self.route.start_point = GEOSGeometry('POINT(2 3)', srid=4326) self.route.save()
def test_query_two_routes_return(self): """Test query with multiple returns.""" profile1 = Profile() user1 = User.objects.create_user(username='******', password='******') user1.save() self.client.force_authenticate(user=user1) profile1.user = user1 profile1.carbrand = 'Audi' profile1.carseat = 1 profile1.petsallowed = True profile1.save() route1 = Route() route1.user = profile1 route1.start_point = GEOSGeometry('POINT(2.0000000000000009 3.0000000000000009)', srid=4326) route1.save() response = self.client.get('/query/', {'lat': '3.0000000000000002', 'lng': '2.0000000000000001'}) self.assertEqual(response.data[0]['start_point'], u'SRID=4326;POINT (2.0000000000000000 3.0000000000000000)') self.assertEqual(response.data[1]['start_point'], u'SRID=4326;POINT (2.0000000000000009 3.0000000000000009)')