class TestPackageResource(Common, unittest.TestCase):
def setUp(self):
super(TestPackageResource, self).setUp()
sys.path.append(self.base)
self.res = PackageResource(self.mod)
def tearDown(self):
del self.res
del sys.modules[self.mod]
sys.path.remove(self.base)
super(TestPackageResource, self).tearDown()
def test_filename(self):
with self.res.filename('a.yml') as filename:
with open(filename, encoding='utf-8') as f:
self.assertEqual(
load(f.read())['path'],
'a.yml')
self.assertTrue(os.path.exists(filename))
with self.res.get(['subdir', 'b.yml']).filename() as filename:
with open(filename, encoding='utf-8') as f:
self.assertEqual(
load(f.read())['path'],
'subdir/b.yml')
self.assertTrue(os.path.exists(filename))
class TestEggResource(Common, unittest.TestCase):
def setUp(self):
super(TestEggResource, self).setUp()
cwd = os.getcwd()
os.chdir(self.base)
with captured_output('stdout'):
with captured_output('stderr'):
setuptools.setup(
name=self.mod,
packages=[self.mod],
script_args=['bdist_egg', '--quiet'],
package_data={self.mod:[
'a.yml',
os.path.join('subdir', 'b.yml')]})
os.chdir(cwd)
self.eggs = os.listdir(os.path.join(self.base, 'dist'))
for egg in self.eggs:
sys.path.append(os.path.join(self.base, 'dist', egg))
self.res = PackageResource(self.mod)
def tearDown(self):
del self.res
del sys.modules[self.mod]
for egg in self.eggs:
sys.path.remove(os.path.join(self.base, 'dist', egg))
super(TestEggResource, self).tearDown()
def test_is_valid(self):
"""
Check that the test uses a .egg resource.
Only then the test results confirm that the package is working.
"""
self.assertTrue(self.res._is_extracted)
def test_filename(self):
with self.res.filename('a.yml') as filename:
with open(filename, encoding='utf-8') as f:
self.assertEqual(
load(f.read())['path'],
'a.yml')
self.assertFalse(os.path.exists(filename))
with self.res.get(['subdir', 'b.yml']).filename() as filename:
with open(filename, encoding='utf-8') as f:
self.assertEqual(
load(f.read())['path'],
'subdir/b.yml')
self.assertFalse(os.path.exists(filename))
# The resource should be accessible again, even though the file was
# deleted when exiting the context manager:
with self.res.get(['subdir', 'b.yml']).filename() as filename:
with open(filename, encoding='utf-8') as f:
self.assertEqual(
load(f.read())['path'],
'subdir/b.yml')
self.assertFalse(os.path.exists(filename))
def test_use_filename_twice(self):
with self.res.filename('a.yml') as filename:
self.assertTrue(os.path.exists(filename))
self.assertFalse(os.path.exists(filename))
with self.res.filename('a.yml') as filename:
self.assertTrue(os.path.exists(filename))
with open(filename, encoding='utf-8') as f:
self.assertEqual(
load(f.read())['path'],
'a.yml')
