def test_unregisted_restype(self):
rm = resource.TestResourceManager(
resource.LocalResourceManagerBackend()).create_session()
with self.assertRaises(ValueError):
rm.acquire_resource("xxx")
with self.assertRaises(ValueError):
rm.release_resource("xxx", 12)
def test_nofile_raise(self):
fm = resource.TestResourceManager(
resource.LocalResourceManagerBackend()).create_session()
self.assertRaisesRegexp(Exception, "文件不存在", fm.get_file, 'a.txt')
self.assertRaisesRegexp(Exception, "文件不存在", resource.get_file, 'a.txt')
self.assertRaisesRegexp(Exception, "目录不存在", fm.list_dir, '')
self.assertRaisesRegexp(Exception, "目录不存在", resource.list_dir, '')
def test_duplicated_raise(self):
_, local_dir = _create_local_testfile()
dup_dir = _copy_testfile(local_dir)
self.addCleanup(shutil.rmtree, dup_dir, True)
fm = resource.TestResourceManager(resource.LocalResourceManagerBackend()).create_session()
self.assertRaisesRegex(Exception, "multiple results", fm.get_file, test_file_name)
self.assertRaisesRegex(Exception, "multiple results", fm.list_dir, test_dir_name)
self.assertRaisesRegex(Exception, "multiple results", resource.get_file, test_file_name)
self.assertRaisesRegex(Exception, "multiple results", resource.list_dir, test_dir_name)
def test_get_file(self):
local_file, link_file, res_dir = _create_testfile()
self.addCleanup(_rm_testfile)
fm = resource.TestResourceManager(
resource.LocalResourceManagerBackend()).create_session()
self.assertEqual(local_file, fm.get_file('a.txt'))
self.assertEqual(link_file, fm.get_file('test.txt'))
self.assertEqual(local_file, resource.get_file('a.txt'))
self.assertEqual(link_file, resource.get_file('test.txt'))
def test_duplicated_raise(self):
_, _, res_dir = _create_testfile()
_copy_testfile(res_dir)
self.addCleanup(_rm_testfile)
fm = resource.TestResourceManager(
resource.LocalResourceManagerBackend()).create_session()
self.assertRaisesRegexp(Exception, "存在多个", fm.get_file, 'a.txt')
self.assertRaisesRegexp(Exception, "存在多个", fm.list_dir, '')
self.assertRaisesRegexp(Exception, "存在多个", resource.get_file, 'a.txt')
self.assertRaisesRegexp(Exception, "存在多个", resource.list_dir, '')
def test_get_local_file(self):
fm = resource.TestResourceManager(
resource.LocalResourceManagerBackend()).create_session()
self.assertEqual(self.local_file, fm.get_file(test_file_name))
self.assertEqual(self.local_file, resource.get_file(test_file_name))
paths = []
for it in os.listdir(self.local_dir):
paths.append(smart_text(os.path.join(self.local_dir, it)))
self.assertEqual(paths, fm.list_dir(''))
self.assertEqual(paths, resource.list_dir(''))
def test_list_dir(self):
_, _, res_dir = _create_testfile()
self.addCleanup(_rm_testfile)
fm = resource.TestResourceManager(
resource.LocalResourceManagerBackend()).create_session()
paths = []
from testbase.util import _to_unicode
for it in os.listdir(res_dir):
paths.append(_to_unicode(os.path.join(res_dir, it)))
self.assertEqual(paths, fm.list_dir(''))
self.assertEqual(paths, resource.list_dir(''))
def test_get_local_file(self):
fm = resource.TestResourceManager(resource.LocalResourceManagerBackend()).create_session()
self.assertEqual(self.local_file, fm.get_file(test_file_name))
self.assertEqual(self.local_file, resource.get_file(test_file_name))
paths = os.listdir(os.path.join(self.local_dir, test_dir_name))
for i, path in enumerate(paths):
if path.endswith(".link"):
paths[i] = path[:-5]
list_result = fm.list_dir(test_dir_name)
self.assertEqual(paths, list_result)
list_result = resource.list_dir(test_dir_name)
self.assertEqual(paths, list_result)
def test_nofile_raise(self):
fm = resource.TestResourceManager(resource.LocalResourceManagerBackend()).create_session()
self.assertRaisesRegex(Exception, "not found", fm.get_file, 'a_xxx.txt')
self.assertRaisesRegex(Exception, "not found", resource.get_file, 'a_xxx.txt')
self.assertRaisesRegex(Exception, "not found", fm.list_dir, 'xxx_xxx')
self.assertRaisesRegex(Exception, "not found", resource.list_dir, 'xxx_xxx')
