mirror of
https://github.com/pre-commit/pre-commit.git
synced 2026-01-14 04:50:20 -06:00
117 lines
3.2 KiB
Python
117 lines
3.2 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import unicode_literals
|
|
|
|
import io
|
|
import mock
|
|
import os
|
|
import os.path
|
|
import pytest
|
|
from plumbum import local
|
|
|
|
import pre_commit.constants as C
|
|
from pre_commit import five
|
|
from pre_commit.prefixed_command_runner import PrefixedCommandRunner
|
|
from pre_commit.runner import Runner
|
|
from pre_commit.store import Store
|
|
from testing.fixtures import make_consuming_repo
|
|
|
|
|
|
git = local['git']
|
|
|
|
|
|
@pytest.yield_fixture
|
|
def tmpdir_factory(tmpdir):
|
|
class TmpdirFactory(object):
|
|
def __init__(self):
|
|
self.tmpdir_count = 0
|
|
|
|
def get(self):
|
|
path = os.path.join(tmpdir.strpath, five.text(self.tmpdir_count))
|
|
self.tmpdir_count += 1
|
|
os.mkdir(path)
|
|
return path
|
|
|
|
yield TmpdirFactory()
|
|
|
|
|
|
@pytest.yield_fixture
|
|
def in_tmpdir(tmpdir_factory):
|
|
path = tmpdir_factory.get()
|
|
with local.cwd(path):
|
|
yield path
|
|
|
|
|
|
@pytest.yield_fixture
|
|
def in_merge_conflict(tmpdir_factory):
|
|
path = make_consuming_repo(tmpdir_factory, 'script_hooks_repo')
|
|
with local.cwd(path):
|
|
local['touch']('dummy')
|
|
git('add', 'dummy')
|
|
git('add', C.CONFIG_FILE)
|
|
git('commit', '-m', 'Add config.')
|
|
|
|
conflict_path = tmpdir_factory.get()
|
|
git('clone', path, conflict_path)
|
|
with local.cwd(conflict_path):
|
|
git('checkout', 'origin/master', '-b', 'foo')
|
|
with io.open('conflict_file', 'w') as conflict_file:
|
|
conflict_file.write('herp\nderp\n')
|
|
git('add', 'conflict_file')
|
|
with io.open('foo_only_file', 'w') as foo_only_file:
|
|
foo_only_file.write('foo')
|
|
git('add', 'foo_only_file')
|
|
git('commit', '-m', 'conflict_file')
|
|
git('checkout', 'origin/master', '-b', 'bar')
|
|
with io.open('conflict_file', 'w') as conflict_file:
|
|
conflict_file.write('harp\nddrp\n')
|
|
git('add', 'conflict_file')
|
|
with io.open('bar_only_file', 'w') as bar_only_file:
|
|
bar_only_file.write('bar')
|
|
git('add', 'bar_only_file')
|
|
git('commit', '-m', 'conflict_file')
|
|
git('merge', 'foo', retcode=None)
|
|
yield os.path.join(conflict_path)
|
|
|
|
|
|
@pytest.yield_fixture(scope='session', autouse=True)
|
|
def dont_write_to_home_directory():
|
|
"""pre_commit.store.Store will by default write to the home directory
|
|
We'll mock out `Store.get_default_directory` to raise invariantly so we
|
|
don't construct a `Store` object that writes to our home directory.
|
|
"""
|
|
class YouForgotToExplicitlyChooseAStoreDirectory(AssertionError):
|
|
pass
|
|
|
|
with mock.patch.object(
|
|
Store,
|
|
'get_default_directory',
|
|
side_effect=YouForgotToExplicitlyChooseAStoreDirectory,
|
|
):
|
|
yield
|
|
|
|
|
|
@pytest.yield_fixture
|
|
def mock_out_store_directory(tmpdir_factory):
|
|
tmpdir = tmpdir_factory.get()
|
|
with mock.patch.object(
|
|
Store,
|
|
'get_default_directory',
|
|
return_value=tmpdir,
|
|
):
|
|
yield tmpdir
|
|
|
|
|
|
@pytest.yield_fixture
|
|
def store(tmpdir_factory):
|
|
yield Store(os.path.join(tmpdir_factory.get(), '.pre-commit'))
|
|
|
|
|
|
@pytest.yield_fixture
|
|
def cmd_runner(tmpdir_factory):
|
|
yield PrefixedCommandRunner(tmpdir_factory.get())
|
|
|
|
|
|
@pytest.yield_fixture
|
|
def runner_with_mocked_store(mock_out_store_directory):
|
|
yield Runner('/')
|