diff -r f7f9d5ac6d60 Lib/packaging/tests/test_support.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Lib/packaging/tests/test_support.py Mon Oct 10 21:49:21 2011 +0200 @@ -0,0 +1,90 @@ +"""Tests for packaging.tests.support""" + +import os +import unittest + +from packaging.dist import Distribution +from packaging.tests import support + +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 = {} + dirs = {} + + class Tester(support.TempdirManager, unittest.TestCase): + + def test_mktempfile(self2): + tmpFile = self2.mktempfile() + files['from_test_mktempfile'] = tmpFile.name + self.assertTrue(os.path.exists(tmpFile.name)) + self.assertTrue(os.path.isfile(tmpFile.name)) + + def test_mkdtemp(self2): + tmpdir = self2.mkdtemp() + dirs['from_test_mkdtemp'] = tmpdir + self.assertTrue(os.path.exists(tmpdir)) + self.assertTrue(os.path.isdir(tmpdir)) + + def test_write_file(self2): + tmpdir = self2.mkdtemp() + dirs['from_test_write_file'] = tmpdir + self2.write_file((tmpdir, 'file1'), 'me file 1') + file1 = os.path.join(tmpdir, 'file1') + self.assertTrue(os.path.exists(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() + dirs['from_test_create_dist'] = project_dir + self.assertTrue(os.path.exists(project_dir)) + self.assertTrue(os.path.isdir(project_dir)) + self.assertTrue(hasattr(dist, 'get_fullname')) + + def test_assertIsFile(self2): + try: + self2.assertIsFile(__file__) + except: + raise + self.assertRaises(AssertionError, self2.assertIsFile, 'foo') + + def test_assertIsNotFile(self2): + try: + self2.assertIsNotFile(os.path.dirname(__file__)) + except: + raise + + tester = Tester() + tester.setUp() + tester.test_mktempfile() + tester.test_mkdtemp() + tester.test_write_file() + tester.test_create_dist() + tester.test_assertIsFile() + tester.test_assertIsNotFile() + tester.tearDown() + + # Check clean-up + self.assertFalse(os.path.exists(files['from_test_mktempfile'])) + self.assertFalse(os.path.exists(dirs['from_test_mkdtemp'])) + self.assertFalse(os.path.exists(dirs['from_test_write_file'])) + self.assertFalse(os.path.exists(dirs['from_test_create_dist'])) + + +def test_suite(): + return unittest.makeSuite(TestingSupportTestCase) + +if __name__ == "__main__": + unittest.main(defaultTest="test_suite") +