sweeper/test/test_sweeper.py

#!/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()