tests/forbid_new_submodules_test.py

from __future__ import absolute_import

import subprocess

import pytest

from pre_commit_hooks.forbid_new_submodules import main


@pytest.fixture
def git_dir_with_git_dir(tmpdir):
    with tmpdir.as_cwd():
        subprocess.check_call(('git', 'init', '.'))
        subprocess.check_call((
            'git', 'commit', '-m', 'init', '--allow-empty', '--no-gpg-sign',
        ))
        subprocess.check_call(('git', 'init', 'foo'))
        subprocess.check_call(
            ('git', 'commit', '-m', 'init', '--allow-empty', '--no-gpg-sign'),
            cwd=tmpdir.join('foo').strpath,
        )
        yield


@pytest.mark.parametrize(
    'cmd',
    (
        # Actually add the submodule
        ('git', 'submodule', 'add', './foo'),
        # Sneaky submodule add (that doesn't show up in .gitmodules)
        ('git', 'add', 'foo'),
    ),
)
def test_main_new_submodule(git_dir_with_git_dir, capsys, cmd):
    subprocess.check_call(cmd)
    assert main() == 1
    out, _ = capsys.readouterr()
    assert out.startswith('foo: new submodule introduced\n')


def test_main_no_new_submodule(git_dir_with_git_dir):
    open('test.py', 'a+').close()
    subprocess.check_call(('git', 'add', 'test.py'))
    assert main() == 0