36 lines
977 B
Python
36 lines
977 B
Python
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
import mozunit
|
|
|
|
from mozversioncontrol import get_repository_object
|
|
|
|
|
|
def test_context_manager(repo):
|
|
cmd = {
|
|
"git": ["show", "--no-patch"],
|
|
"hg": ["tip"],
|
|
"jj": ["show", "@-"],
|
|
"src": ["echo", "src"],
|
|
}[repo.vcs]
|
|
|
|
vcs = get_repository_object(repo.dir)
|
|
output_subprocess = vcs._run(*cmd)
|
|
if repo.vcs == "hg":
|
|
assert vcs._client.server is None
|
|
if repo.vcs != "src":
|
|
assert "Initial commit" in output_subprocess
|
|
|
|
with vcs:
|
|
if repo.vcs == "hg":
|
|
assert vcs._client.server is not None
|
|
output_client = vcs._run(*cmd)
|
|
|
|
if repo.vcs == "hg":
|
|
assert vcs._client.server is None
|
|
assert output_subprocess == output_client
|
|
|
|
|
|
if __name__ == "__main__":
|
|
mozunit.main()
|