test_is_shallow.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 shutil

import subprocess

import tempfile

from pathlib import Path

import mozunit

import pytest

from mozversioncontrol import get_repository_object

def test_is_shallow_full_clone(repo):

    assert not get_repository_object(repo.dir).is_shallow

@pytest.mark.parametrize("vcs", ["git", "jj"])

def test_is_shallow_shallow_clone(vcs, monkeypatch):

    if vcs == "jj":

        if not shutil.which("jj"):

            pytest.skip("jj unavailable")

        monkeypatch.setenv("JJ_CONFIG", "")

        monkeypatch.delenv("MOZ_AVOID_JJ_VCS", raising=False)

    # Use tempfile since pytest's tempdir is too long for jj on Windows

    with tempfile.TemporaryDirectory(prefix=f"{vcs}-shallow") as td:

        src = Path(td) / "src"

        src.mkdir()

        for cmd in (

            ["git", "init", "-b", "master"],

            ["git", "config", "user.name", "Testing McTesterson"],

            ["git", "config", "user.email", "test@example.org"],

            ["git", "commit", "--allow-empty", "-m", "first"],

            ["git", "commit", "--allow-empty", "-m", "second"],

):

            subprocess.check_call(cmd, cwd=src)

        clone = Path(td) / "clone"

        subprocess.check_call(["git", "clone", "--depth", "1", src.as_uri(), clone])

        if vcs == "jj":

            subprocess.check_call(["jj", "git", "init", "--colocate"], cwd=clone)

        repo = get_repository_object(clone)

        assert repo.name == vcs

        assert repo.is_shallow

if __name__ == "__main__":

    mozunit.main()