1 files changed, 34 insertions, 1 deletions
diff --git a/chimere/tests.py b/chimere/tests.py
index ad54532..fde2702 100644
--- a/chimere/tests.py
+++ b/chimere/tests.py
@@ -4,9 +4,12 @@
 
 import datetime
 import lxml.etree
+from io import StringIO
 import os
 from shutil import copyfile
 import json
+import sys
+
 test_path = os.path.abspath(__file__)
 test_dir_path = os.path.dirname(test_path) + os.sep
 
@@ -15,6 +18,7 @@ from django.contrib.gis.geos import GEOSGeometry
 from django.conf import settings
 from django.core import mail
 from django.core.files import File
+from django.core.management import call_command
 from django.core.urlresolvers import reverse
 from django.template import Context
 from django.test import TestCase
@@ -404,7 +408,6 @@ class MainUITest(TestCase):
         self.assertEqual(200, response.status_code)
 
 
-
 class AdministratorsTest(TestCase):
     def setUp(self):
         self.areas = areas_setup()
@@ -693,3 +696,33 @@ class RapprochementTest(TestCase):
 class RouteTest(TestCase):
     def setUp(self):
         self.subcategories = subcategory_setup()
+
+
+class V3ToV2Test(TestCase):
+    def setUp(self):
+        self.areas = areas_setup()
+        self.markers = marker_setup()
+        self.routes = route_setup()
+
+    def test_markers(self):
+        sysout = sys.stdout
+        sys.stdout = StringIO()
+        call_command('v3_to_v2_markers')
+        vals = sys.stdout.getvalue()
+        markers = json.loads(vals)
+        self.assertEqual(len(markers), 3)
+        self.assertNotIn('weight', markers[0]['fields'])
+        self.assertNotIn('normalised_weight', markers[0]['fields'])
+        sys.stdout = sysout
+
+    def test_routes(self):
+        sysout = sys.stdout
+        sys.stdout = StringIO()
+        call_command('v3_to_v2_routes')
+        vals = sys.stdout.getvalue()
+        routes = json.loads(vals)
+        self.assertEqual(len(routes), 2)
+        self.assertNotIn('weight', routes[0]['fields'])
+        self.assertNotIn('normalised_weight', routes[0]['fields'])
+        self.assertNotIn('color', routes[0]['fields'])
+        sys.stdout = sysout