Graham McIntire
1da78a80e5
Stages 6, 7, 10 of the HRDPS plan, plus the Elixir read-side merge that makes Rust's `.hrdps.prop` writes show up on /map. Read side: - `ScoresFile.path_for_hrdps/2` + `read_hrdps/2` — companions to the HRRR equivalents. Same decode path; different filename. - `ScoresFile.read_bounds/3` now reads HRRR + HRDPS files and concatenates the cells. Files are disjoint by construction (Grid.hrdps_only_points excludes CONUS) so no de-dup needed. - `ScoresFile.read_point/4` checks `.prop`, then `.hrdps.prop`, then legacy `.ntms` — first hit wins. Cells outside their owning region read as no-data in the other file's grid, so the order doesn't matter for correctness. - `Propagation.warm_cache_and_broadcast/2` reads both files and merges before broadcasting to the cluster ScoreCache. A single missing file (HRDPS pre-cycle, HRRR briefly absent) is now OK — the cache warms with whichever side is available. Activation: - runtime.exs cron: HRDPS seed worker fires at HH:35 of each cycle hour (05:35Z, 11:35Z, 17:35Z, 23:35Z) — 5h after cycle, 30 min staggered from HRRR's */15 schedule. - HrdpsGridWorker moduledoc updated: no longer dormant; Rust prop-grid-rs handles source='hrdps' rows. - map_live North America bbox extended to (24.5, 60.0, -141.0, -52.0) for the out-of-coverage banner check; visitors anywhere in CONUS or the Canadian extent now stay banner-free. Cleanup: - mix hrdps.probe deleted — its risks are retired and the production HrdpsClient covers the same surface. What's still in motion (not blocking the Canadian /map): - Per-valid_time profile + scalar artifacts for HRDPS (would let /weather show Canadian temperature/refractivity layers). Rust pipeline currently skips those so the HRRR-written companion files aren't clobbered — separate stage when /weather UX is decided. - FreshnessMonitor HRDPS staleness tracking — HRDPS has its own cron and a different cadence than HRRR; HRRR-stale logic doesn't apply cleanly. Defer until prod tells us what alarms we actually want.
386 lines
14 KiB
Elixir
386 lines
14 KiB
Elixir
defmodule Microwaveprop.Propagation.ScoresFileTest do
|
|
# async: false — tests mutate the global Application env to redirect
|
|
# the scores dir to a private tmp tree. Running these in parallel
|
|
# would have concurrent tests trample each other's directory setting.
|
|
use ExUnit.Case, async: false
|
|
|
|
alias Microwaveprop.Propagation.Grid
|
|
alias Microwaveprop.Propagation.ScoresFile
|
|
|
|
setup do
|
|
dir =
|
|
Path.join(
|
|
System.tmp_dir!(),
|
|
"scores_file_test_#{System.unique_integer([:positive])}"
|
|
)
|
|
|
|
File.mkdir_p!(dir)
|
|
|
|
prev = Application.get_env(:microwaveprop, :propagation_scores_dir)
|
|
Application.put_env(:microwaveprop, :propagation_scores_dir, dir)
|
|
|
|
on_exit(fn ->
|
|
File.rm_rf!(dir)
|
|
|
|
if prev do
|
|
Application.put_env(:microwaveprop, :propagation_scores_dir, prev)
|
|
else
|
|
Application.delete_env(:microwaveprop, :propagation_scores_dir)
|
|
end
|
|
end)
|
|
|
|
%{dir: dir}
|
|
end
|
|
|
|
describe "path_for/2" do
|
|
test "nests files under base_dir/{band_mhz}/ with an ISO-8601 valid_time", %{dir: dir} do
|
|
path = ScoresFile.path_for(10_000, ~U[2026-04-14 18:00:00Z])
|
|
assert path == Path.join([dir, "10000", "2026-04-14T18:00:00Z.prop"])
|
|
end
|
|
|
|
test "writes with the new .prop extension", %{dir: dir} do
|
|
valid_time = ~U[2026-04-14 18:00:00Z]
|
|
ScoresFile.write!(10_000, valid_time, [%{lat: 25.0, lon: -125.0, score: 50}])
|
|
written = Path.join([dir, "10000", "2026-04-14T18:00:00Z.prop"])
|
|
assert File.exists?(written)
|
|
refute File.exists?(Path.join([dir, "10000", "2026-04-14T18:00:00Z.ntms"]))
|
|
end
|
|
end
|
|
|
|
describe "legacy_path_for/2" do
|
|
test "still points at the old .ntms extension for backwards reads", %{dir: dir} do
|
|
path = ScoresFile.legacy_path_for(10_000, ~U[2026-04-14 18:00:00Z])
|
|
assert path == Path.join([dir, "10000", "2026-04-14T18:00:00Z.ntms"])
|
|
end
|
|
end
|
|
|
|
describe "legacy .ntms file readability" do
|
|
test "read/2 falls through to the legacy .ntms path when .prop is missing", %{dir: dir} do
|
|
valid_time = ~U[2026-04-14 18:00:00Z]
|
|
# Write a valid file at the .prop path, then rename to .ntms to
|
|
# simulate a file written before the rename that survived through
|
|
# a rolling deploy. The decoder must accept the old magic bytes
|
|
# embedded in the header of already-on-disk files as well, but the
|
|
# simpler check — that the file is found by extension — is what
|
|
# this test targets.
|
|
ScoresFile.write!(10_000, valid_time, [%{lat: 25.0, lon: -125.0, score: 77}])
|
|
prop_path = Path.join([dir, "10000", "2026-04-14T18:00:00Z.prop"])
|
|
ntms_path = Path.join([dir, "10000", "2026-04-14T18:00:00Z.ntms"])
|
|
File.rename!(prop_path, ntms_path)
|
|
|
|
assert {:ok, payload} = ScoresFile.read(10_000, valid_time)
|
|
assert payload.band_mhz == 10_000
|
|
end
|
|
|
|
test "read_point/4 accepts a legacy .ntms file with NTMS magic", %{dir: dir} do
|
|
valid_time = ~U[2026-04-14 18:00:00Z]
|
|
ScoresFile.write!(10_000, valid_time, [%{lat: 25.0, lon: -125.0, score: 88}])
|
|
prop_path = Path.join([dir, "10000", "2026-04-14T18:00:00Z.prop"])
|
|
ntms_path = Path.join([dir, "10000", "2026-04-14T18:00:00Z.ntms"])
|
|
|
|
# Flip the 4-byte magic back to the legacy "NTMS" so both the
|
|
# extension and the in-file identifier reflect the pre-rename
|
|
# state. The reader must still serve the cell.
|
|
raw = File.read!(prop_path)
|
|
<<_magic::binary-size(4), rest::binary>> = raw
|
|
File.write!(ntms_path, <<"NTMS", rest::binary>>, [:binary])
|
|
File.rm!(prop_path)
|
|
|
|
assert ScoresFile.read_point(10_000, valid_time, 25.0, -125.0) == 88
|
|
end
|
|
|
|
test "list_valid_times/1 picks up .ntms and .prop files together", %{dir: _dir} do
|
|
t1 = ~U[2026-04-14 17:00:00Z]
|
|
t2 = ~U[2026-04-14 18:00:00Z]
|
|
|
|
ScoresFile.write!(10_000, t1, [%{lat: 25.0, lon: -125.0, score: 10}])
|
|
ScoresFile.write!(10_000, t2, [%{lat: 25.0, lon: -125.0, score: 20}])
|
|
|
|
# Rename the earlier one to .ntms to simulate a mixed-extension
|
|
# directory during a rolling deploy.
|
|
band_dir =
|
|
Path.join([
|
|
Application.fetch_env!(:microwaveprop, :propagation_scores_dir),
|
|
"10000"
|
|
])
|
|
|
|
File.rename!(
|
|
Path.join(band_dir, "2026-04-14T17:00:00Z.prop"),
|
|
Path.join(band_dir, "2026-04-14T17:00:00Z.ntms")
|
|
)
|
|
|
|
times = ScoresFile.list_valid_times(10_000)
|
|
assert length(times) == 2
|
|
assert t1 in times
|
|
assert t2 in times
|
|
end
|
|
end
|
|
|
|
describe "write!/3 + read/2 round trip" do
|
|
test "preserves the score for each scored grid point", %{dir: _dir} do
|
|
valid_time = ~U[2026-04-14 18:00:00Z]
|
|
|
|
scores = [
|
|
%{lat: 25.0, lon: -125.0, score: 10},
|
|
%{lat: 25.0, lon: -124.875, score: 20},
|
|
%{lat: 25.125, lon: -125.0, score: 30}
|
|
]
|
|
|
|
ScoresFile.write!(10_000, valid_time, scores)
|
|
|
|
assert {:ok, payload} = ScoresFile.read(10_000, valid_time)
|
|
assert payload.band_mhz == 10_000
|
|
assert DateTime.compare(payload.valid_time, valid_time) == :eq
|
|
assert payload.lat_min == Grid.bounds().lat_min
|
|
assert payload.lon_min == Grid.bounds().lon_min
|
|
assert payload.step == Grid.step()
|
|
|
|
# The three scored cells we wrote come back at their grid positions.
|
|
assert score_at(payload, 25.0, -125.0) == 10
|
|
assert score_at(payload, 25.0, -124.875) == 20
|
|
assert score_at(payload, 25.125, -125.0) == 30
|
|
end
|
|
|
|
test "unscored cells read back as 255 (no-data sentinel)" do
|
|
valid_time = ~U[2026-04-14 18:00:00Z]
|
|
ScoresFile.write!(10_000, valid_time, [%{lat: 25.0, lon: -125.0, score: 42}])
|
|
|
|
{:ok, payload} = ScoresFile.read(10_000, valid_time)
|
|
assert score_at(payload, 25.0, -125.0) == 42
|
|
# Any cell we didn't write should be the no-data sentinel.
|
|
assert score_at(payload, 26.5, -100.0) == 255
|
|
end
|
|
end
|
|
|
|
describe "read/2 errors" do
|
|
test "returns {:error, :enoent} when no file exists for the band/time" do
|
|
assert {:error, :enoent} = ScoresFile.read(10_000, ~U[2026-04-14 18:00:00Z])
|
|
end
|
|
|
|
test "returns {:error, :invalid_format} for a file that isn't a score grid" do
|
|
path = ScoresFile.path_for(10_000, ~U[2026-04-14 18:00:00Z])
|
|
File.mkdir_p!(Path.dirname(path))
|
|
File.write!(path, "not a score file")
|
|
|
|
assert {:error, :invalid_format} = ScoresFile.read(10_000, ~U[2026-04-14 18:00:00Z])
|
|
end
|
|
end
|
|
|
|
describe "atomic write" do
|
|
test "leaves no .tmp files behind after a successful write", %{dir: dir} do
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 18:00:00Z], [%{lat: 25.0, lon: -125.0, score: 1}])
|
|
tmp_files = find_files_matching(dir, ~r/\.tmp\./)
|
|
assert tmp_files == []
|
|
end
|
|
end
|
|
|
|
describe "list_valid_times/1" do
|
|
test "returns files sorted ascending for a band" do
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 18:00:00Z], [])
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 16:00:00Z], [])
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 17:00:00Z], [])
|
|
# Unrelated band's files shouldn't leak in.
|
|
ScoresFile.write!(24_000, ~U[2026-04-14 20:00:00Z], [])
|
|
|
|
assert ScoresFile.list_valid_times(10_000) == [
|
|
~U[2026-04-14 16:00:00Z],
|
|
~U[2026-04-14 17:00:00Z],
|
|
~U[2026-04-14 18:00:00Z]
|
|
]
|
|
end
|
|
|
|
test "returns an empty list when the band dir doesn't exist" do
|
|
assert ScoresFile.list_valid_times(75_000) == []
|
|
end
|
|
|
|
test "dedupes when a valid_time exists as both `.prop` and `.ntms`" do
|
|
# Simulate the rolling rename window: the new writer just dropped a
|
|
# .prop file; the prior cycle's .ntms file for the same valid_time
|
|
# is still on NFS. A naive implementation surfaces both → the /map
|
|
# timeline renders the same chip twice.
|
|
valid_time = ~U[2026-04-21 22:00:00Z]
|
|
ScoresFile.write!(10_000, valid_time, [])
|
|
|
|
base = Application.fetch_env!(:microwaveprop, :propagation_scores_dir)
|
|
legacy = Path.join([base, "10000", "2026-04-21T22-00-00Z.ntms"])
|
|
:ok = File.cp!(ScoresFile.path_for(10_000, valid_time), legacy)
|
|
|
|
assert ScoresFile.list_valid_times(10_000) == [valid_time]
|
|
end
|
|
end
|
|
|
|
describe "latest_valid_time/0" do
|
|
test "returns the max valid_time across every band on disk" do
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 16:00:00Z], [])
|
|
ScoresFile.write!(24_000, ~U[2026-04-14 18:00:00Z], [])
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 17:00:00Z], [])
|
|
|
|
assert ScoresFile.latest_valid_time() == ~U[2026-04-14 18:00:00Z]
|
|
end
|
|
|
|
test "returns nil on an empty tree" do
|
|
assert ScoresFile.latest_valid_time() == nil
|
|
end
|
|
end
|
|
|
|
describe "read_bounds/3" do
|
|
test "returns every scored cell (no-data excluded) when bounds is nil" do
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 18:00:00Z], [
|
|
%{lat: 25.0, lon: -125.0, score: 10},
|
|
%{lat: 25.125, lon: -125.0, score: 20}
|
|
])
|
|
|
|
scores = ScoresFile.read_bounds(10_000, ~U[2026-04-14 18:00:00Z])
|
|
|
|
assert [
|
|
%{lat: 25.0, lon: -125.0, score: 10},
|
|
%{lat: 25.125, lon: -125.0, score: 20}
|
|
] = Enum.sort_by(scores, & &1.lat)
|
|
end
|
|
|
|
test "restricts results to the bounding box when provided" do
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 18:00:00Z], [
|
|
%{lat: 25.0, lon: -125.0, score: 10},
|
|
%{lat: 49.875, lon: -66.0, score: 90}
|
|
])
|
|
|
|
bounds = %{"south" => 24.0, "north" => 26.0, "west" => -126.0, "east" => -124.0}
|
|
scores = ScoresFile.read_bounds(10_000, ~U[2026-04-14 18:00:00Z], bounds)
|
|
|
|
assert scores == [%{lat: 25.0, lon: -125.0, score: 10}]
|
|
end
|
|
|
|
test "returns an empty list when the file doesn't exist" do
|
|
assert ScoresFile.read_bounds(10_000, ~U[2026-04-14 18:00:00Z]) == []
|
|
end
|
|
|
|
test "merges cells from a sibling .hrdps.prop file when present" do
|
|
# Write the HRRR file via the public writer, then copy it to the
|
|
# HRDPS path. Both files now contain the same cell — read_bounds
|
|
# should return both, doubling the cell list. (Production HRDPS
|
|
# files have a different bbox; this test only verifies the merge
|
|
# plumbing routes through both paths.)
|
|
vt = ~U[2026-04-14 18:00:00Z]
|
|
ScoresFile.write!(10_000, vt, [%{lat: 25.0, lon: -125.0, score: 42}])
|
|
|
|
hrrr_path = ScoresFile.path_for(10_000, vt)
|
|
hrdps_path = ScoresFile.path_for_hrdps(10_000, vt)
|
|
File.cp!(hrrr_path, hrdps_path)
|
|
|
|
scores = ScoresFile.read_bounds(10_000, vt)
|
|
assert length(scores) == 2
|
|
assert Enum.all?(scores, &(&1.score == 42))
|
|
end
|
|
|
|
test "returns HRDPS cells alone when only the .hrdps.prop file exists" do
|
|
vt = ~U[2026-04-14 18:00:00Z]
|
|
# Use write! to land a file at the HRRR path, then move it to
|
|
# the HRDPS path so only the HRDPS file remains.
|
|
ScoresFile.write!(10_000, vt, [%{lat: 25.0, lon: -125.0, score: 80}])
|
|
|
|
File.rename!(
|
|
ScoresFile.path_for(10_000, vt),
|
|
ScoresFile.path_for_hrdps(10_000, vt)
|
|
)
|
|
|
|
assert [%{score: 80}] = ScoresFile.read_bounds(10_000, vt)
|
|
end
|
|
end
|
|
|
|
describe "read_point/4" do
|
|
test "returns the score for a grid cell by (lat, lon) lookup" do
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 18:00:00Z], [
|
|
%{lat: 25.0, lon: -125.0, score: 42}
|
|
])
|
|
|
|
assert ScoresFile.read_point(10_000, ~U[2026-04-14 18:00:00Z], 25.0, -125.0) == 42
|
|
end
|
|
|
|
test "returns nil for cells that weren't scored" do
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 18:00:00Z], [
|
|
%{lat: 25.0, lon: -125.0, score: 42}
|
|
])
|
|
|
|
assert ScoresFile.read_point(10_000, ~U[2026-04-14 18:00:00Z], 40.0, -100.0) == nil
|
|
end
|
|
|
|
test "returns nil when the file is missing" do
|
|
assert ScoresFile.read_point(10_000, ~U[2026-04-14 18:00:00Z], 25.0, -125.0) == nil
|
|
end
|
|
end
|
|
|
|
describe "retain_window/2" do
|
|
test "deletes files outside [run_time, run_time + hours] across every band" do
|
|
run_time = ~U[2026-04-14 16:00:00Z]
|
|
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 15:00:00Z], [])
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 16:00:00Z], [])
|
|
ScoresFile.write!(10_000, ~U[2026-04-14 17:00:00Z], [])
|
|
ScoresFile.write!(24_000, ~U[2026-04-14 14:00:00Z], [])
|
|
ScoresFile.write!(24_000, ~U[2026-04-14 16:00:00Z], [])
|
|
ScoresFile.write!(24_000, ~U[2026-04-15 10:00:00Z], [])
|
|
ScoresFile.write!(24_000, ~U[2026-04-15 11:00:00Z], [])
|
|
|
|
# Window: 16:00 -> 10:00 next day (18 hours forward).
|
|
assert ScoresFile.retain_window(run_time, 18) == 3
|
|
|
|
assert ScoresFile.list_valid_times(10_000) == [
|
|
~U[2026-04-14 16:00:00Z],
|
|
~U[2026-04-14 17:00:00Z]
|
|
]
|
|
|
|
assert ScoresFile.list_valid_times(24_000) == [
|
|
~U[2026-04-14 16:00:00Z],
|
|
~U[2026-04-15 10:00:00Z]
|
|
]
|
|
end
|
|
|
|
test "returns 0 and leaves foreign files alone", %{dir: dir} do
|
|
run_time = ~U[2026-04-14 16:00:00Z]
|
|
stray = Path.join(dir, "some_other_file.txt")
|
|
File.write!(stray, "keep me")
|
|
|
|
assert ScoresFile.retain_window(run_time, 18) == 0
|
|
assert File.exists?(stray)
|
|
end
|
|
end
|
|
|
|
describe "prune_older_than/1" do
|
|
test "deletes files whose valid_time is strictly before the cutoff", %{dir: _dir} do
|
|
old = ~U[2026-04-14 10:00:00Z]
|
|
fresh = ~U[2026-04-14 18:00:00Z]
|
|
|
|
ScoresFile.write!(10_000, old, [%{lat: 25.0, lon: -125.0, score: 1}])
|
|
ScoresFile.write!(10_000, fresh, [%{lat: 25.0, lon: -125.0, score: 2}])
|
|
|
|
cutoff = ~U[2026-04-14 15:00:00Z]
|
|
assert ScoresFile.prune_older_than(cutoff) == 1
|
|
|
|
assert {:error, :enoent} = ScoresFile.read(10_000, old)
|
|
assert {:ok, _} = ScoresFile.read(10_000, fresh)
|
|
end
|
|
|
|
test "leaves foreign files in the scores dir alone", %{dir: dir} do
|
|
stray = Path.join(dir, "some_other_file.txt")
|
|
File.write!(stray, "do not touch")
|
|
|
|
assert ScoresFile.prune_older_than(~U[2099-01-01 00:00:00Z]) == 0
|
|
assert File.exists?(stray)
|
|
end
|
|
end
|
|
|
|
# Look up the score for (lat, lon) in the binary returned by read/2.
|
|
defp score_at(payload, lat, lon) do
|
|
row = round((lat - payload.lat_min) / payload.step)
|
|
col = round((lon - payload.lon_min) / payload.step)
|
|
idx = row * payload.n_cols + col
|
|
:binary.at(payload.scores, idx)
|
|
end
|
|
|
|
defp find_files_matching(dir, pattern) do
|
|
dir
|
|
|> Path.join("**/*")
|
|
|> Path.wildcard()
|
|
|> Enum.filter(fn path -> File.regular?(path) and Regex.match?(pattern, path) end)
|
|
end
|
|
end
|