import os import tempfile from distutils2.dist import Distribution from distutils2.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 = '' f = open(file1, 'r') try: text = f.read() finally: f.close() 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")