mirror of
https://github.com/nikdoof/tvorganise.git
synced 2025-12-13 06:42:16 +00:00
61 lines
1.6 KiB
Python
Executable File
61 lines
1.6 KiB
Python
Executable File
#!/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=None):
|
|
|
|
if not suite:
|
|
suite = unittest.TestSuite()
|
|
|
|
suite.addTest(unittest.makeSuite(testTvOrganise))
|
|
suite.addTest(unittest.makeSuite(testFindFiles))
|
|
|
|
return suite
|
|
|
|
def run():
|
|
unittest.TextTestRunner(verbosity=2).run(suite())
|
|
|