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
| """Tests for distutils.command.clean."""
|
| import os
| from distutils.command.clean import clean
| from distutils.tests import support
|
|
| class TestClean(support.TempdirManager):
| def test_simple_run(self):
| pkg_dir, dist = self.create_dist()
| cmd = clean(dist)
|
| # let's add some elements clean should remove
| dirs = [
| (d, os.path.join(pkg_dir, d))
| for d in (
| 'build_temp',
| 'build_lib',
| 'bdist_base',
| 'build_scripts',
| 'build_base',
| )
| ]
|
| for name, path in dirs:
| os.mkdir(path)
| setattr(cmd, name, path)
| if name == 'build_base':
| continue
| for f in ('one', 'two', 'three'):
| self.write_file(os.path.join(path, f))
|
| # let's run the command
| cmd.all = 1
| cmd.ensure_finalized()
| cmd.run()
|
| # make sure the files where removed
| for _name, path in dirs:
| assert not os.path.exists(path), f'{path} was not removed'
|
| # let's run the command again (should spit warnings but succeed)
| cmd.all = 1
| cmd.ensure_finalized()
| cmd.run()
|
|