adds some tests for the file naming functions

Author: beaufour

setup.py

@@ -19,6 +19,11 @@ def readme():
           'python-dateutil == 1.5',
           'PyYAML==3.11',
       ],
+      test_suite='tests.get_tests',
+      tests_require=[
+          'unittest2==0.8.0',
+          'attrdict==0.5.1',
+      ],
       entry_points={
           'console_scripts': ['flickr_download=flickr_download.flick_download:main'],
       },

tests/__init__.py

@@ -0,0 +1,7 @@
+import os.path
+import unittest2
+
+
+def get_tests():
+    start_dir = os.path.dirname(__file__)
+    return unittest2.TestLoader().discover(start_dir, pattern="*.py")

tests/filename_handlers.py

@@ -0,0 +1,48 @@
+import unittest
+
+from attrdict import AttrDict
+
+from flickr_download.filename_handlers import get_filename_handler
+
+
+class TestFilenameHandlers(unittest.TestCase):
+    def setUp(self):
+        self._pset = AttrDict({'title': 'Some Set', 'id': 999})
+        self._photo = AttrDict({'title': 'Some Photo', 'id': 123})
+        self._suffix = ""
+
+    def test_title(self):
+        fn = get_filename_handler('title')
+        self.assertEqual(fn(self._pset, self._photo, self._suffix),
+                         'Some Photo.jpg')
+
+    def test_title_and_id(self):
+        fn = get_filename_handler('title_and_id')
+        self.assertEqual(fn(self._pset, self._photo, self._suffix),
+                         'Some Photo-123.jpg')
+
+    def test_id(self):
+        fn = get_filename_handler('id')
+        self.assertEqual(fn(self._pset, self._photo, self._suffix),
+                         '123.jpg')
+
+    def test_title_increment(self):
+        fn = get_filename_handler('title_increment')
+        # Ensure increment on same title
+        self.assertEqual(fn(self._pset, self._photo, self._suffix),
+                         'Some Photo.jpg')
+        self.assertEqual(fn(self._pset, self._photo, self._suffix),
+                         'Some Photo(1).jpg')
+
+        # Ensure no increment on different title
+        photo2 = AttrDict({'title': 'Some Other Photo', 'id': 124})
+        self.assertEqual(fn(self._pset, photo2, self._suffix),
+                         'Some Other Photo.jpg')
+
+        # Ensure no increment on same title, but different set
+        pset2 = AttrDict({'title': 'Some Other Set', 'id': 1000})
+        self.assertEqual(fn(pset2, self._photo, self._suffix),
+                         'Some Photo.jpg')
+
+if __name__ == '__main__':
+    unittest.main()