from __future__ import absolute_import
import mozunit
from mozversioncontrol import get_repository_object
def test_context_manager(repo):
is_git = repo.vcs == 'git'
cmd = ['show', '--no-patch'] if is_git else ['tip']
vcs = get_repository_object(repo.strpath)
output_subprocess = vcs._run(*cmd)
assert is_git or vcs._client.server is None
assert "Initial commit" in output_subprocess
with vcs:
assert is_git or vcs._client.server is not None
output_client = vcs._run(*cmd)
assert is_git or vcs._client.server is None
assert output_subprocess == output_client
if __name__ == '__main__':
mozunit.main()