94 lines
3.6 KiB
Python
94 lines
3.6 KiB
Python
from pathlib import Path
|
||
|
||
from hockey_data.service import HockeyDataService
|
||
|
||
ROOT = Path(__file__).resolve().parents[1]
|
||
APP_JS = (ROOT / "ui_builder/static/app.js").read_text(encoding="utf-8")
|
||
|
||
|
||
def _penalty(side: str, remaining: int) -> dict:
|
||
return {
|
||
"side": side,
|
||
"infraction": {"code": "TEST"},
|
||
"preset": "2",
|
||
"durationMs": 120000,
|
||
"remainingMs": remaining,
|
||
"finished": False,
|
||
}
|
||
|
||
|
||
def test_equal_coincidental_penalties_have_no_side_caption():
|
||
settings = {
|
||
"strength_regulation_skaters": 5,
|
||
"strength_min_skaters": 3,
|
||
"strength_state_labels": {"regulation": {"4x4": {"ru": "4 на 4"}}},
|
||
}
|
||
payload = HockeyDataService._strength_payload(
|
||
settings,
|
||
stage="regular",
|
||
current_period="2",
|
||
timer_state={"penalty_board": {"penalties": [_penalty("home", 100000), _penalty("away", 100000)]}},
|
||
language="ru",
|
||
)
|
||
assert payload["strength_label"] == "4×4"
|
||
assert payload["advantage_side"] == ""
|
||
assert payload["home_label"] == "4 на 4"
|
||
assert payload["away_label"] == "4 на 4"
|
||
|
||
|
||
def test_two_vs_one_penalties_caption_only_advantage_team():
|
||
settings = {
|
||
"strength_regulation_skaters": 5,
|
||
"strength_min_skaters": 3,
|
||
"strength_state_labels": {"regulation": {"4x3": {"ru": "Играют в большинстве"}}},
|
||
}
|
||
payload = HockeyDataService._strength_payload(
|
||
settings,
|
||
stage="regular",
|
||
current_period="2",
|
||
timer_state={"penalty_board": {"penalties": [
|
||
_penalty("home", 103000),
|
||
_penalty("home", 120000),
|
||
_penalty("away", 103000),
|
||
]}},
|
||
language="ru",
|
||
)
|
||
assert payload["strength_label"] == "3×4"
|
||
assert payload["advantage_side"] == "away"
|
||
assert payload["home_label"] == "Играют в большинстве"
|
||
assert payload["away_label"] == "Играют в большинстве"
|
||
|
||
|
||
def test_runtime_plan_uses_one_plate_and_global_shortest_transition_timer():
|
||
start = APP_JS.index("function penaltyDisplayEntriesByTargetSide")
|
||
end = APP_JS.index("function hockeyPenaltySideMappingDetail", start)
|
||
block = APP_JS[start:end]
|
||
assert 'const all = [...home, ...away].sort' in block
|
||
assert 'const transitionEntry = all[0] || null;' in block
|
||
assert 'home: advantageSide === "home" && transitionEntry ? [transitionEntry] : []' in block
|
||
assert 'away: advantageSide === "away" && transitionEntry ? [transitionEntry] : []' in block
|
||
assert 'if (!advantageSide)' in block
|
||
assert 'return { home: [], away: [], routedToAdvantage: false' in block
|
||
|
||
|
||
def test_final_full_strength_action_only_runs_after_last_penalty_and_real_advantage():
|
||
start = APP_JS.index("function fireConfiguredTimerFinishActions")
|
||
end = APP_JS.index("async function runShortcutSequenceStep", start)
|
||
block = APP_JS[start:end]
|
||
assert 'Number(meta.remaining_total || 0) > 0' in block
|
||
assert '!Boolean(meta.had_advantage)' in block
|
||
assert 'full_strength_side' in block
|
||
assert 'side !== fullStrengthSide' in block
|
||
|
||
|
||
def test_rebalance_sends_old_plate_out_before_new_plate_in():
|
||
start = APP_JS.index("async function rebalanceVmixPenaltyTargets")
|
||
end = APP_JS.index("function finishActionMatchesSource", start)
|
||
block = APP_JS[start:end]
|
||
assert 'const outCommands = [];' in block
|
||
assert 'const setCommands = [];' in block
|
||
assert 'const inCommands = [];' in block
|
||
assert 'const stopCommands = [];' in block
|
||
assert 'const runCommands = [];' in block
|
||
assert 'const commands = [...outCommands, ...stopCommands, ...setCommands, ...runCommands, ...inCommands];' in block
|