79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
import os
|
|
import tempfile
|
|
|
|
from packaging.dist import Distribution
|
|
from packaging.tests import support, unittest
|
|
|
|
|
|
class TestingSupportTestCase(unittest.TestCase):
|
|
|
|
def test_fake_dec(self):
|
|
@support.fake_dec(1, 2, k=3)
|
|
def func(arg0, *args, **kargs):
|
|
return arg0, args, kargs
|
|
self.assertEqual(func(-1, -2, k=-3), (-1, (-2,), {'k': -3}))
|
|
|
|
def test_TempdirManager(self):
|
|
files = {}
|
|
|
|
class Tester(support.TempdirManager, unittest.TestCase):
|
|
|
|
def test_mktempfile(self2):
|
|
tmpfile = self2.mktempfile()
|
|
files['test_mktempfile'] = tmpfile.name
|
|
self.assertTrue(os.path.isfile(tmpfile.name))
|
|
|
|
def test_mkdtemp(self2):
|
|
tmpdir = self2.mkdtemp()
|
|
files['test_mkdtemp'] = tmpdir
|
|
self.assertTrue(os.path.isdir(tmpdir))
|
|
|
|
def test_write_file(self2):
|
|
tmpdir = self2.mkdtemp()
|
|
files['test_write_file'] = tmpdir
|
|
self2.write_file((tmpdir, 'file1'), 'me file 1')
|
|
file1 = os.path.join(tmpdir, 'file1')
|
|
self.assertTrue(os.path.isfile(file1))
|
|
text = ''
|
|
with open(file1, 'r') as f:
|
|
text = f.read()
|
|
self.assertEqual(text, 'me file 1')
|
|
|
|
def test_create_dist(self2):
|
|
project_dir, dist = self2.create_dist()
|
|
files['test_create_dist'] = project_dir
|
|
self.assertTrue(os.path.isdir(project_dir))
|
|
self.assertIsInstance(dist, Distribution)
|
|
|
|
def test_assertIsFile(self2):
|
|
fd, fn = tempfile.mkstemp()
|
|
os.close(fd)
|
|
self.addCleanup(support.unlink, fn)
|
|
self2.assertIsFile(fn)
|
|
self.assertRaises(AssertionError, self2.assertIsFile, 'foO')
|
|
|
|
def test_assertIsNotFile(self2):
|
|
tmpdir = self2.mkdtemp()
|
|
self2.assertIsNotFile(tmpdir)
|
|
|
|
tester = Tester()
|
|
for name in ('test_mktempfile', 'test_mkdtemp', 'test_write_file',
|
|
'test_create_dist', 'test_assertIsFile',
|
|
'test_assertIsNotFile'):
|
|
tester.setUp()
|
|
try:
|
|
getattr(tester, name)()
|
|
finally:
|
|
tester.tearDown()
|
|
|
|
# check clean-up
|
|
if name in files:
|
|
self.assertFalse(os.path.exists(files[name]))
|
|
|
|
|
|
def test_suite():
|
|
return unittest.makeSuite(TestingSupportTestCase)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(defaultTest="test_suite")
|