"""net_rates: a container's bytes a second, from its cumulative network
counters against the last ones seen for it (#6115). Pure; no DB."""
from datetime import datetime, timedelta, timezone

from plugins.docker.ingest import net_rates

T0 = datetime(2026, 10, 19, 18, 0, tzinfo=timezone.utc)


def _c(name, rx, tx, cid="abc", status="running"):
    return {"name": name, "container_id": cid, "status": status,
            "net_rx_bytes": rx, "net_tx_bytes": tx}


def test_rates_count_from_the_stored_row_and_then_from_each_snapshot():
    before = {"web": (T0, "abc", 1_000, 2_000)}
    snaps = [(T0 + timedelta(seconds=10), [_c("web", 11_000, 52_000)]),
             (T0 + timedelta(seconds=20), [_c("web", 11_000, 62_000)])]
    assert net_rates(before, snaps) == [
        (T0 + timedelta(seconds=10), "web", 1_000.0, 5_000.0),
        (T0 + timedelta(seconds=20), "web", 0.0, 1_000.0),
    ]


def test_no_rate_for_a_new_recreated_or_restarted_container_until_its_next_snapshot():
    before = {"db": (T0, "old", 100, 100), "app": (T0, "abc", 5_000, 5_000)}
    t1, t2 = T0 + timedelta(seconds=10), T0 + timedelta(seconds=20)
    snaps = [(t1, [_c("new", 10, 10), _c("db", 200, 200, cid="fresh"), _c("app", 10, 10)]),
             (t2, [_c("new", 110, 210), _c("db", 300, 300, cid="fresh"), _c("app", 20, 30)])]
    assert net_rates(before, snaps) == [
        (t2, "new", 10.0, 20.0), (t2, "db", 10.0, 10.0), (t2, "app", 1.0, 2.0)]


def test_stopped_containers_and_missing_counters_are_skipped():
    before = {"web": (T0, "abc", 0, 0), "old": (T0, "abc", 0, 0)}
    snaps = [(T0 + timedelta(seconds=10), [_c("web", None, None), _c("old", 9, 9, status="exited")])]
    assert net_rates(before, snaps) == []