56 lines
1.7 KiB
Python
56 lines
1.7 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
|
|
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)
|
|
|
|
# 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_iter_file_dups_dups(self):
|
|
it = iter_file_dups([os.path.join(mydir, 'testfiles_dups')])
|
|
dups_exist = False
|
|
for x in it:
|
|
dups_exist = True
|
|
filepath, h, dups = x
|
|
self.assertNotIn(filepath, dups)
|
|
self.assertTrue(len(dups) > 0)
|
|
self.assertTrue(dups_exist)
|
|
|
|
def test_iter_file_dups_nodups(self):
|
|
it = iter_file_dups([os.path.join(mydir, 'testfiles_nodups')])
|
|
dups_exist = False
|
|
for x in it:
|
|
dups_exist = True
|
|
break
|
|
self.assertFalse(dups_exist)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|