python/mozversioncontrol/test/test_branch.py
author Norisz Fay <nfay@mozilla.com>
Fri, 21 Jan 2022 11:26:20 +0200
changeset 605083 6ca2ae7f66684fae2b65257496074d7f2b0510b3
parent 575992 57833f8a204c40f9be0239a430e779a06c8ac604
permissions -rw-r--r--
Merge autoland to mozilla-central. a=merge

# 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/.

from distutils.version import LooseVersion

import mozunit
import pytest

from mozversioncontrol import get_repository_object


STEPS = {
    "hg": [
        """
        hg bookmark test
        """,
        """
        echo "bar" > foo
        hg commit -m "second commit"
        """,
    ],
    "git": [
        """
        git checkout -b test
        """,
        """
        echo "bar" > foo
        git commit -a -m "second commit"
        """,
    ],
}


def test_branch(repo):
    vcs = get_repository_object(repo.strpath)
    if vcs.name == "git" and LooseVersion(vcs.tool_version) < LooseVersion("2.22.0"):
        pytest.xfail("`git branch --show-current` not implemented yet")

    if vcs.name == "git":
        assert vcs.branch == "master"
    else:
        assert vcs.branch is None

    next(repo.step)
    assert vcs.branch == "test"

    next(repo.step)
    assert vcs.branch == "test"

    vcs.update(vcs.head_ref)
    assert vcs.branch is None

    vcs.update("test")
    assert vcs.branch == "test"


if __name__ == "__main__":
    mozunit.main()