mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-14 13:00:10 -06:00
120 lines
2.9 KiB
Python
120 lines
2.9 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import os.path
|
|
import random
|
|
|
|
import pytest
|
|
|
|
from pre_commit.util import CalledProcessError
|
|
from pre_commit.util import clean_path_on_failure
|
|
from pre_commit.util import cmd_output
|
|
from pre_commit.util import memoize_by_cwd
|
|
from pre_commit.util import tmpdir
|
|
from testing.util import cwd
|
|
|
|
|
|
def test_CalledProcessError_str():
|
|
error = CalledProcessError(
|
|
1, [str('git'), str('status')], 0, (str('stdout'), str('stderr')),
|
|
)
|
|
assert str(error) == (
|
|
"Command: ['git', 'status']\n"
|
|
"Return code: 1\n"
|
|
"Expected return code: 0\n"
|
|
"Output: \n"
|
|
" stdout\n"
|
|
"Errors: \n"
|
|
" stderr\n"
|
|
)
|
|
|
|
|
|
def test_CalledProcessError_str_nooutput():
|
|
error = CalledProcessError(
|
|
1, [str('git'), str('status')], 0, (str(''), str('')),
|
|
)
|
|
assert str(error) == (
|
|
"Command: ['git', 'status']\n"
|
|
"Return code: 1\n"
|
|
"Expected return code: 0\n"
|
|
"Output: (none)\n"
|
|
"Errors: (none)\n"
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def memoized_by_cwd():
|
|
@memoize_by_cwd
|
|
def func(arg):
|
|
return arg + str(random.getrandbits(64))
|
|
|
|
return func
|
|
|
|
|
|
def test_memoized_by_cwd_returns_same_twice_in_a_row(memoized_by_cwd):
|
|
ret = memoized_by_cwd('baz')
|
|
ret2 = memoized_by_cwd('baz')
|
|
assert ret is ret2
|
|
|
|
|
|
def test_memoized_by_cwd_returns_different_for_different_args(memoized_by_cwd):
|
|
ret = memoized_by_cwd('baz')
|
|
ret2 = memoized_by_cwd('bar')
|
|
assert ret.startswith('baz')
|
|
assert ret2.startswith('bar')
|
|
assert ret != ret2
|
|
|
|
|
|
def test_memoized_by_cwd_changes_with_different_cwd(memoized_by_cwd):
|
|
ret = memoized_by_cwd('baz')
|
|
with cwd('.git'):
|
|
ret2 = memoized_by_cwd('baz')
|
|
|
|
assert ret != ret2
|
|
|
|
|
|
def test_clean_on_failure_noop(in_tmpdir):
|
|
with clean_path_on_failure('foo'):
|
|
pass
|
|
|
|
|
|
def test_clean_path_on_failure_does_nothing_when_not_raising(in_tmpdir):
|
|
with clean_path_on_failure('foo'):
|
|
os.mkdir('foo')
|
|
assert os.path.exists('foo')
|
|
|
|
|
|
def test_clean_path_on_failure_cleans_for_normal_exception(in_tmpdir):
|
|
class MyException(Exception):
|
|
pass
|
|
|
|
with pytest.raises(MyException):
|
|
with clean_path_on_failure('foo'):
|
|
os.mkdir('foo')
|
|
raise MyException
|
|
|
|
assert not os.path.exists('foo')
|
|
|
|
|
|
def test_clean_path_on_failure_cleans_for_system_exit(in_tmpdir):
|
|
class MySystemExit(SystemExit):
|
|
pass
|
|
|
|
with pytest.raises(MySystemExit):
|
|
with clean_path_on_failure('foo'):
|
|
os.mkdir('foo')
|
|
raise MySystemExit
|
|
|
|
assert not os.path.exists('foo')
|
|
|
|
|
|
def test_tmpdir():
|
|
with tmpdir() as tempdir:
|
|
assert os.path.exists(tempdir)
|
|
assert not os.path.exists(tempdir)
|
|
|
|
|
|
def test_cmd_output_exe_not_found():
|
|
ret, out, _ = cmd_output('i-dont-exist', retcode=None)
|
|
assert ret == 1
|
|
assert out == 'Executable `i-dont-exist` not found'
|