56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
# Author: Darko Poljak <darko.poljak@gmail.com>
|
|
# License: GPLv3
|
|
|
|
import unittest
|
|
from sweeper import file_dups, iter_file_dups, file_dups_immediate
|
|
import os
|
|
|
|
mydir = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
|
|
class TestSweeper(unittest.TestCase):
|
|
def test_file_dups_dups(self):
|
|
dups = file_dups([os.path.join(mydir, 'testfiles_dups')])
|
|
dups_exist = False
|
|
for h, flist in dups.items():
|
|
if len(flist) > 1:
|
|
dups_exist = True
|
|
self.assertTrue(dups_exist)
|
|
|
|
def test_file_dups_nodups(self):
|
|
dups = file_dups([os.path.join(mydir, 'testfiles_nodups')])
|
|
for h, flist in dups.items():
|
|
self.assertTrue(len(flist) == 1)
|
|
|
|
def test_iter_fule_dups_rethash(self):
|
|
for item in iter_file_dups([os.path.join(mydir, 'testfiles_dups')],
|
|
rethash=True):
|
|
self.assertTrue(type(item).__name__ == 'tuple')
|
|
|
|
def test_iter_fule_dups_norethash(self):
|
|
for item in iter_file_dups([os.path.join(mydir, 'testfiles_dups')]):
|
|
self.assertTrue(type(item).__name__ == 'list')
|
|
|
|
# does not actually test safe_mode, we would need to find
|
|
# hash collision
|
|
def test_file_dups_safe_mode(self):
|
|
dups = file_dups([os.path.join(mydir, 'testfiles_dups')],
|
|
safe_mode=True)
|
|
for h, flist in dups.items():
|
|
if len(flist) > 1:
|
|
dups_exist = True
|
|
self.assertTrue(dups_exist)
|
|
|
|
def test_file_dups_immediate_dups(self):
|
|
it = file_dups_immediate([os.path.join(mydir, 'testfiles_dups')])
|
|
dups_exist = False
|
|
for x in it:
|
|
dups_exist = True
|
|
break
|
|
self.assertTrue(dups_exist)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|