from __future__ import absolute_import

from __future__ import unicode_literals

import argparse

import mock

import pytest

from pre_commit import main

from pre_commit.error_handler import PreCommitSystemExit

from pre_commit.util import cwd

from testing.auto_namedtuple import auto_namedtuple

@pytest.yield_fixture

def mock_commands():

with mock.patch.object(main, 'autoupdate') as autoupdate_mock:

with mock.patch.object(main, 'clean') as clean_mock:

with mock.patch.object(main, 'install') as install_mock:

with mock.patch.object(main, 'uninstall') as uninstall_mock:

with mock.patch.object(main, 'run') as run_mock:

yield auto_namedtuple(

autoupdate_mock=autoupdate_mock,

clean_mock=clean_mock,

install_mock=install_mock,

uninstall_mock=uninstall_mock,

run_mock=run_mock,

)

class CalledExit(Exception):

pass

@pytest.yield_fixture

def argparse_exit_mock():

with mock.patch.object(

argparse.ArgumentParser, 'exit', side_effect=CalledExit,

) as exit_mock:

yield exit_mock

@pytest.yield_fixture

def argparse_parse_args_spy():

parse_args_mock = mock.Mock()

original_parse_args = argparse.ArgumentParser.parse_args

def fake_parse_args(self, args):

# call our spy object

parse_args_mock(args)

return original_parse_args(self, args)

with mock.patch.object(

argparse.ArgumentParser, 'parse_args', fake_parse_args,

):

yield parse_args_mock

def assert_only_one_mock_called(mock_objs):

total_call_count = sum(mock_obj.call_count for mock_obj in mock_objs)

assert total_call_count == 1

def test_overall_help(mock_commands, argparse_exit_mock):

with pytest.raises(CalledExit):

main.main(['--help'])

def test_help_command(

mock_commands, argparse_exit_mock, argparse_parse_args_spy,

):

with pytest.raises(CalledExit):

main.main(['help'])

argparse_parse_args_spy.assert_has_calls([

mock.call(['help']),

mock.call(['--help']),

])

def test_help_other_command(

mock_commands, argparse_exit_mock, argparse_parse_args_spy,

):

with pytest.raises(CalledExit):

main.main(['help', 'run'])

argparse_parse_args_spy.assert_has_calls([

mock.call(['help', 'run']),

mock.call(['run', '--help']),

])

def test_install_command(mock_commands):

main.main(['install'])

assert mock_commands.install_mock.call_count == 1

assert_only_one_mock_called(mock_commands)

def test_uninstall_command(mock_commands):

main.main(['uninstall'])

assert mock_commands.uninstall_mock.call_count == 1

assert_only_one_mock_called(mock_commands)

def test_clean_command(mock_commands):

main.main(['clean'])

assert mock_commands.clean_mock.call_count == 1

assert_only_one_mock_called(mock_commands)

def test_autoupdate_command(mock_commands):

main.main(['autoupdate'])

assert mock_commands.autoupdate_mock.call_count == 1

assert_only_one_mock_called(mock_commands)

def test_run_command(mock_commands):

main.main(['run'])

assert mock_commands.run_mock.call_count == 1

assert_only_one_mock_called(mock_commands)

def test_no_commands_run_command(mock_commands):

main.main([])

assert mock_commands.run_mock.call_count == 1

assert_only_one_mock_called(mock_commands)

def test_help_cmd_in_empty_directory(

mock_commands,

tempdir_factory,

argparse_exit_mock,

argparse_parse_args_spy,

):

path = tempdir_factory.get()

with cwd(path):

with pytest.raises(CalledExit):

main.main(['help', 'run'])

argparse_parse_args_spy.assert_has_calls([

mock.call(['help', 'run']),

mock.call(['run', '--help']),

])

def test_expected_fatal_error_no_git_repo(

tempdir_factory, cap_out, mock_out_store_directory,

):

with cwd(tempdir_factory.get()):

with pytest.raises(PreCommitSystemExit):

main.main([])

assert cap_out.get() == (

'An error has occurred: FatalError: git failed. '

'Is it installed, and are you in a Git repository directory?\n'

'Check the log at ~/.pre-commit/pre-commit.log\n'

)