mirror of
https://github.com/dcs-liberation/dcs_liberation.git
synced 2025-11-10 14:22:26 +00:00
08abe36443
The dataclass contructor will not automatically convert the int in the JSON file to the enum type, so our enum equivalence check was not actually working, and could result in us re-allocating a TACAN channel that was used by the map. Fixing this problem surfaces a latent bug, where we can't actually treat duplicate map TACAN channels as a bug because some channels are used by multiple airports in PG.
102 lines
3.6 KiB
Python
102 lines
3.6 KiB
Python
import pytest
|
|
|
|
from game.radio.tacan import (
|
|
OutOfTacanChannelsError,
|
|
TacanBand,
|
|
TacanChannel,
|
|
TacanRegistry,
|
|
TacanUsage,
|
|
)
|
|
|
|
ALL_VALID_X_TR = [1, *range(31, 46 + 1), *range(64, 126 + 1)]
|
|
ALL_VALID_X_A2A = [*range(37, 63 + 1), *range(100, 126 + 1)]
|
|
|
|
|
|
def test_allocate_first_few_channels() -> None:
|
|
registry = TacanRegistry()
|
|
chan1 = registry.alloc_for_band(TacanBand.X, TacanUsage.TransmitReceive)
|
|
chan2 = registry.alloc_for_band(TacanBand.X, TacanUsage.TransmitReceive)
|
|
chan3 = registry.alloc_for_band(TacanBand.X, TacanUsage.TransmitReceive)
|
|
assert chan1 == TacanChannel(1, TacanBand.X)
|
|
assert chan2 == TacanChannel(31, TacanBand.X)
|
|
assert chan3 == TacanChannel(32, TacanBand.X)
|
|
|
|
|
|
def test_allocate_different_usages() -> None:
|
|
"""Make sure unallocated channels for one use don't make channels unavailable for other usage"""
|
|
registry = TacanRegistry()
|
|
|
|
chanA2AX = registry.alloc_for_band(TacanBand.X, TacanUsage.AirToAir)
|
|
chanA2AY = registry.alloc_for_band(TacanBand.Y, TacanUsage.AirToAir)
|
|
assert chanA2AX == TacanChannel(37, TacanBand.X)
|
|
assert chanA2AY == TacanChannel(37, TacanBand.Y)
|
|
|
|
chanTRX = registry.alloc_for_band(TacanBand.X, TacanUsage.TransmitReceive)
|
|
chanTRY = registry.alloc_for_band(TacanBand.Y, TacanUsage.TransmitReceive)
|
|
assert chanTRX == TacanChannel(1, TacanBand.X)
|
|
assert chanTRY == TacanChannel(1, TacanBand.Y)
|
|
|
|
|
|
def test_reserve_all_valid_transmit_receive() -> None:
|
|
registry = TacanRegistry()
|
|
print("All valid x", ALL_VALID_X_TR)
|
|
|
|
for num in ALL_VALID_X_TR:
|
|
registry.mark_unavailable(TacanChannel(num, TacanBand.X))
|
|
|
|
with pytest.raises(OutOfTacanChannelsError):
|
|
registry.alloc_for_band(TacanBand.X, TacanUsage.TransmitReceive)
|
|
|
|
# Check that we still can allocate an a2a channel even
|
|
# though the T/R channels are used up
|
|
chanA2A = registry.alloc_for_band(TacanBand.X, TacanUsage.AirToAir)
|
|
assert chanA2A == TacanChannel(47, TacanBand.X)
|
|
|
|
|
|
def test_reserve_all_valid_a2a() -> None:
|
|
registry = TacanRegistry()
|
|
print("All valid x", ALL_VALID_X_A2A)
|
|
|
|
for num in ALL_VALID_X_A2A:
|
|
registry.mark_unavailable(TacanChannel(num, TacanBand.X))
|
|
|
|
with pytest.raises(OutOfTacanChannelsError):
|
|
registry.alloc_for_band(TacanBand.X, TacanUsage.AirToAir)
|
|
|
|
# Check that we still can allocate an a2a channel even
|
|
# though the T/R channels are used up
|
|
chanTR = registry.alloc_for_band(TacanBand.X, TacanUsage.TransmitReceive)
|
|
assert chanTR == TacanChannel(1, TacanBand.X)
|
|
|
|
|
|
def test_reserve_again() -> None:
|
|
registry = TacanRegistry()
|
|
registry.mark_unavailable(TacanChannel(1, TacanBand.X))
|
|
registry.mark_unavailable(TacanChannel(1, TacanBand.X))
|
|
|
|
|
|
def test_tacan_parsing() -> None:
|
|
assert TacanChannel.parse("1X") == TacanChannel(1, TacanBand.X)
|
|
assert TacanChannel.parse("1Y") == TacanChannel(1, TacanBand.Y)
|
|
assert TacanChannel.parse("10X") == TacanChannel(10, TacanBand.X)
|
|
assert TacanChannel.parse("100X") == TacanChannel(100, TacanBand.X)
|
|
|
|
with pytest.raises(ValueError):
|
|
TacanChannel.parse("1000X")
|
|
with pytest.raises(ValueError):
|
|
TacanChannel.parse("0X")
|
|
with pytest.raises(ValueError):
|
|
TacanChannel.parse("1Z")
|
|
with pytest.raises(ValueError):
|
|
TacanChannel.parse("X")
|
|
with pytest.raises(ValueError):
|
|
TacanChannel.parse("1")
|
|
with pytest.raises(ValueError):
|
|
TacanChannel.parse("1 X")
|
|
with pytest.raises(ValueError):
|
|
TacanChannel.parse(" 1X")
|
|
with pytest.raises(ValueError):
|
|
TacanChannel.parse("1X ")
|
|
with pytest.raises(ValueError):
|
|
TacanChannel.parse("1x")
|