from __future__ import unicode_literals

import os.path

import stat

import subprocess

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 cmd_output_p

from pre_commit.util import parse_version

from pre_commit.util import rmtree

from pre_commit.util import tmpdir

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'

)

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)'

)

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('dne', retcode=None)

assert ret == 1

assert out == 'Executable `dne` not found'

def test_cmd_output_p_exe_not_found():

ret, out, _ = cmd_output_p('dne', retcode=None, stderr=subprocess.STDOUT)

assert ret == 1

assert out == b'Executable `dne` not found'

def test_parse_version():

assert parse_version('0.0') == parse_version('0.0')

assert parse_version('0.1') > parse_version('0.0')

assert parse_version('2.1') >= parse_version('2')

def test_rmtree_read_only_directories(tmpdir):

"""Simulates the go module tree. See #1042"""

tmpdir.join('x/y/z').ensure_dir().join('a').ensure()

mode = os.stat(str(tmpdir.join('x'))).st_mode

mode_no_w = mode & ~(stat.S_IWUSR | stat.S_IWGRP | stat.S_IWOTH)

tmpdir.join('x/y/z').chmod(mode_no_w)

tmpdir.join('x/y/z').chmod(mode_no_w)

tmpdir.join('x/y/z').chmod(mode_no_w)

rmtree(str(tmpdir.join('x')))