mirror of
https://github.com/balkian/pre-commit-hooks.git
synced 2024-11-24 16:32:28 +00:00
65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
|
"""
|
||
|
|
Tests for `branchversion` module.
|
||
|
|
"""
|
||
|
|
import pytest
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import unittest
|
||
|
|
from contextlib import contextmanager
|
||
|
|
from subprocess import check_output
|
||
|
|
from balkian_pre_commit import branchversion
|
||
|
|
|
||
|
|
|
||
|
|
TEMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)),
|
||
|
|
'test-precommit')
|
||
|
|
|
||
|
|
|
||
|
|
@contextmanager
|
||
|
|
def cd(newdir):
|
||
|
|
prevdir = os.getcwd()
|
||
|
|
os.chdir(os.path.expanduser(newdir))
|
||
|
|
try:
|
||
|
|
yield
|
||
|
|
finally:
|
||
|
|
os.chdir(prevdir)
|
||
|
|
|
||
|
|
|
||
|
|
class TestBalkian_pre_commit(unittest.TestCase):
|
||
|
|
|
||
|
|
def commit_version(self, version, branch='master'):
|
||
|
|
with cd(TEMP_DIR):
|
||
|
|
check_output(['git', 'checkout', '-B', branch])
|
||
|
|
with open('VERSION', 'w') as f:
|
||
|
|
f.write(version)
|
||
|
|
check_output(['git', 'add', 'VERSION'])
|
||
|
|
check_output(['git', 'commit', '-am', 'Committing to branch {}'.format(branch)])
|
||
|
|
|
||
|
|
def initRepo(self):
|
||
|
|
self.tearDown()
|
||
|
|
os.makedirs(TEMP_DIR)
|
||
|
|
check_output(['git', 'init', TEMP_DIR])
|
||
|
|
with cd(TEMP_DIR):
|
||
|
|
with open('README.md', 'w') as f:
|
||
|
|
f.write('Hello')
|
||
|
|
check_output(['git', 'add', '.'])
|
||
|
|
check_output(['git', 'commit', '-am', 'First commit'])
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self.initRepo()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
if os.path.isdir(TEMP_DIR):
|
||
|
|
shutil.rmtree(TEMP_DIR)
|
||
|
|
|
||
|
|
def test_success(self):
|
||
|
|
self.initRepo()
|
||
|
|
with cd(TEMP_DIR):
|
||
|
|
self.commit_version('0.1.0', 'master')
|
||
|
|
assert branchversion.main(['VERSION',])==0
|
||
|
|
|
||
|
|
def test_fail(self):
|
||
|
|
self.initRepo()
|
||
|
|
with cd(TEMP_DIR):
|
||
|
|
self.commit_version('0.1.0-pre.4', 'master')
|
||
|
|
assert branchversion.main(['VERSION',])==1
|