import contextlib
import os
import pytest

from plumbum import local
from pre_commit import git



@contextlib.contextmanager
def in_dir(dir):
    old_path = local.cwd.getpath()
    local.cwd.chdir(dir)
    try:
        yield
    finally:
        local.cwd.chdir(old_path)

@pytest.yield_fixture
def empty_git_dir(tmpdir):
    with in_dir(tmpdir.strpath):
        local['git']['init']()
        yield tmpdir.strpath


def test_get_root(empty_git_dir):
    assert git.get_root() == empty_git_dir

    foo = local.path('foo')
    foo.mkdir()

    with in_dir(foo):
        assert git.get_root() == empty_git_dir


def test_get_pre_commit_path(empty_git_dir):
    assert git.get_pre_commit_path() == '{0}/.git/hooks/pre-commit'.format(empty_git_dir)


def test_create_pre_commit(empty_git_dir):
    git.create_pre_commit()
    assert len(open(git.get_pre_commit_path(), 'r').read()) > 0


def test_remove_pre_commit(empty_git_dir):
    git.remove_pre_commit()

    assert not os.path.exists(git.get_pre_commit_path())

    git.create_pre_commit()
    git.remove_pre_commit()

    assert not os.path.exists(git.get_pre_commit_path())