summaryrefslogtreecommitdiffstats
path: root/Lib/packaging/tests/test_support.py
blob: 0ae9e1b8cfbdc1fa0d1c53def2c8f0da3dd51560 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
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")