Red-DiscordBot/tests/cogs/audio/test_manager.py

77 lines
2.8 KiB
Python

import itertools
import pytest
from redbot.cogs.audio.manager import LavalinkOldVersion, LavalinkVersion
ORDERED_VERSIONS = [
LavalinkOldVersion("3.3.2.3", build_number=1239),
LavalinkOldVersion("3.4.0", build_number=1275),
LavalinkOldVersion("3.4.0", build_number=1350),
# LavalinkVersion is always newer than LavalinkOldVersion
LavalinkVersion(3, 3),
LavalinkVersion(3, 4),
LavalinkVersion(3, 5, rc=1),
LavalinkVersion(3, 5, rc=2),
LavalinkVersion(3, 5, rc=3),
# version with `+red.N` build number is newer than an equivalent upstream version
LavalinkVersion(3, 5, rc=3, red=1),
LavalinkVersion(3, 5, rc=3, red=2),
# all RC versions (including ones with `+red.N`) are older than a stable version
LavalinkVersion(3, 5),
# version with `+red.N` build number is newer than an equivalent upstream version
LavalinkVersion(3, 5, red=1),
LavalinkVersion(3, 5, red=2),
# but newer version number without `+red.N` is still newer
LavalinkVersion(3, 5, 1),
]
@pytest.mark.parametrize(
"raw_version,raw_build_number,expected",
(
# 3-segment version number
("3.4.0", "1350", LavalinkOldVersion("3.4.0", build_number=1350)),
# 4-segment version number
("3.3.2.3", "1239", LavalinkOldVersion("3.3.2.3", build_number=1239)),
# 3-segment version number with 3-digit build number
("3.3.1", "987", LavalinkOldVersion("3.3.1", build_number=987)),
),
)
def test_old_ll_version_parsing(
raw_version: str, raw_build_number: str, expected: LavalinkOldVersion
) -> None:
line = b"Version: %b\nBuild: %b" % (raw_version.encode(), raw_build_number.encode())
assert LavalinkOldVersion.from_version_output(line)
@pytest.mark.parametrize(
"raw_version,expected",
(
# older version format that allowed stripped `.0` and no dot in `rc.4`, used until LL 3.6
("3.5-rc4", LavalinkVersion(3, 5, rc=4)),
("3.5", LavalinkVersion(3, 5)),
# newer version format
("3.6.0-rc.1", LavalinkVersion(3, 6, 0, rc=1)),
# downstream RC version with `+red.N` suffix
("3.7.5-rc.1+red.1", LavalinkVersion(3, 7, 5, rc=1, red=1)),
("3.7.5-rc.1+red.123", LavalinkVersion(3, 7, 5, rc=1, red=123)),
# upstream stable version
("3.7.5", LavalinkVersion(3, 7, 5)),
# downstream stable version with `+red.N` suffix
("3.7.5+red.1", LavalinkVersion(3, 7, 5, red=1)),
("3.7.5+red.123", LavalinkVersion(3, 7, 5, red=123)),
),
)
def test_ll_version_parsing(raw_version: str, expected: LavalinkVersion) -> None:
line = b"Version: " + raw_version.encode()
assert LavalinkVersion.from_version_output(line)
def test_ll_version_comparison() -> None:
it1, it2 = itertools.tee(ORDERED_VERSIONS)
next(it2, None)
for a, b in zip(it1, it2):
assert a < b