Import: allow use of CSV for imports

1 files changed, 24 insertions, 1 deletions

diff --git a/chimere/tests.py b/chimere/tests.py
index 89c6ee3..f97afab 100644
--- a/chimere/tests.py
+++ b/chimere/tests.py
@@ -19,7 +19,7 @@ from django.test.client import Client
 
 from chimere.admin import managed_modified, MarkerAdmin
 from chimere.models import Icon, Importer, Category, SubCategory, Marker,\
-                           Route, News, Map, MapUsers, MapGroups
+                           Route, News, Map, MapUsers, MapGroups, PropertyModel
 from chimere.forms import MarkerForm, MapAdminForm
 from chimere.templatetags.chimere_tags import display_news
 from chimere.utils import ShapefileManager
@@ -133,6 +133,14 @@ def map_setup():
                 public_propose=False, public_write=False))
     return maps
 
+def propertymodel_setup():
+    pms = []
+    pms.append(PropertyModel.objects.create(name='Address', order=10, available=False,
+                                            mandatory=False, slug='address', type='T'))
+    pms.append(PropertyModel.objects.create(name='City', order=20, available=False,
+                                            mandatory=False, slug='town', type='T'))
+    return pms
+
 def users_setup():
     adminuser = User.objects.create_superuser('admin',
                                               'admin@test.com',
@@ -315,6 +323,21 @@ class GeoRSSImporterTest(TestCase, ImporterTest):
 
         self.marker_importers = [(importer1, 1), (importer2, 32)]
 
+class CSVImporterTest(TestCase, ImporterTest):
+    def setUp(self):
+        subcategories = subcategory_setup()
+        pm = propertymodel_setup()
+        importer1 = Importer.objects.create(importer_type='CSV',
+                         source=test_dir_path+'tests/test.csv')
+        importer1.categories.add(subcategories[0])
+        self.old_settings = settings.CHIMERE_NOMINATIM_FIELDS
+        settings.CHIMERE_NOMINATIM_FIELDS = {'street':'address',
+                              'city':'town', 'country':u'Great Britain'}
+        self.marker_importers = [(importer1, 2)]
+
+    def tearDown(self):
+        settings.CHIMERE_NOMINATIM_FIELDS = self.old_settings
+
 class FeedsTest(TestCase):
     def setUp(self):
         self.areas = areas_setup()