summaryrefslogtreecommitdiffstats
path: root/python/mozversioncontrol/test/test_context_manager.py
blob: 3186a144d92438bf4a1be239f5fc0f98fcf32ef4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# 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):
    is_git = repo.vcs == "git"
    cmd = ["show", "--no-patch"] if is_git else ["tip"]

    vcs = get_repository_object(repo.dir)
    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()