import unittest

from test import test_support



import posixpath, os

from posixpath import realpath, abspath, join, dirname, basename, relpath



# An absolute path to a temporary filename for testing. We can't rely on TESTFN

# being an absolute path, so we need this.



ABSTFN = abspath(test_support.TESTFN)



def safe_rmdir(dirname):

    try:

        os.rmdir(dirname)

    except OSError:

        pass



class PosixPathTest(unittest.TestCase):



    def setUp(self):

        self.tearDown()



    def tearDown(self):

        for suffix in ["", "1", "2"]:

            test_support.unlink(test_support.TESTFN + suffix)

            safe_rmdir(test_support.TESTFN + suffix)



    def assertIs(self, a, b):

        self.assert_(a is b)



    def test_normcase(self):

        # Check that normcase() is idempotent

        p = "FoO/./BaR"

        p = posixpath.normcase(p)

        self.assertEqual(p, posixpath.normcase(p))



        self.assertRaises(TypeError, posixpath.normcase)



    def test_join(self):

        self.assertEqual(posixpath.join("/foo", "bar", "/bar", "baz"), "/bar/baz")

        self.assertEqual(posixpath.join("/foo", "bar", "baz"), "/foo/bar/baz")

        self.assertEqual(posixpath.join("/foo/", "bar/", "baz/"), "/foo/bar/baz/")



        self.assertRaises(TypeError, posixpath.join)



    def test_splitdrive(self):

        self.assertEqual(posixpath.splitdrive("/foo/bar"), ("", "/foo/bar"))



        self.assertRaises(TypeError, posixpath.splitdrive)



    def test_split(self):

        self.assertEqual(posixpath.split("/foo/bar"), ("/foo", "bar"))

        self.assertEqual(posixpath.split("/"), ("/", ""))

        self.assertEqual(posixpath.split("foo"), ("", "foo"))

        self.assertEqual(posixpath.split("////foo"), ("////", "foo"))

        self.assertEqual(posixpath.split("//foo//bar"), ("//foo", "bar"))



        self.assertRaises(TypeError, posixpath.split)



    def splitextTest(self, path, filename, ext):

        self.assertEqual(posixpath.splitext(path), (filename, ext))

        self.assertEqual(posixpath.splitext("/" + path), ("/" + filename, ext))

        self.assertEqual(posixpath.splitext("abc/" + path), ("abc/" + filename, ext))

        self.assertEqual(posixpath.splitext("abc.def/" + path), ("abc.def/" + filename, ext))

        self.assertEqual(posixpath.splitext("/abc.def/" + path), ("/abc.def/" + filename, ext))

        self.assertEqual(posixpath.splitext(path + "/"), (filename + ext + "/", ""))



    def test_splitext(self):

        self.splitextTest("foo.bar", "foo", ".bar")

        self.splitextTest("foo.boo.bar", "foo.boo", ".bar")

        self.splitextTest("foo.boo.biff.bar", "foo.boo.biff", ".bar")

        self.splitextTest(".csh.rc", ".csh", ".rc")

        self.splitextTest("nodots", "nodots", "")

        self.splitextTest(".cshrc", ".cshrc", "")

        self.splitextTest("...manydots", "...manydots", "")

        self.splitextTest("...manydots.ext", "...manydots", ".ext")

        self.splitextTest(".", ".", "")

        self.splitextTest("..", "..", "")

        self.splitextTest("........", "........", "")

        self.splitextTest("", "", "")

        self.assertRaises(TypeError, posixpath.splitext)



    def test_isabs(self):

        self.assertIs(posixpath.isabs(""), False)

        self.assertIs(posixpath.isabs("/"), True)

        self.assertIs(posixpath.isabs("/foo"), True)

        self.assertIs(posixpath.isabs("/foo/bar"), True)

        self.assertIs(posixpath.isabs("foo/bar"), False)



        self.assertRaises(TypeError, posixpath.isabs)



    def test_splitdrive(self):

        self.assertEqual(posixpath.splitdrive("/foo/bar"), ("", "/foo/bar"))



        self.assertRaises(TypeError, posixpath.splitdrive)



    def test_basename(self):

        self.assertEqual(posixpath.basename("/foo/bar"), "bar")

        self.assertEqual(posixpath.basename("/"), "")

        self.assertEqual(posixpath.basename("foo"), "foo")

        self.assertEqual(posixpath.basename("////foo"), "foo")

        self.assertEqual(posixpath.basename("//foo//bar"), "bar")



        self.assertRaises(TypeError, posixpath.basename)



    def test_dirname(self):

        self.assertEqual(posixpath.dirname("/foo/bar"), "/foo")

        self.assertEqual(posixpath.dirname("/"), "/")

        self.assertEqual(posixpath.dirname("foo"), "")

        self.assertEqual(posixpath.dirname("////foo"), "////")

        self.assertEqual(posixpath.dirname("//foo//bar"), "//foo")



        self.assertRaises(TypeError, posixpath.dirname)



    def test_commonprefix(self):

        self.assertEqual(

            posixpath.commonprefix([]),

            ""

        )

        self.assertEqual(

            posixpath.commonprefix(["/home/swenson/spam", "/home/swen/spam"]),

            "/home/swen"

        )

        self.assertEqual(

            posixpath.commonprefix(["/home/swen/spam", "/home/swen/eggs"]),

            "/home/swen/"

        )

        self.assertEqual(

            posixpath.commonprefix(["/home/swen/spam", "/home/swen/spam"]),

            "/home/swen/spam"

        )



        testlist = ['', 'abc', 'Xbcd', 'Xb', 'XY', 'abcd', 'aXc', 'abd', 'ab', 'aX', 'abcX']

        for s1 in testlist:

            for s2 in testlist:

                p = posixpath.commonprefix([s1, s2])

                self.assert_(s1.startswith(p))

                self.assert_(s2.startswith(p))

                if s1 != s2:

                    n = len(p)

                    self.assertNotEqual(s1[n:n+1], s2[n:n+1])



    def test_getsize(self):

        f = open(test_support.TESTFN, "wb")

        try:

            f.write("foo")

            f.close()

            self.assertEqual(posixpath.getsize(test_support.TESTFN), 3)

        finally:

            if not f.closed:

                f.close()



    def test_time(self):

        f = open(test_support.TESTFN, "wb")

        try:

            f.write("foo")

            f.close()

            f = open(test_support.TESTFN, "ab")

            f.write("bar")

            f.close()

            f = open(test_support.TESTFN, "rb")

            d = f.read()

            f.close()

            self.assertEqual(d, "foobar")



            self.assert_(

                posixpath.getctime(test_support.TESTFN) <=

                posixpath.getmtime(test_support.TESTFN)

            )

        finally:

            if not f.closed:

                f.close()



    def test_islink(self):

        self.assertIs(posixpath.islink(test_support.TESTFN + "1"), False)

        f = open(test_support.TESTFN + "1", "wb")

        try:

            f.write("foo")

            f.close()

            self.assertIs(posixpath.islink(test_support.TESTFN + "1"), False)

            if hasattr(os, "symlink"):

                os.symlink(test_support.TESTFN + "1", test_support.TESTFN + "2")

                self.assertIs(posixpath.islink(test_support.TESTFN + "2"), True)

                os.remove(test_support.TESTFN + "1")

                self.assertIs(posixpath.islink(test_support.TESTFN + "2"), True)

                self.assertIs(posixpath.exists(test_support.TESTFN + "2"), False)

                self.assertIs(posixpath.lexists(test_support.TESTFN + "2"), True)

        finally:

            if not f.close():

                f.close()



        self.assertRaises(TypeError, posixpath.islink)



    def test_exists(self):

        self.assertIs(posixpath.exists(test_support.TESTFN), False)

        f = open(test_support.TESTFN, "wb")

        try:

            f.write("foo")

            f.close()

            self.assertIs(posixpath.exists(test_support.TESTFN), True)

            self.assertIs(posixpath.lexists(test_support.TESTFN), True)

        finally:

            if not f.close():

                f.close()



        self.assertRaises(TypeError, posixpath.exists)



    def test_isdir(self):

        self.assertIs(posixpath.isdir(test_support.TESTFN), False)

        f = open(test_support.TESTFN, "wb")

        try:

            f.write("foo")

            f.close()

            self.assertIs(posixpath.isdir(test_support.TESTFN), False)

            os.remove(test_support.TESTFN)

            os.mkdir(test_support.TESTFN)

            self.assertIs(posixpath.isdir(test_support.TESTFN), True)

            os.rmdir(test_support.TESTFN)

        finally:

            if not f.close():

                f.close()



        self.assertRaises(TypeError, posixpath.isdir)



    def test_isfile(self):

        self.assertIs(posixpath.isfile(test_support.TESTFN), False)

        f = open(test_support.TESTFN, "wb")

        try:

            f.write("foo")

            f.close()

            self.assertIs(posixpath.isfile(test_support.TESTFN), True)

            os.remove(test_support.TESTFN)

            os.mkdir(test_support.TESTFN)

            self.assertIs(posixpath.isfile(test_support.TESTFN), False)

            os.rmdir(test_support.TESTFN)

        finally:

            if not f.close():

                f.close()



        self.assertRaises(TypeError, posixpath.isdir)



    def test_samefile(self):

        f = open(test_support.TESTFN + "1", "wb")

        try:

            f.write("foo")

            f.close()

            self.assertIs(

                posixpath.samefile(

                    test_support.TESTFN + "1",

                    test_support.TESTFN + "1"

                ),

                True

            )

            # If we don't have links, assume that os.stat doesn't return resonable

            # inode information and thus, that samefile() doesn't work

            if hasattr(os, "symlink"):

                os.symlink(

                    test_support.TESTFN + "1",

                    test_support.TESTFN + "2"

                )

                self.assertIs(

                    posixpath.samefile(

                        test_support.TESTFN + "1",

                        test_support.TESTFN + "2"

                    ),

                    True

                )

                os.remove(test_support.TESTFN + "2")

                f = open(test_support.TESTFN + "2", "wb")

                f.write("bar")

                f.close()

                self.assertIs(

                    posixpath.samefile(

                        test_support.TESTFN + "1",

                        test_support.TESTFN + "2"

                    ),

                    False

                )

        finally:

            if not f.close():

                f.close()



        self.assertRaises(TypeError, posixpath.samefile)



    def test_samestat(self):

        f = open(test_support.TESTFN + "1", "wb")

        try:

            f.write("foo")

            f.close()

            self.assertIs(

                posixpath.samestat(

                    os.stat(test_support.TESTFN + "1"),

                    os.stat(test_support.TESTFN + "1")

                ),

                True

            )

            # If we don't have links, assume that os.stat() doesn't return resonable

            # inode information and thus, that samefile() doesn't work

            if hasattr(os, "symlink"):

                if hasattr(os, "symlink"):

                    os.symlink(test_support.TESTFN + "1", test_support.TESTFN + "2")

                    self.assertIs(

                        posixpath.samestat(

                            os.stat(test_support.TESTFN + "1"),

                            os.stat(test_support.TESTFN + "2")

                        ),

                        True

                    )

                    os.remove(test_support.TESTFN + "2")

                f = open(test_support.TESTFN + "2", "wb")

                f.write("bar")

                f.close()

                self.assertIs(

                    posixpath.samestat(

                        os.stat(test_support.TESTFN + "1"),

                        os.stat(test_support.TESTFN + "2")

                    ),

                    False

                )

        finally:

            if not f.close():

                f.close()



        self.assertRaises(TypeError, posixpath.samestat)



    def test_ismount(self):

        self.assertIs(posixpath.ismount("/"), True)



        self.assertRaises(TypeError, posixpath.ismount)



    def test_expanduser(self):

        self.assertEqual(posixpath.expanduser("foo"), "foo")

        try:

            import pwd

        except ImportError:

            pass

        else:

            self.assert_(isinstance(posixpath.expanduser("~/"), basestring))

            # if home directory == root directory, this test makes no sense

            if posixpath.expanduser("~") != '/':

                self.assertEqual(

                    posixpath.expanduser("~") + "/",

                    posixpath.expanduser("~/")

                )

            self.assert_(isinstance(posixpath.expanduser("~root/"), basestring))

            self.assert_(isinstance(posixpath.expanduser("~foo/"), basestring))



            with test_support.EnvironmentVarGuard() as env:

                env.set('HOME', '/')

                self.assertEqual(posixpath.expanduser("~"), "/")



        self.assertRaises(TypeError, posixpath.expanduser)



    def test_expandvars(self):

        oldenv = os.environ.copy()

        try:

            os.environ.clear()

            os.environ["foo"] = "bar"

            os.environ["{foo"] = "baz1"

            os.environ["{foo}"] = "baz2"

            self.assertEqual(posixpath.expandvars("foo"), "foo")

            self.assertEqual(posixpath.expandvars("$foo bar"), "bar bar")

            self.assertEqual(posixpath.expandvars("${foo}bar"), "barbar")

            self.assertEqual(posixpath.expandvars("$[foo]bar"), "$[foo]bar")

            self.assertEqual(posixpath.expandvars("$bar bar"), "$bar bar")

            self.assertEqual(posixpath.expandvars("$?bar"), "$?bar")

            self.assertEqual(posixpath.expandvars("${foo}bar"), "barbar")

            self.assertEqual(posixpath.expandvars("$foo}bar"), "bar}bar")

            self.assertEqual(posixpath.expandvars("${foo"), "${foo")

            self.assertEqual(posixpath.expandvars("${{foo}}"), "baz1}")

            self.assertEqual(posixpath.expandvars("$foo$foo"), "barbar")

            self.assertEqual(posixpath.expandvars("$bar$bar"), "$bar$bar")

        finally:

            os.environ.clear()

            os.environ.update(oldenv)



        self.assertRaises(TypeError, posixpath.expandvars)



    def test_normpath(self):

        self.assertEqual(posixpath.normpath(""), ".")

        self.assertEqual(posixpath.normpath("/"), "/")

        self.assertEqual(posixpath.normpath("//"), "//")

        self.assertEqual(posixpath.normpath("///"), "/")

        self.assertEqual(posixpath.normpath("///foo/.//bar//"), "/foo/bar")

        self.assertEqual(posixpath.normpath("///foo/.//bar//.//..//.//baz"), "/foo/baz")

        self.assertEqual(posixpath.normpath("///..//./foo/.//bar"), "/foo/bar")



        self.assertRaises(TypeError, posixpath.normpath)



    def test_abspath(self):

        self.assert_("foo" in posixpath.abspath("foo"))



        self.assertRaises(TypeError, posixpath.abspath)



    def test_realpath(self):

        self.assert_("foo" in realpath("foo"))

        self.assertRaises(TypeError, posixpath.realpath)



    if hasattr(os, "symlink"):

        def test_realpath_basic(self):

            # Basic operation.

            try:

                os.symlink(ABSTFN+"1", ABSTFN)

                self.assertEqual(realpath(ABSTFN), ABSTFN+"1")

            finally:

                test_support.unlink(ABSTFN)



        def test_realpath_symlink_loops(self):

            # Bug #930024, return the path unchanged if we get into an infinite

            # symlink loop.

            try:

                old_path = abspath('.')

                os.symlink(ABSTFN, ABSTFN)

                self.assertEqual(realpath(ABSTFN), ABSTFN)



                os.symlink(ABSTFN+"1", ABSTFN+"2")

                os.symlink(ABSTFN+"2", ABSTFN+"1")

                self.assertEqual(realpath(ABSTFN+"1"), ABSTFN+"1")

                self.assertEqual(realpath(ABSTFN+"2"), ABSTFN+"2")



                # Test using relative path as well.

                os.chdir(dirname(ABSTFN))

                self.assertEqual(realpath(basename(ABSTFN)), ABSTFN)

            finally:

                os.chdir(old_path)

                test_support.unlink(ABSTFN)

                test_support.unlink(ABSTFN+"1")

                test_support.unlink(ABSTFN+"2")



        def test_realpath_resolve_parents(self):

            # We also need to resolve any symlinks in the parents of a relative

            # path passed to realpath. E.g.: current working directory is

            # /usr/doc with 'doc' being a symlink to /usr/share/doc. We call

            # realpath("a"). This should return /usr/share/doc/a/.

            try:

                old_path = abspath('.')

                os.mkdir(ABSTFN)

                os.mkdir(ABSTFN + "/y")

                os.symlink(ABSTFN + "/y", ABSTFN + "/k")



                os.chdir(ABSTFN + "/k")

                self.assertEqual(realpath("a"), ABSTFN + "/y/a")

            finally:

                os.chdir(old_path)

                test_support.unlink(ABSTFN + "/k")

                safe_rmdir(ABSTFN + "/y")

                safe_rmdir(ABSTFN)



        def test_realpath_resolve_before_normalizing(self):

            # Bug #990669: Symbolic links should be resolved before we

            # normalize the path. E.g.: if we have directories 'a', 'k' and 'y'

            # in the following hierarchy:

            # a/k/y

            #

            # and a symbolic link 'link-y' pointing to 'y' in directory 'a',

            # then realpath("link-y/..") should return 'k', not 'a'.

            try:

                old_path = abspath('.')

                os.mkdir(ABSTFN)

                os.mkdir(ABSTFN + "/k")

                os.mkdir(ABSTFN + "/k/y")

                os.symlink(ABSTFN + "/k/y", ABSTFN + "/link-y")



                # Absolute path.

                self.assertEqual(realpath(ABSTFN + "/link-y/.."), ABSTFN + "/k")

                # Relative path.

                os.chdir(dirname(ABSTFN))

                self.assertEqual(realpath(basename(ABSTFN) + "/link-y/.."), ABSTFN + "/k")

            finally:

                os.chdir(old_path)

                test_support.unlink(ABSTFN + "/link-y")

                safe_rmdir(ABSTFN + "/k/y")

                safe_rmdir(ABSTFN + "/k")

                safe_rmdir(ABSTFN)



        def test_realpath_resolve_first(self):

            # Bug #1213894: The first component of the path, if not absolute,

            # must be resolved too.



            try:

                old_path = abspath('.')

                os.mkdir(ABSTFN)

                os.mkdir(ABSTFN + "/k")

                os.symlink(ABSTFN, ABSTFN + "link")

                os.chdir(dirname(ABSTFN))



                base = basename(ABSTFN)

                self.assertEqual(realpath(base + "link"), ABSTFN)

                self.assertEqual(realpath(base + "link/k"), ABSTFN + "/k")

            finally:

                os.chdir(old_path)

                test_support.unlink(ABSTFN + "link")

                safe_rmdir(ABSTFN + "/k")

                safe_rmdir(ABSTFN)



    def test_relpath(self):

        (real_getcwd, os.getcwd) = (os.getcwd, lambda: r"/home/user/bar")

        try:

            curdir = os.path.split(os.getcwd())[-1]

            self.assertRaises(ValueError, posixpath.relpath, "")

            self.assertEqual(posixpath.relpath("a"), "a")

            self.assertEqual(posixpath.relpath(posixpath.abspath("a")), "a")

            self.assertEqual(posixpath.relpath("a/b"), "a/b")

            self.assertEqual(posixpath.relpath("../a/b"), "../a/b")

            self.assertEqual(posixpath.relpath("a", "../b"), "../"+curdir+"/a")

            self.assertEqual(posixpath.relpath("a/b", "../c"), "../"+curdir+"/a/b")

            self.assertEqual(posixpath.relpath("a", "b/c"), "../../a")

            self.assertEqual(posixpath.relpath("a", "a"), ".")

        finally:

            os.getcwd = real_getcwd



def test_main():

    test_support.run_unittest(PosixPathTest)



if __name__=="__main__":

    test_main()

