add test for tags

Author: dethi

src/models.py

@@ -67,13 +67,13 @@ def __init__(self, model, client):
         if self.custom_objectID:
             if not (hasattr(model, self.custom_objectID) or
                     (self.custom_objectID in all_fields)):
-                raise AlgoliaIndex('{} is not an attribute of {}.'.format(
+                raise AlgoliaIndexError('{} is not an attribute of {}.'.format(
                     self.custom_objectID, model))
 
         # Check tags
         if self.tags:
             if not (hasattr(model, self.tags) or (self.tags in all_fields)):
-                raise AlgoliaIndex('{} is not an attribute of {}'.format(
+                raise AlgoliaIndexError('{} is not an attribute of {}'.format(
                     self.tags, model))
 
         # Check geo_field

tests/models.py

@@ -7,6 +7,7 @@ class Example(models.Model):
     address = models.CharField(max_length=200)
     lat = models.FloatField()
     lng = models.FloatField()
+    category = []
 
     def location(self):
         return (self.lat, self.lng)

tests/test_index.py

@@ -16,6 +16,7 @@ def setUp(self):
                                 address='Finland',
                                 lat=63.3,
                                 lng=-32.0)
+        self.instance.category = ['Shop', 'Grocery']
 
     def test_default_index_name(self):
         index = AlgoliaIndex(Example, self.client)
@@ -36,6 +37,21 @@ class ExampleIndex(AlgoliaIndex):
         except:
             self.assertRegexpMatches(index.index_name, regex)
 
+    def test_custom_objectID(self):
+        class ExampleIndex(AlgoliaIndex):
+            custom_objectID = 'uid'
+
+        index = ExampleIndex(Example, self.client)
+        obj = index._build_object(self.instance)
+        self.assertEqual(obj['objectID'], 4)
+
+    def test_invalid_custom_objectID(self):
+        class ExampleIndex(AlgoliaIndex):
+            custom_objectID = 'uuid'
+
+        with self.assertRaises(AlgoliaIndexError):
+            index = ExampleIndex(Example, self.client)
+
     def test_geo_fields(self):
         class ExampleIndex(AlgoliaIndex):
             geo_field = 'location'
@@ -44,13 +60,27 @@ class ExampleIndex(AlgoliaIndex):
         obj = index._build_object(self.instance)
         self.assertEqual(obj['_geoloc'], {'lat': 63.3, 'lng': -32.0})
 
-    def test_custom_objectID(self):
+    def test_invalid_geo_fields(self):
         class ExampleIndex(AlgoliaIndex):
-            custom_objectID = 'uid'
+            geo_field = 'position'
+
+        with self.assertRaises(AlgoliaIndexError):
+            index = ExampleIndex(Example, self.client)
+
+    def test_tags(self):
+        class ExampleIndex(AlgoliaIndex):
+            tags = 'category'
 
         index = ExampleIndex(Example, self.client)
         obj = index._build_object(self.instance)
-        self.assertEqual(obj['objectID'], 4)
+        self.assertListEqual(obj['_tags'], self.instance.category)
+
+    def test_invalid_tags(self):
+        class ExampleIndex(AlgoliaIndex):
+            tags = 'categories'
+
+        with self.assertRaises(AlgoliaIndexError):
+            index = ExampleIndex(Example, self.client)
 
     def test_one_field(self):
         class ExampleIndex(AlgoliaIndex):
@@ -64,6 +94,7 @@ class ExampleIndex(AlgoliaIndex):
         self.assertNotIn('lat', obj)
         self.assertNotIn('lng', obj)
         self.assertNotIn('location', obj)
+        self.assertNotIn('category', obj)
         self.assertEqual(len(obj), 2)
 
     def test_multiple_fields(self):
@@ -78,6 +109,7 @@ class ExampleIndex(AlgoliaIndex):
         self.assertNotIn('lat', obj)
         self.assertNotIn('lng', obj)
         self.assertNotIn('location', obj)
+        self.assertNotIn('category', obj)
         self.assertEqual(len(obj), 3)
 
     def test_invalid_fields(self):