diff --git a/tracking/test_views.py b/tracking/test_views.py index d5aaf93..9c59600 100644 --- a/tracking/test_views.py +++ b/tracking/test_views.py @@ -1,73 +1,61 @@ +import random from datetime import timedelta + from django.contrib.auth.models import User from django.contrib.gis.geos import Point from django.test import TestCase from django.urls import reverse from django.utils import timezone from mixer.backend.django import mixer -import random from tracking.models import Device, LoggedPoint class ViewTestCase(TestCase): - def setUp(self): point = Point(random.uniform(32.0, 34.0), random.uniform(-115.0, -116.0)) self.device = mixer.blend(Device, seen=timezone.now(), point=point) - # Generate a short tracking history. mixer.blend(LoggedPoint, device=self.device, seen=self.device.seen, point=point) - point.x = point.x + random.uniform(-0.01, 0.01) - point.y = point.y + random.uniform(-0.01, 0.01) - mixer.blend(LoggedPoint, device=self.device, seen=self.device.seen - timedelta(minutes=3), point=point) - point.x = point.x + random.uniform(-0.01, 0.01) - point.y = point.y + random.uniform(-0.01, 0.01) - mixer.blend(LoggedPoint, device=self.device, seen=self.device.seen - timedelta(minutes=6), point=point) - point.x = point.x + random.uniform(-0.01, 0.01) - point.y = point.y + random.uniform(-0.01, 0.01) - mixer.blend(LoggedPoint, device=self.device, seen=self.device.seen - timedelta(minutes=9), point=point) - + # Generate a short tracking history. + for i in range(1, 5): + point.x = point.x + random.uniform(-0.01, 0.01) + point.y = point.y + random.uniform(-0.01, 0.01) + mixer.blend(LoggedPoint, device=self.device, seen=self.device.seen - timedelta(minutes=i), point=point) # Login self.client.force_login(User.objects.create(username="testuser")) def test_device_csv_download(self): - """Test the devices.csv download view - """ + """Test the devices.csv download view""" url = reverse("device_csv") response = self.client.get(url) self.assertEqual(response.status_code, 200) def test_device_geojson_download(self): - """Test the devices.geojson download view - """ + """Test the devices.geojson download view""" url = reverse("device_geojson") response = self.client.get(url) self.assertEqual(response.status_code, 200) def test_device_history_geojson_view(self): - """Test the device history GeoJSON view - """ + """Test the device history GeoJSON view""" url = reverse("device_history_geojson", kwargs={"device_id": self.device.pk}) response = self.client.get(url) self.assertEqual(response.status_code, 200) def test_device_history_csv_view(self): - """Test the device history CSV view - """ + """Test the device history CSV view""" url = reverse("device_history_csv", kwargs={"device_id": self.device.pk}) response = self.client.get(url) self.assertEqual(response.status_code, 200) def test_device_route_geojson_view(self): - """Test the device route GeoJSON view - """ + """Test the device route GeoJSON view""" url = reverse("device_route_geojson", kwargs={"device_id": self.device.pk}) response = self.client.get(url) self.assertEqual(response.status_code, 200) def test_resource_map_view(self): - """Test the resource map view - """ + """Test the resource map view""" url = reverse("resource_map") response = self.client.get(url) self.assertEqual(response.status_code, 200)