Add property to v3 to v2 marker export

1 files changed, 20 insertions, 2 deletions

diff --git a/chimere/tests.py b/chimere/tests.py
index fde2702..c16f87b 100644
--- a/chimere/tests.py
+++ b/chimere/tests.py
@@ -24,7 +24,7 @@ from django.template import Context
 from django.test import TestCase
 
 from chimere.models import Area, Icon, Importer, Category, SubCategory, \
-    Marker, Route, News, SubCategoryUserLimit, Polygon
+    Marker, Route, News, SubCategoryUserLimit, Polygon, PropertyModel
 from chimere.forms import MarkerForm, AreaAdminForm, notifySubmission
 from chimere.templatetags.chimere_tags import display_news
 from chimere.utils import ShapefileManager
@@ -46,6 +46,19 @@ def areas_setup():
     return [area_1, area_2, area_3]
 
 
+def properties_setup():
+    properties = []
+    properties.append(
+        PropertyModel.objects.create(name="Property 1", slug='prop-1',
+                                     order=10, available=True, mandatory=False)
+    )
+    properties.append(
+        PropertyModel.objects.create(name="Property 2", slug='prop-2',
+                                     order=20, available=True, mandatory=False)
+    )
+    return properties
+
+
 def subcategory_setup():
     category = Category.objects.create(
         name='Main category', available=True, order=1, description='')
@@ -703,6 +716,8 @@ class V3ToV2Test(TestCase):
         self.areas = areas_setup()
         self.markers = marker_setup()
         self.routes = route_setup()
+        properties_setup()
+        self.markers[0].setProperty('prop-1', 'myproperty')
 
     def test_markers(self):
         sysout = sys.stdout
@@ -710,9 +725,12 @@ class V3ToV2Test(TestCase):
         call_command('v3_to_v2_markers')
         vals = sys.stdout.getvalue()
         markers = json.loads(vals)
-        self.assertEqual(len(markers), 3)
+        self.assertEqual(len(markers), 4)
         self.assertNotIn('weight', markers[0]['fields'])
         self.assertNotIn('normalised_weight', markers[0]['fields'])
+        self.assertEqual(markers[-1]['fields']['value'], 'myproperty')
+        self.assertNotIn('polygon', markers[-1]['fields'])
+        self.assertNotIn('route', markers[-1]['fields'])
         sys.stdout = sysout
 
     def test_routes(self):