mirror of
https://github.com/PyCQA/flake8.git
synced 2026-04-03 11:56:52 +00:00
59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
"""Module containing the tests for flake8.hooks."""
|
|
import os
|
|
import unittest
|
|
|
|
try:
|
|
from unittest import mock
|
|
except ImportError:
|
|
import mock
|
|
|
|
import flake8.hooks
|
|
from flake8.util import is_windows
|
|
|
|
|
|
def excluded(filename):
|
|
return filename.endswith('afile.py')
|
|
|
|
|
|
class TestGitHook(unittest.TestCase):
|
|
if is_windows:
|
|
# On Windows, absolute paths start with a drive letter, for example C:
|
|
# Here we build a fake absolute path starting with the current drive
|
|
# letter, for example C:\fake\temp
|
|
current_drive, ignore_tail = os.path.splitdrive(os.getcwd())
|
|
fake_abs_path = os.path.join(current_drive, os.path.sep, 'fake', 'tmp')
|
|
else:
|
|
fake_abs_path = os.path.join(os.path.sep, 'fake', 'tmp')
|
|
|
|
@mock.patch('os.makedirs')
|
|
@mock.patch('flake8.hooks.open', create=True)
|
|
@mock.patch('shutil.rmtree')
|
|
@mock.patch('tempfile.mkdtemp', return_value=fake_abs_path)
|
|
@mock.patch('flake8.hooks.run',
|
|
return_value=(None,
|
|
[os.path.join('foo', 'afile.py'),
|
|
os.path.join('foo', 'bfile.py')],
|
|
None))
|
|
@mock.patch('flake8.hooks.get_style_guide')
|
|
def test_prepends_tmp_directory_to_exclude(self, get_style_guide, run,
|
|
*args):
|
|
style_guide = get_style_guide.return_value = mock.Mock()
|
|
style_guide.options.exclude = [os.path.join('foo', 'afile.py')]
|
|
style_guide.options.filename = [os.path.join('foo', '*')]
|
|
style_guide.excluded = excluded
|
|
|
|
flake8.hooks.git_hook()
|
|
|
|
dirname, filename = os.path.split(
|
|
os.path.abspath(os.path.join('foo', 'bfile.py')))
|
|
if is_windows:
|
|
# In Windows, the absolute path in dirname will start with a drive
|
|
# letter. Here, we discad the drive letter.
|
|
ignore_drive, dirname = os.path.splitdrive(dirname)
|
|
tmpdir = os.path.join(self.fake_abs_path, dirname[1:])
|
|
tmpfile = os.path.join(tmpdir, 'bfile.py')
|
|
style_guide.check_files.assert_called_once_with([tmpfile])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|