

import os, filecmp, shutil, tempfile, shutil

import unittest

from test import test_support



class FileCompareTestCase(unittest.TestCase):

    def setUp(self):

        self.name = test_support.TESTFN

        self.name_same = test_support.TESTFN + '-same'

        self.name_diff = test_support.TESTFN + '-diff'

        data = 'Contents of file go here.\n'

        for name in [self.name, self.name_same, self.name_diff]:

            output = open(name, 'w')

            output.write(data)

            output.close()



        output = open(self.name_diff, 'a+')

        output.write('An extra line.\n')

        output.close()

        self.dir = tempfile.gettempdir()



    def tearDown(self):

        os.unlink(self.name)

        os.unlink(self.name_same)

        os.unlink(self.name_diff)



    def test_matching(self):

        self.failUnless(filecmp.cmp(self.name, self.name_same),

                        "Comparing file to itself fails")

        self.failUnless(filecmp.cmp(self.name, self.name_same, shallow=False),

                        "Comparing file to itself fails")

        self.failUnless(filecmp.cmp(self.name, self.name, shallow=False),

                        "Comparing file to identical file fails")

        self.failUnless(filecmp.cmp(self.name, self.name),

                        "Comparing file to identical file fails")



    def test_different(self):

        self.failIf(filecmp.cmp(self.name, self.name_diff),

                    "Mismatched files compare as equal")

        self.failIf(filecmp.cmp(self.name, self.dir),

                    "File and directory compare as equal")



class DirCompareTestCase(unittest.TestCase):

    def setUp(self):

        tmpdir = tempfile.gettempdir()

        self.dir = os.path.join(tmpdir, 'dir')

        self.dir_same = os.path.join(tmpdir, 'dir-same')

        self.dir_diff = os.path.join(tmpdir, 'dir-diff')

        self.caseinsensitive = os.path.normcase('A') == os.path.normcase('a')

        data = 'Contents of file go here.\n'

        for dir in [self.dir, self.dir_same, self.dir_diff]:

            shutil.rmtree(dir, True)

            os.mkdir(dir)

            if self.caseinsensitive and dir is self.dir_same:

                fn = 'FiLe'     # Verify case-insensitive comparison

            else:

                fn = 'file'

            output = open(os.path.join(dir, fn), 'w')

            output.write(data)

            output.close()



        output = open(os.path.join(self.dir_diff, 'file2'), 'w')

        output.write('An extra file.\n')

        output.close()



    def tearDown(self):

        shutil.rmtree(self.dir)

        shutil.rmtree(self.dir_same)

        shutil.rmtree(self.dir_diff)



    def test_cmpfiles(self):

        self.failUnless(filecmp.cmpfiles(self.dir, self.dir, ['file']) ==

                        (['file'], [], []),

                        "Comparing directory to itself fails")

        self.failUnless(filecmp.cmpfiles(self.dir, self.dir_same, ['file']) ==

                        (['file'], [], []),

                        "Comparing directory to same fails")



        # Try it with shallow=False

        self.failUnless(filecmp.cmpfiles(self.dir, self.dir, ['file'],

                                         shallow=False) ==

                        (['file'], [], []),

                        "Comparing directory to itself fails")

        self.failUnless(filecmp.cmpfiles(self.dir, self.dir_same, ['file'],

                                         shallow=False),

                        "Comparing directory to same fails")



        # Add different file2

        output = open(os.path.join(self.dir, 'file2'), 'w')

        output.write('Different contents.\n')

        output.close()



        self.failIf(filecmp.cmpfiles(self.dir, self.dir_same,

                                     ['file', 'file2']) ==

                    (['file'], ['file2'], []),

                    "Comparing mismatched directories fails")





    def test_dircmp(self):

        # Check attributes for comparison of two identical directories

        d = filecmp.dircmp(self.dir, self.dir_same)

        if self.caseinsensitive:

            self.assertEqual([d.left_list, d.right_list],[['file'], ['FiLe']])

        else:

            self.assertEqual([d.left_list, d.right_list],[['file'], ['file']])

        self.failUnless(d.common == ['file'])

        self.failUnless(d.left_only == d.right_only == [])

        self.failUnless(d.same_files == ['file'])

        self.failUnless(d.diff_files == [])



        # Check attributes for comparison of two different directories

        d = filecmp.dircmp(self.dir, self.dir_diff)

        self.failUnless(d.left_list == ['file'])

        self.failUnless(d.right_list == ['file', 'file2'])

        self.failUnless(d.common == ['file'])

        self.failUnless(d.left_only == [])

        self.failUnless(d.right_only == ['file2'])

        self.failUnless(d.same_files == ['file'])

        self.failUnless(d.diff_files == [])



        # Add different file2

        output = open(os.path.join(self.dir, 'file2'), 'w')

        output.write('Different contents.\n')

        output.close()

        d = filecmp.dircmp(self.dir, self.dir_diff)

        self.failUnless(d.same_files == ['file'])

        self.failUnless(d.diff_files == ['file2'])





def test_main():

    test_support.run_unittest(FileCompareTestCase, DirCompareTestCase)



if __name__ == "__main__":

    test_main()

