fune/python/mozversioncontrol/test/test_get_upstream_remote.py

# 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

STEPS = {
    "hg": [],
    "git": [
        """
        git remote add unified hg::https://hg.mozilla.org/mozilla-unified
        git remote add central hg::https://hg.mozilla.org/central
        """,
        "git remote remove unified",
        "git remote remove central",
    ],
}


def test_get_upstream_remote(repo):
    # Test is only relevant for Git.
    if not repo.vcs == "git":
        return

    vcs = get_repository_object(repo.dir)
    remote = vcs.get_mozilla_upstream_remote()

    repo.execute_next_step()
    remote = vcs.get_mozilla_upstream_remote()
    assert remote == "unified", "Unified remote should be preferred."

    repo.execute_next_step()
    remote = vcs.get_mozilla_upstream_remote()
    assert (
        remote == "central"
    ), "Any other official looking remote should be returned second."

    repo.execute_next_step()
    remote = vcs.get_mozilla_upstream_remote()
    assert (
        remote is None
    ), "`None` should be returned if an official remote isn't found."


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