Progress bar + thumbnail generator

spec/Indexer_spec.py

@@ -8,9 +8,14 @@ class IndexerTest(TestCase):
 
         docs = [{"name": "doc1"}, {"name": "doc2"}]
 
-        result = Indexer.create_bulk_index_string(docs, "indexName")
+        result = Indexer.create_bulk_index_string(docs, 1)
+
+        self.assertTrue(result == '{"index":{}}\n'
+                                  '{"directory": 1, "name": "doc1"}\n'
+                                  '{"index":{}}\n'
+                                  '{"directory": 1, "name": "doc2"}\n'
+                        or result == '{"index":{}}\n'
+                                  '{"name": "doc1", "directory": 1}\n'
+                                  '{"index":{}}\n'
+                                  '{"name": "doc2", "directory": 1}\n')
 
-        self.assertEqual(result, '{"index":{"_index":"indexName","_type":"file"}}\n'
-                                 '{"name": "doc1"}\n'
-                                 '{"index":{"_index":"indexName","_type":"file"}}\n'
-                                 '{"name": "doc2"}\n')

spec/ThumbnailGenerator_spec.py

@@ -0,0 +1,24 @@
+from unittest import TestCase
+from thumbnail import ThumbnailGenerator
+from PIL import Image
+import os
+
+
+class ThumbnailGeneratorTest(TestCase):
+
+    def test_generate(self):
+
+        generator = ThumbnailGenerator(300)
+        # Original image is 420x315
+        generator.generate("test_folder/sample_1.jpg", "test_thumb1.jpg")
+
+        img = Image.open("test_thumb1.jpg")
+        width, height = img.size
+        img.close()
+
+        self.assertEqual(300, width)
+        self.assertEqual(225, height)
+
+        if os.path.isfile("test_thumb1.jpg"):
+            os.remove("test_thumb1.jpg")
+