test_context_manager.py

Enable keyboard shortcuts

# 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"],

        # For jj, remove most of the header to avoid relative timestamps (one

        # output might say "(now)", the next "(1 second ago)" for the same

        # timestamp.)

        "jj": ["show", "@-", "--template", "description ++ '\\n'"],

        "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()