more tests

Author: dethi

tests/test_commands.py

@@ -0,0 +1,61 @@
+from django.test import TestCase
+from django.core.management import call_command
+
+from django.contrib.algoliasearch import AlgoliaIndex
+from django.contrib.algoliasearch import register
+from django.contrib.algoliasearch import unregister
+
+try:
+    from StringIO import StringIO
+except:
+    from io import StringIO
+
+from .models import Example
+
+
+class CommandsTestCase(TestCase):
+    def setUp(self):
+        class ExampleIndex(AlgoliaIndex):
+            field = 'name'
+
+        register(Example, ExampleIndex)
+        self.output = StringIO()
+
+    def tearDown(self):
+        unregister(Example)
+
+    def test_reindex(self):
+        call_command('algolia_reindex', stdout=self.output)
+        self.output.seek(0)
+        result = self.output.readlines()
+        self.assertEqual(len(result), 2)
+
+        regex = r'Example --> \d+'
+        try:
+            self.assertRegex(result[1], regex)
+        except AttributeError:
+            self.assertRegexpMatches(result[1], regex)
+
+    def test_clearindex(self):
+        call_command('algolia_clearindex', stdout=self.output)
+        self.output.seek(0)
+        result = self.output.readlines()
+        self.assertEqual(len(result), 2)
+
+        regex = r'Example'
+        try:
+            self.assertRegex(result[1], regex)
+        except AttributeError:
+            self.assertRegexpMatches(result[1], regex)
+
+    def test_applysettings(self):
+        call_command('algolia_applysettings', stdout=self.output)
+        self.output.seek(0)
+        result = self.output.readlines()
+        self.assertEqual(len(result), 2)
+
+        regex = r'Example'
+        try:
+            self.assertRegex(result[1], regex)
+        except AttributeError:
+            self.assertRegexpMatches(result[1], regex)

tests/test_index.py

@@ -23,7 +23,7 @@ def test_default_index_name(self):
         regex = r'django(\d+.\d+)?_Example_test'
         try:
             self.assertRegex(index.index_name, regex)
-        except:
+        except AttributeError:
             self.assertRegexpMatches(index.index_name, regex)
 
     def test_custom_index_name(self):
@@ -34,7 +34,7 @@ class ExampleIndex(AlgoliaIndex):
         regex = r'django(\d+.\d+)?_customName_test'
         try:
             self.assertRegex(index.index_name, regex)
-        except:
+        except AttributeError:
             self.assertRegexpMatches(index.index_name, regex)
 
     def test_custom_objectID(self):