nix-update/tests/test_branch.py

#!/usr/bin/env python3
import unittest.mock
from pathlib import Path
from typing import BinaryIO
from urllib.parse import urlparse

import conftest

from nix_update.version import fetch_latest_version
from nix_update.version.version import VersionPreference

TEST_ROOT = Path(__file__).parent.resolve()


def fake_urlopen(url: str) -> BinaryIO:
    if url.endswith("releases.atom"):
        return open(TEST_ROOT.joinpath("test_branch_releases.atom"), "rb")
    else:
        return open(TEST_ROOT.joinpath("test_branch_commits_master.atom"), "rb")


def test_branch(helpers: conftest.Helpers) -> None:
    del helpers
    with unittest.mock.patch("urllib.request.urlopen", fake_urlopen):
        assert (
            fetch_latest_version(
                urlparse("https://github.com/Mic92/nix-update"),
                VersionPreference.BRANCH,
                "(.*)",
                "master",
            ).number
            == "1.2.0-unstable-2024-02-19"
        )