tyler.durden/powerline
1
0
Fork 0
mirror of https://github.com/powerline/powerline.git synced 2025-07-31 01:35:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix git/hg repo tests

Browse Source
This commit is contained in:
ZyX 2013-03-10 15:34:27 +04:00 committed by Kim Silkebækken
parent a6d48232b0
commit 4374e1f71c
1 changed files with 31 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
tests/test_lib.py
View File

@ -45,93 +45,99 @@ use_mercurial = use_bzr = sys.version_info < (3, 0)
class TestVCS(TestCase): class TestVCS(TestCase):
def test_git(self): def test_git(self):
repo = guess(path='git_repo') repo = guess(path=GIT_REPO)
self.assertNotEqual(repo, None) self.assertNotEqual(repo, None)
self.assertEqual(repo.branch(), 'master') self.assertEqual(repo.branch(), 'master')
self.assertEqual(repo.status(), ' ') self.assertEqual(repo.status(), ' ')
self.assertEqual(repo.status('file'), None) self.assertEqual(repo.status('file'), None)
with open(os.path.join('git_repo', 'file'), 'w') as f: with open(os.path.join(GIT_REPO, 'file'), 'w') as f:
f.write('abc') f.write('abc')
f.flush() f.flush()
self.assertEqual(repo.status(), ' U') self.assertEqual(repo.status(), ' U')
self.assertEqual(repo.status('file'), '??') self.assertEqual(repo.status('file'), '??')
call(['git', 'add', '.'], cwd='git_repo') call(['git', 'add', '.'], cwd=GIT_REPO)
self.assertEqual(repo.status(), ' I ') self.assertEqual(repo.status(), ' I ')
self.assertEqual(repo.status('file'), 'A ') self.assertEqual(repo.status('file'), 'A ')
f.write('def') f.write('def')
f.flush() f.flush()
self.assertEqual(repo.status(), 'DI ') self.assertEqual(repo.status(), 'DI ')
self.assertEqual(repo.status('file'), 'AM') self.assertEqual(repo.status('file'), 'AM')
os.remove(os.path.join('git_repo', 'file')) os.remove(os.path.join(GIT_REPO, 'file'))
if use_mercurial: if use_mercurial:
def test_mercurial(self): def test_mercurial(self):
repo = guess(path='hg_repo') repo = guess(path=HG_REPO)
self.assertNotEqual(repo, None) self.assertNotEqual(repo, None)
self.assertEqual(repo.branch(), 'default') self.assertEqual(repo.branch(), 'default')
with open(os.path.join('hg_repo', 'file'), 'w') as f: with open(os.path.join(HG_REPO, 'file'), 'w') as f:
f.write('abc') f.write('abc')
f.flush() f.flush()
self.assertEqual(repo.status(), ' U') self.assertEqual(repo.status(), ' U')
self.assertEqual(repo.status('file'), 'U') self.assertEqual(repo.status('file'), 'U')
call(['hg', 'add', '.'], cwd='hg_repo', stdout=PIPE) call(['hg', 'add', '.'], cwd=HG_REPO, stdout=PIPE)
self.assertEqual(repo.status(), 'D ') self.assertEqual(repo.status(), 'D ')
self.assertEqual(repo.status('file'), 'A') self.assertEqual(repo.status('file'), 'A')
os.remove(os.path.join('hg_repo', 'file')) os.remove(os.path.join(HG_REPO, 'file'))
if use_bzr: if use_bzr:
def test_bzr(self): def test_bzr(self):
repo = guess(path='bzr_repo') repo = guess(path=BZR_REPO)
self.assertNotEqual(repo, None, 'No bzr repo found. Do you have bzr installed?') self.assertNotEqual(repo, None, 'No bzr repo found. Do you have bzr installed?')
self.assertEqual(repo.branch(), 'test_powerline') self.assertEqual(repo.branch(), 'test_powerline')
self.assertEqual(repo.status(), None) self.assertEqual(repo.status(), None)
with open(os.path.join('bzr_repo', 'file'), 'w') as f: with open(os.path.join(BZR_REPO, 'file'), 'w') as f:
f.write('abc') f.write('abc')
self.assertEqual(repo.status(), ' U') self.assertEqual(repo.status(), ' U')
self.assertEqual(repo.status('file'), '? ') self.assertEqual(repo.status('file'), '? ')
call(['bzr', 'add', '.'], cwd='bzr_repo', stdout=PIPE) call(['bzr', 'add', '.'], cwd=BZR_REPO, stdout=PIPE)
self.assertEqual(repo.status(), 'D ') self.assertEqual(repo.status(), 'D ')
self.assertEqual(repo.status('file'), '+N') self.assertEqual(repo.status('file'), '+N')
call(['bzr', 'commit', '-m', 'initial commit'], cwd='bzr_repo', stdout=PIPE, stderr=PIPE) call(['bzr', 'commit', '-m', 'initial commit'], cwd=BZR_REPO, stdout=PIPE, stderr=PIPE)
self.assertEqual(repo.status(), None) self.assertEqual(repo.status(), None)
with open(os.path.join('bzr_repo', 'file'), 'w') as f: with open(os.path.join(BZR_REPO, 'file'), 'w') as f:
f.write('def') f.write('def')
self.assertEqual(repo.status(), 'D ') self.assertEqual(repo.status(), 'D ')
self.assertEqual(repo.status('file'), ' M') self.assertEqual(repo.status('file'), ' M')
self.assertEqual(repo.status('notexist'), None) self.assertEqual(repo.status('notexist'), None)
os.remove(os.path.join('bzr_repo', 'file')) os.remove(os.path.join(BZR_REPO, 'file'))
old_HGRCPATH = None old_HGRCPATH = None
old_cwd = None old_cwd = None
GIT_REPO = 'git_repo' + os.environ.get('PYTHON', '')
HG_REPO = 'hg_repo' + os.environ.get('PYTHON', '')
BZR_REPO = 'bzr_repo' + os.environ.get('PYTHON', '')
def setUpModule(): def setUpModule():
global old_cwd global old_cwd
global old_HGRCPATH global old_HGRCPATH
old_cwd = os.getcwd() old_cwd = os.getcwd()
os.chdir(os.path.dirname(__file__)) os.chdir(os.path.dirname(__file__))
call(['git', 'init', '--quiet', 'git_repo']) call(['git', 'init', '--quiet', GIT_REPO])
call(['git', 'config', '--local', 'user.name', 'Foo'], cwd='git_repo') assert os.path.isdir(GIT_REPO)
call(['git', 'config', '--local', 'user.email', 'bar@example.org'], cwd='git_repo') call(['git', 'config', '--local', 'user.name', 'Foo'], cwd=GIT_REPO)
call(['git', 'commit', '--allow-empty', '--message', 'Initial commit', '--quiet'], cwd='git_repo') call(['git', 'config', '--local', 'user.email', 'bar@example.org'], cwd=GIT_REPO)
call(['git', 'commit', '--allow-empty', '--message', 'Initial commit', '--quiet'], cwd=GIT_REPO)
if use_mercurial: if use_mercurial:
old_HGRCPATH = os.environ.get('HGRCPATH') old_HGRCPATH = os.environ.get('HGRCPATH')
os.environ['HGRCPATH'] = '' os.environ['HGRCPATH'] = ''
call(['hg', 'init', 'hg_repo']) call(['hg', 'init', HG_REPO])
with open(os.path.join('hg_repo', '.hg', 'hgrc'), 'w') as hgrc: with open(os.path.join(HG_REPO, '.hg', 'hgrc'), 'w') as hgrc:
hgrc.write('[ui]\n') hgrc.write('[ui]\n')
hgrc.write('username = Foo <bar@example.org>\n') hgrc.write('username = Foo <bar@example.org>\n')
if use_bzr: if use_bzr:
call(['bzr', 'init', '--quiet', 'bzr_repo']) call(['bzr', 'init', '--quiet', BZR_REPO])
call(['bzr', 'config', 'email=Foo <bar@example.org>'], cwd='bzr_repo') call(['bzr', 'config', 'email=Foo <bar@example.org>'], cwd=BZR_REPO)
call(['bzr', 'config', 'nickname=test_powerline'], cwd='bzr_repo') call(['bzr', 'config', 'nickname=test_powerline'], cwd=BZR_REPO)
call(['bzr', 'config', 'create_signatures=0'], cwd='bzr_repo') call(['bzr', 'config', 'create_signatures=0'], cwd=BZR_REPO)
def tearDownModule(): def tearDownModule():
global old_cwd global old_cwd
global old_HGRCPATH global old_HGRCPATH
for repo_dir in (['git_repo'] + (['hg_repo'] if use_mercurial else []) + (['bzr_repo'] if use_bzr else [])): for repo_dir in [GIT_REPO] + ([HG_REPO] if use_mercurial else []) + ([BZR_REPO] if use_bzr else []):
for root, dirs, files in list(os.walk(repo_dir, topdown=False)): for root, dirs, files in list(os.walk(repo_dir, topdown=False)):
for file in files: for file in files:
os.remove(os.path.join(root, file)) os.remove(os.path.join(root, file))