From 0cafb2a8f142e919026524a85977270aff5cf0bc Mon Sep 17 00:00:00 2001
From: Andrew Williams <andy@tensixtyone.com>
Date: Tue, 26 Jan 2010 16:31:30 +0000
Subject: [PATCH] Start of a simple test suite

---
 test/tvorganise-test.py | 57 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100755 test/tvorganise-test.py

diff --git a/test/tvorganise-test.py b/test/tvorganise-test.py
new file mode 100755
index 0000000..24e1e6b
--- /dev/null
+++ b/test/tvorganise-test.py
@@ -0,0 +1,57 @@
+#!/usr/bin/env python
+#encoding:utf-8
+
+import tvorganise
+import unittest
+import os
+import shutil
+
+class testTvOrganise(unittest.TestCase):
+    """
+    Test class for TvOrganise module
+    """
+
+    def setUp(self):
+        pass
+
+
+class testFindFiles(unittest.TestCase):
+    """
+    Test find_files() function
+    """
+
+    def setUp(self):
+        os.makedirs("/tmp/find-files-test/folder1")
+        os.makedirs("/tmp/find-files-test/folder2")
+
+	open('/tmp/find-files-test/folder1/file1', 'w').close() 
+        open('/tmp/find-files-test/folder1/file2', 'w').close()
+        open('/tmp/find-files-test/folder2/file3', 'w').close()
+        open('/tmp/find-files-test/folder2/file4', 'w').close()
+        open('/tmp/find-files-test/folder2/file5', 'w').close()
+
+    def tearDown(self):
+        shutil.rmtree("/tmp/find-files-test/")        
+
+    def testFolderList(self):
+        self.assertEqual(len(tvorganise.find_files("/tmp/find-files-test/")),5)
+        self.assertEqual(len(tvorganise.find_files("/tmp/find-files-test/folder1")),2)
+        self.assertEqual(len(tvorganise.find_files("/tmp/find-files-test/folder2")),3)
+
+    def testMultipleInput(self):
+        input = ['/tmp/find-files-test/folder1/', '/tmp/find-files-test/folder2/']
+        self.assertEqual(len(tvorganise.find_files(input)),5)
+
+    def testCrapInput(self):
+        self.assertEqual(len(tvorganise.find_files("belfrhe")),0)
+
+def suite():
+    suite = unittest.TestSuite()
+    suite.addTest(unittest.makeSuite(testTvOrganise))
+    suite.addTest(unittest.makeSuite(testFindFiles))
+ 
+    return suite
+
+if __name__ == '__main__':
+    unittest.TextTestRunner(verbosity=2).run(suite())
+