90 lines
3.3 KiB
Python
90 lines
3.3 KiB
Python
from unittest.case import TestCase
|
|
import oe, oe.path
|
|
import tempfile
|
|
import os
|
|
import errno
|
|
import shutil
|
|
|
|
class TestRealPath(TestCase):
|
|
DIRS = [ "a", "b", "etc", "sbin", "usr", "usr/bin", "usr/binX", "usr/sbin", "usr/include", "usr/include/gdbm" ]
|
|
FILES = [ "etc/passwd", "b/file" ]
|
|
LINKS = [
|
|
( "bin", "/usr/bin", "/usr/bin" ),
|
|
( "binX", "usr/binX", "/usr/binX" ),
|
|
( "c", "broken", "/broken" ),
|
|
( "etc/passwd-1", "passwd", "/etc/passwd" ),
|
|
( "etc/passwd-2", "passwd-1", "/etc/passwd" ),
|
|
( "etc/passwd-3", "/etc/passwd-1", "/etc/passwd" ),
|
|
( "etc/shadow-1", "/etc/shadow", "/etc/shadow" ),
|
|
( "etc/shadow-2", "/etc/shadow-1", "/etc/shadow" ),
|
|
( "prog-A", "bin/prog-A", "/usr/bin/prog-A" ),
|
|
( "prog-B", "/bin/prog-B", "/usr/bin/prog-B" ),
|
|
( "usr/bin/prog-C", "../../sbin/prog-C", "/sbin/prog-C" ),
|
|
( "usr/bin/prog-D", "/sbin/prog-D", "/sbin/prog-D" ),
|
|
( "usr/binX/prog-E", "../sbin/prog-E", None ),
|
|
( "usr/bin/prog-F", "../../../sbin/prog-F", "/sbin/prog-F" ),
|
|
( "loop", "a/loop", None ),
|
|
( "a/loop", "../loop", None ),
|
|
( "b/test", "file/foo", "/b/file/foo" ),
|
|
]
|
|
|
|
LINKS_PHYS = [
|
|
( "./", "/", "" ),
|
|
( "binX/prog-E", "/usr/sbin/prog-E", "/sbin/prog-E" ),
|
|
]
|
|
|
|
EXCEPTIONS = [
|
|
( "loop", errno.ELOOP ),
|
|
( "b/test", errno.ENOENT ),
|
|
]
|
|
|
|
def __del__(self):
|
|
try:
|
|
#os.system("tree -F %s" % self.tmpdir)
|
|
shutil.rmtree(self.tmpdir)
|
|
except:
|
|
pass
|
|
|
|
def setUp(self):
|
|
self.tmpdir = tempfile.mkdtemp(prefix = "oe-test_path")
|
|
self.root = os.path.join(self.tmpdir, "R")
|
|
|
|
os.mkdir(os.path.join(self.tmpdir, "_real"))
|
|
os.symlink("_real", self.root)
|
|
|
|
for d in self.DIRS:
|
|
os.mkdir(os.path.join(self.root, d))
|
|
for f in self.FILES:
|
|
open(os.path.join(self.root, f), "w")
|
|
for l in self.LINKS:
|
|
os.symlink(l[1], os.path.join(self.root, l[0]))
|
|
|
|
def __realpath(self, file, use_physdir, assume_dir = True):
|
|
return oe.path.realpath(os.path.join(self.root, file), self.root,
|
|
use_physdir, assume_dir = assume_dir)
|
|
|
|
def test_norm(self):
|
|
for l in self.LINKS:
|
|
if l[2] == None:
|
|
continue
|
|
|
|
target_p = self.__realpath(l[0], True)
|
|
target_l = self.__realpath(l[0], False)
|
|
|
|
if l[2] != False:
|
|
self.assertEqual(target_p, target_l)
|
|
self.assertEqual(l[2], target_p[len(self.root):])
|
|
|
|
def test_phys(self):
|
|
for l in self.LINKS_PHYS:
|
|
target_p = self.__realpath(l[0], True)
|
|
target_l = self.__realpath(l[0], False)
|
|
|
|
self.assertEqual(l[1], target_p[len(self.root):])
|
|
self.assertEqual(l[2], target_l[len(self.root):])
|
|
|
|
def test_loop(self):
|
|
for e in self.EXCEPTIONS:
|
|
self.assertRaisesRegex(OSError, r'\[Errno %u\]' % e[1],
|
|
self.__realpath, e[0], False, False)
|