OwlCyberSecurity - MANAGER

Edit File: test_clean.py

"""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()