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)')
class TestEndpoints(APITestCase):
"""Test request and response of endpoints."""
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 tearDown(self):
"""Teardown."""
User.objects.all().delete()
Profile.objects.all().delete()
Route.objects.all().delete()
def test_user_signup(self):
"""Test that a user exists once they have signed up."""
post = self.client.post('/users/signup', {'username': '******',
'password': '******',
'email': '*****@*****.**'})
second_get = self.client.get('/users/{}/'.format(post.data['id']))
self.assertContains(second_get, 'jane')
def test_post_user_already_exists(self):
"""Test response when user exists in the database."""
post = self.client.post('/users/signup', {'username': '******',
'password': '******',
'email': '*****@*****.**'})
self.assertEqual(post.data, {'username': [u'A user with that username already exists.']})
def test_get_user(self):
"""Test that a user returned when a get request is performed."""
response = self.client.get('/users/{}/'.format(self.user.id))
print(response.data)
self.assertEqual(response.data['username'], 'foo')
def test_post_user_is_user(self):
"""Test that a user is created when a post request is performed."""
post = self.client.post('/users/signup', {'username': '******',
'password': '******',
'email': '*****@*****.**'})
self.assertEquals(post.data['username'], 'joe')
self.assertEquals(post.data['email'], '*****@*****.**')
def test_get_route(self):
"""Test that you can get a route."""
response = self.client.get('/routes/{}/'.format(self.route.id))
self.assertEqual(response.data['start_point'],
u'SRID=4326;POINT (2.0000000000000000 3.0000000000000000)')
def test_get_route_no_route_exists(self):
"""Test that you can get a route."""
response = self.client.get('/routes/500/')
self.assertEqual(response.data, {u'detail': u'Not found.'})
def test_get_profile(self):
"""Test that you can get a profile."""
response = self.client.get('/profiles/{}/'.format(self.profile.id))
self.assertEqual(response.data['carbrand'], 'Audi')
def test_add_route(self):
"""Test that a route is added."""
post = self.client.post('/routes/add/', {'lat': '2',
'lng': '3'})
second_get = self.client.get('/routes/{}/'.format(post.data['id']))
self.assertContains(second_get, post.data['id'])
def test_query_get_exact_point(self):
"""Test query on same point returns point."""
response = self.client.get('/query/', {'lat': '3',
'lng': '2'})
self.assertEqual(response.data[0]['start_point'], u'SRID=4326;POINT (2.0000000000000000 3.0000000000000000)')
def test_query_get_near_point(self):
"""Test query on nearby coordinates returns nearest."""
response = self.client.get('/query/', {'lat': '3.0000000000000001',
'lng': '2.0000000000000002'})
self.assertEqual(response.data[0]['start_point'], u'SRID=4326;POINT (2.0000000000000000 3.0000000000000000)')
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)')
