60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
# Author: Darko Poljak <darko.poljak@gmail.com>
|
|
# License: GPLv3
|
|
|
|
import unittest
|
|
from sweeper import Sweeper
|
|
import os
|
|
|
|
mydir = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
|
|
class TestSweeper(unittest.TestCase):
|
|
def test_file_dups_dups(self):
|
|
swp = Sweeper(topdirs=[os.path.join(mydir, 'testfiles_dups')])
|
|
dups = swp.file_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):
|
|
swp = Sweeper(topdirs=[os.path.join(mydir, 'testfiles_nodups')])
|
|
dups = swp.file_dups()
|
|
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):
|
|
swp = Sweeper(topdirs=[os.path.join(mydir, 'testfiles_dups')],
|
|
safe_mode=True)
|
|
dups = swp.file_dups()
|
|
for h, flist in dups.items():
|
|
if len(flist) > 1:
|
|
dups_exist = True
|
|
self.assertTrue(dups_exist)
|
|
|
|
def test_iter_file_dups_dups(self):
|
|
swp = Sweeper(topdirs=[os.path.join(mydir, 'testfiles_dups')])
|
|
dups_exist = False
|
|
for x in swp:
|
|
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):
|
|
swp = Sweeper([os.path.join(mydir, 'testfiles_nodups')])
|
|
dups_exist = False
|
|
for x in swp:
|
|
dups_exist = True
|
|
break
|
|
self.assertFalse(dups_exist)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|