hyperhive/hive-c0re/src/workers/wanted.rs
iris 513554fe9a swarm: add a declared "paused" agent wanted state
mara (#4170): swarm-ui's wanted-state dropdown could only ever declare
up/offline/destroy, with no way to swarm-declare the existing hive-local
turn-loop pause (`hivectl agent pause|resume`).

`AgentState::Paused` is not a fifth peer of Up/Offline/Destroyed on the
power axis this enum otherwise answers — it's Up plus an orthogonal
turn-loop pause. `hive-c0re`'s `workers::wanted` reconcile loop now
decides the two axes independently (`decide` for power, the new
`decide_pause` for the marker), so a stopped agent declared Paused
converges with both a Start and a Pause in the same pass.

Known, deliberate limitation: a Paused declaration on an agent this
hive has never deployed only reaches Deploy this pass — writing the
pause marker into a harness dir that may not exist yet was judged not
worth the risk, so it converges on the next pass once the agent is
present instead.

swarm-ui's WantedMenu gains a fourth "paused" option (warning-tone
badge). No separate "resume" entry — selecting "up" from a paused row
already clears the marker via the same decide_pause path.

Pause/resume marker writes go through one shared
Coordinator::set_paused_by_name helper, used by both the interactive
dashboard pause/resume handlers and this reconcile loop, instead of
each duplicating the parse-name/write-marker/track-rescan shape.
swarm-ui's "offline" and "paused" confirm dialogs share one
confirmTarget state and one ConfirmDialog instead of two near-identical
copies.

Closes #4170
2026-09-11 01:35:40 +02:00

799 lines
32 KiB
Rust

//! Converging this hive onto the agent set the swarm controller declares.
//!
//! The controller gives each hive its own `hive-wanted-<hive>` bucket (see
//! [`swarm_queue_client::wanted`]); this reads its own and acts on it. It is
//! the **repair** path, not the fast one: a deploy event
//! ([`crate::swarm_status`]) is core NATS, so a hive that was down never hears
//! it, and until this existed that left the hive on its old shape with nothing
//! looking wrong.
//!
//! # Two things a declaration does not mean
//!
//! **Absence is not a deletion order.** An agent missing from the value — or
//! no value, or no bucket — is one the controller has said nothing about.
//! Swarm-side lifecycle does not yet cover agents that predate it, so
//! converging "to exactly this set" would tear down every agent the swarm has
//! not adopted.
//!
//! **An unknown state is not a partial instruction.** [`AgentState`] is closed,
//! so a value this build cannot read fails the whole decode: the hive converges
//! nothing rather than obeying the agents it happened to understand.
//!
//! # What "diverged" is measured against
//!
//! The hive's **stored power intent**, not the container's observed running
//! state. An agent down while its intent says `Up` is already the boot
//! reconcile's work; a loop reading `is_running` would insert a start DAG
//! behind that reconcile's back on every boot.
//!
//! One consequence, deliberate: a declaration outranks a local stop — an agent
//! declared `Up` returns at the next boot, because the controller owns that.
use std::collections::{BTreeMap, BTreeSet};
use std::sync::Arc;
use anyhow::{Context, Result};
use swarm_queue_client::wanted::{AgentState, HiveWanted};
use crate::coordinator::Coordinator;
use crate::power::Wanted;
/// Read this hive's declaration and converge the agents it names.
///
/// `Ok(())` covers three quiet cases that are **not** an empty declaration: no
/// hive name, no bucket, no key. Each logs its own line, because a loop that
/// silently does nothing and one that is working look identical otherwise.
/// Anything else — an unreachable queue, an undecodable value, an agent list
/// this hive cannot enumerate — is an error, so the node fails visibly on the
/// dashboard instead of reading as "nothing was declared".
///
/// # Errors
/// Propagates a failed queue connect, KV read, decode, or agent enumeration.
pub async fn pull(coord: &Arc<Coordinator>) -> Result<()> {
let Some(hive) = crate::container_view::hive_swarm_names().0 else {
tracing::debug!("wanted state: HYPERHIVE_HIVE_NAME unset; no key to read");
return Ok(());
};
let Some(client) = crate::swarm_queue::client().await else {
// Absent or failed — either way already reported by the shared
// connector (an `info` log or a `swarm_queue_config` banner).
return Ok(());
};
// An unconnected client does not fail a JetStream request, it hangs on it.
// A hung node is worse than a failed one: the dashboard shows it running.
swarm_queue_client::ensure_connected(&client)?;
let Some(store) = swarm_queue_client::wanted::open_read_only(&client, &hive).await else {
tracing::info!(%hive, "wanted state: no bucket yet; nothing has been declared");
return Ok(());
};
let Some(raw) = store
.get(&hive)
.await
.context("reading this hive's wanted state")?
else {
tracing::info!(%hive, "wanted state: no key for this hive; nothing has been declared");
return Ok(());
};
let declared: HiveWanted =
serde_json::from_slice(&raw).context("decoding this hive's wanted state")?;
converge(coord, &declared).await
}
/// Converge again each time the controller republishes this hive's declaration.
///
/// The **fast** path, and it does not replace [`pull`]: this hears only what is
/// published while it is listening, so a hive that was down still learns the
/// current declaration from the boot read. Both, not either.
///
/// ⚠️ A refused watch and a quiet one are told apart here, unlike the core-NATS
/// subscriptions in [`crate::swarm_status`]: a watch is a `JetStream` consumer,
/// so the request is answered, and a hive lacking the grant gets `None` rather
/// than silence. That is why this warns instead of returning quietly.
pub async fn watch_declarations(
client: async_nats::Client,
coord: Arc<Coordinator>,
hive: String,
mut shutdown: tokio::sync::watch::Receiver<bool>,
) {
use futures_util::StreamExt as _;
let Some(mut updates) = swarm_queue_client::wanted::watch(&client, &hive).await else {
tracing::warn!(
%hive,
"wanted state: cannot watch this hive's bucket; changes will be \
picked up at the next boot instead"
);
return;
};
tracing::info!(%hive, "wanted state: watching for declarations");
loop {
tokio::select! {
entry = updates.next() => {
let Some(entry) = entry else {
// Same reading as the sibling subscriptions: `async-nats`
// reconnects underneath a live watch, so an ended stream is
// the connection going away for good rather than a blip to
// spin on.
tracing::warn!(%hive, "wanted state: watch closed");
return;
};
match entry {
Ok(entry) => apply_entry(&coord, &hive, &entry).await,
// The watch survives one bad entry; the stream ending is
// the case above.
Err(e) => tracing::warn!(%hive, error = %e, "wanted state: watch error"),
}
}
_ = shutdown.changed() => {
tracing::info!("wanted state: shutdown signal received");
return;
}
}
}
}
/// Converge one watched update, or decline to.
///
/// A delete is **not** a deletion order — the module docs' rule, and the reason
/// this is not simply "decode and converge": the controller removing the key
/// says nothing about the agents this hive runs, so acting on it would tear
/// down the very set that absence is defined not to touch.
async fn apply_entry(
coord: &Arc<Coordinator>,
hive: &str,
entry: &async_nats::jetstream::kv::Entry,
) {
if !carries_a_declaration(entry.operation) {
tracing::info!(%hive, "wanted state: declaration withdrawn; nothing to converge");
return;
}
let declared: HiveWanted = match serde_json::from_slice(&entry.value) {
Ok(declared) => declared,
// Warned rather than propagated: this end and the controller share one
// type, so a decode failure means they disagree about it — and the
// watch must keep running to pick up the next, possibly good, value.
Err(e) => {
tracing::warn!(%hive, error = %e, "wanted state: undecodable declaration");
return;
}
};
if let Err(e) = converge(coord, &declared).await {
tracing::warn!(%hive, error = ?e, "wanted state: converging a watched update failed");
}
}
/// Whether a watched operation carries a declaration to converge to.
///
/// Pure, and separate from [`apply_entry`], so the module's "absence is not a
/// deletion order" rule is enforced by a test rather than only asserted in
/// prose — converging on a removed key is the one mistake here that would tear
/// down agents nobody asked to stop.
fn carries_a_declaration(operation: async_nats::jetstream::kv::Operation) -> bool {
use async_nats::jetstream::kv::Operation;
match operation {
Operation::Put => true,
Operation::Delete | Operation::Purge => false,
}
}
/// Queue whatever the declaration asks for and this hive is not already doing.
async fn converge(coord: &Arc<Coordinator>, declared: &HiveWanted) -> Result<()> {
// Fail closed, for the reason the deploy event's own arm gives: without
// the list, "this hive does not have that agent" cannot be told from
// "this hive cannot see its agents", and acting on the guess creates a
// container over one that already exists.
let present: BTreeSet<String> = crate::lifecycle::agents_for_meta_listing()
.await
.context("enumerating this hive's agents")?
.into_iter()
.map(|spec| spec.name)
.collect();
let mut intents = BTreeMap::new();
for agent in declared.agents.keys() {
match coord.power.get(agent) {
Ok(intent) => {
intents.insert(agent.clone(), intent);
}
// Left out of the map, which is what makes `plan` skip it.
Err(e) => {
tracing::warn!(%agent, error = ?e, "wanted state: cannot read power intent; skipping");
}
}
}
// The pause marker is a stat, not a store read — it cannot fail the way
// `coord.power.get` above can, so there is no "unreadable, skip" branch
// to mirror. A name that does not parse as an `Ident` is skipped with a
// warning instead, the same defensive shape `crash_watch` uses for the
// same reason: a declaration is attacker-adjacent input (published by
// the swarm controller, not typed by an operator at this hive), so a
// malformed name should not panic the convergence loop.
let mut paused = BTreeMap::new();
for agent in declared.agents.keys() {
match hive_types::Ident::parse(agent) {
Ok(id) => {
paused.insert(agent.clone(), Coordinator::is_paused(&id));
}
Err(e) => {
tracing::warn!(%agent, error = %e, "wanted state: invalid agent name; skipping pause check");
}
}
}
let plan = plan(declared, &present, &intents, &paused);
tracing::info!(
declared = declared.agents.len(),
deploy = plan.deploy.len(),
start = plan.start.len(),
stop = plan.stop.len(),
destroy = plan.destroy.len(),
pause = plan.pause.len(),
resume = plan.resume.len(),
"wanted state: converging"
);
let mut queued = false;
for agent in &plan.deploy {
match crate::swarm_status::queue_first_deploy(coord, agent) {
Ok(_) => queued = true,
// Per agent rather than the whole sweep: one agent that cannot be
// queued is not a reason to leave the others undeclared.
Err(e) => tracing::warn!(%agent, error = %e, "wanted state: queueing a deploy failed"),
}
}
if !plan.start.is_empty() {
crate::job_queue::power::start_many(coord, &plan.start)
.await
.context("queueing the declared starts")?;
}
if !plan.stop.is_empty() {
crate::job_queue::power::stop_many(coord, &plan.stop, true)
.await
.context("queueing the declared stops")?;
}
// No purge: this issue's own scope is tearing the container down, not
// wiping its persistent state. `actions::destroy` submits its own DAG
// node and emits its own queue snapshot, so there is nothing to await
// or fold into `queued` here — same fire-and-forget contract every other
// lifecycle op in that module has.
for agent in &plan.destroy {
crate::actions::destroy(coord, agent, false);
}
if queued {
// The power ops and destroy emit their own; the first deploys do not.
coord.emit_rebuild_queue_snapshot();
}
// `set_paused_by_name` — same helper `dashboard::lifecycle_ops`'s
// `/api/pause`/`/api/resume` use, not the job-queue DAG those
// name-check in their own doc comments — this loop already has its own
// per-agent retry story (redeclare-or-rewatch), so there is nothing
// here to wait on an acknowledgement for. One loop over both lists
// paired with their target `paused` value, not two near-identical
// copies (argus, PR review).
let mut rescan = false;
for (agents, paused) in [(&plan.pause, true), (&plan.resume, false)] {
for agent in agents {
match Coordinator::set_paused_by_name(agent, paused).await {
Ok(()) => rescan = true,
Err(e) => {
tracing::warn!(%agent, paused, error = %e, "wanted state: set-paused failed");
}
}
}
}
if rescan {
// Same call `post_pause`/`post_resume` make, so the dashboard's
// "paused" badge flips without waiting for the next periodic sweep.
coord.rescan_containers_and_emit().await;
}
Ok(())
}
/// Everything one declaration asks this hive to queue.
///
/// `pause`/`resume` are a second, independent axis from the other four —
/// see [`decide_pause`]'s own comment for why an agent can land in one of
/// the power-axis lists *and* one of these in the same pass.
#[derive(Debug, Default, PartialEq, Eq)]
struct Plan {
deploy: Vec<String>,
start: Vec<String>,
stop: Vec<String>,
destroy: Vec<String>,
pause: Vec<String>,
resume: Vec<String>,
}
/// Turn one declaration into that plan — pure, so what the loop does with a
/// given declaration is testable without a queue, a container or a store.
///
/// `intents` is the hive's stored power intent per agent, and an agent
/// **missing from the map** is one whose intent could not be read: it is
/// skipped. That is not `Some(None)` — "no row yet" is a reading and gets
/// converged, "the read failed" is not and would be a guess.
///
/// An agent the declaration does not name reaches none of the three lists.
/// That is the whole of "absence is not a deletion order": the agents this
/// hive runs that the swarm has not adopted are never even inspected.
fn plan(
declared: &HiveWanted,
present: &BTreeSet<String>,
intents: &BTreeMap<String, Option<Wanted>>,
paused: &BTreeMap<String, bool>,
) -> Plan {
let mut plan = Plan::default();
for (agent, decl) in &declared.agents {
// The pause axis does not gate on a readable power intent the way
// the match below does — it has no intent row to fail reading, only
// a stat that always answers — so it is evaluated unconditionally,
// even for an agent this pass otherwise skips.
let is_present = present.contains(agent);
let is_paused = paused.get(agent).copied().unwrap_or(false);
match decide_pause(decl.state, is_present, is_paused) {
PauseConverge::Pause => plan.pause.push(agent.clone()),
PauseConverge::Resume => plan.resume.push(agent.clone()),
PauseConverge::Nothing => {}
}
let Some(intent) = intents.get(agent) else {
continue;
};
match decide(decl.state, is_present, *intent) {
Converge::Deploy => plan.deploy.push(agent.clone()),
Converge::Start => plan.start.push(agent.clone()),
Converge::Stop => plan.stop.push(agent.clone()),
Converge::Destroy => plan.destroy.push(agent.clone()),
Converge::Nothing => {}
}
}
plan
}
/// What the loop will do about one declared agent.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
enum Converge {
/// Declared, and this hive has no container for it — the missed-deploy
/// repair.
Deploy,
Start,
Stop,
/// Declared destroyed, and this hive still has a container for it — run
/// the teardown once. Irreversible, so unlike `Start`/`Stop` there is no
/// intent to compare against: presence alone decides it.
Destroy,
/// The hive already agrees with the declaration.
Nothing,
}
/// The whole decision, pure: no queue, no store, no container. The three
/// inputs are exactly what [`converge`] reads per agent, so the table below is
/// the behaviour rather than a model of it.
///
/// `Paused` shares every one of `Up`'s arms here — on the power axis a
/// paused agent is still a running one, just with its turn loop parked.
/// [`decide_pause`] is where the two diverge.
fn decide(state: AgentState, present: bool, intent: Option<Wanted>) -> Converge {
match state {
AgentState::Up | AgentState::Paused if !present => Converge::Deploy,
AgentState::Up | AgentState::Paused => {
if intent == Some(Wanted::Up) {
Converge::Nothing
} else {
Converge::Start
}
}
// Declared offline/destroyed and not here: creating a container in
// order to leave it stopped, or to immediately re-destroy it, is not
// what either declaration asks for — absence plus a stopped-or-gone
// declaration is agreement, not a repair to queue.
AgentState::Offline | AgentState::Destroyed if !present => Converge::Nothing,
AgentState::Offline => {
if intent == Some(Wanted::Offline) {
Converge::Nothing
} else {
Converge::Stop
}
}
// Power intent plays no part here — a destroyed agent has no
// running/stopped distinction left to converge to, only present or
// not.
AgentState::Destroyed => Converge::Destroy,
}
}
/// What the loop will do about one declared agent's turn-loop pause state.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
enum PauseConverge {
Pause,
Resume,
/// Either the pause marker already matches the declaration, or there is
/// no container yet to hold one — see this function's own comment on
/// `!present`.
Nothing,
}
/// The pause-axis half of [`decide`], deliberately **not** folded into it:
/// the two vary independently (an agent can need `Converge::Start` and
/// `PauseConverge::Pause` in the same pass — declared `Paused`, currently
/// stopped), so a single combined enum would need one variant per
/// combination rather than the two small ones here.
///
/// `present` gates this exactly like [`decide`] gates deploy-vs-repair on
/// the power axis: a `Paused` declaration on an absent agent reaches
/// `Converge::Deploy` above, and the pause marker converges on the *next*
/// pass once that deploy has made the agent present, not this one — writing
/// a pause marker into a harness dir that may not exist yet is the same
/// "repair against a guess" this module's `Offline`/`Destroyed` present-gate
/// already refuses on the power axis.
fn decide_pause(state: AgentState, present: bool, currently_paused: bool) -> PauseConverge {
if !present {
return PauseConverge::Nothing;
}
match state {
AgentState::Paused if currently_paused => PauseConverge::Nothing,
AgentState::Paused => PauseConverge::Pause,
AgentState::Up if currently_paused => PauseConverge::Resume,
// `Up` already unpaused: agreement. `Offline`/`Destroyed`: the pause
// marker is moot either way (no turn loop runs on a stopped or gone
// container to park), so it is left as-is rather than cleared —
// whatever it says gets re-evaluated the moment either state moves
// back to `Up` or `Paused`.
AgentState::Up | AgentState::Offline | AgentState::Destroyed => PauseConverge::Nothing,
}
}
#[cfg(test)]
mod tests {
use std::collections::{BTreeMap, BTreeSet};
use super::{Converge, PauseConverge, Plan, carries_a_declaration, decide, decide_pause, plan};
use crate::power::Wanted;
use swarm_queue_client::wanted::{AgentState, AgentWanted, HiveWanted};
fn declaration(agents: &[(&str, AgentState)]) -> HiveWanted {
HiveWanted {
agents: agents
.iter()
.map(|(name, state)| ((*name).to_owned(), AgentWanted { state: *state }))
.collect(),
}
}
fn set(names: &[&str]) -> BTreeSet<String> {
names.iter().map(|n| (*n).to_owned()).collect()
}
fn intents(entries: &[(&str, Option<Wanted>)]) -> BTreeMap<String, Option<Wanted>> {
entries
.iter()
.map(|(name, intent)| ((*name).to_owned(), *intent))
.collect()
}
fn paused(entries: &[(&str, bool)]) -> BTreeMap<String, bool> {
entries
.iter()
.map(|(name, is_paused)| ((*name).to_owned(), *is_paused))
.collect()
}
/// The property the whole migration rides on: this hive's own agents are
/// not converged, they are not looked at. `adopted` is the control — the
/// same call must still act on what the declaration *does* name, or this
/// passes on a loop that plans nothing at all.
#[test]
fn an_agent_the_declaration_does_not_name_is_never_touched() {
let declared = declaration(&[("adopted", AgentState::Up)]);
let planned = plan(
&declared,
&set(&["adopted", "legacy"]),
&intents(&[("adopted", Some(Wanted::Offline)), ("legacy", None)]),
&paused(&[]),
);
assert_eq!(
planned,
Plan {
deploy: vec![],
start: vec!["adopted".to_owned()],
stop: vec![],
destroy: vec![],
pause: vec![],
resume: vec![],
}
);
}
/// An intent that could not be read is skipped rather than treated as "no
/// row". `readable` is the control for the same call.
#[test]
fn an_unreadable_intent_skips_only_that_agent() {
let declared = declaration(&[("unreadable", AgentState::Up), ("readable", AgentState::Up)]);
let planned = plan(
&declared,
&set(&["unreadable", "readable"]),
&intents(&[("readable", None)]),
&paused(&[]),
);
assert_eq!(planned.start, vec!["readable".to_owned()]);
assert!(planned.deploy.is_empty() && planned.stop.is_empty());
}
/// The pause axis does not gate on a readable intent row at all — a
/// `paused` agent this hive has no power-intent row for still gets its
/// pause marker converged, unlike the power axis right above.
#[test]
fn pause_converges_even_when_the_power_intent_is_unreadable() {
let declared = declaration(&[("unreadable", AgentState::Paused)]);
let planned = plan(
&declared,
&set(&["unreadable"]),
&intents(&[]),
&paused(&[("unreadable", false)]),
);
assert_eq!(planned.pause, vec!["unreadable".to_owned()]);
assert!(planned.start.is_empty() && planned.deploy.is_empty());
}
/// A declared-paused agent that is present, running and already carries
/// the marker needs nothing on either axis.
#[test]
fn an_already_paused_agent_converges_to_nothing() {
let declared = declaration(&[("parked", AgentState::Paused)]);
let planned = plan(
&declared,
&set(&["parked"]),
&intents(&[("parked", Some(Wanted::Up))]),
&paused(&[("parked", true)]),
);
assert_eq!(planned, Plan::default());
}
/// The case `decide_pause`'s own comment calls out: declared `Paused` but
/// currently stopped needs a power `Start` *and* a `Pause`, in the same
/// pass — this is the test that would fail if the two axes were folded
/// into one enum instead of decided independently.
#[test]
fn a_stopped_agent_declared_paused_gets_both_a_start_and_a_pause() {
let declared = declaration(&[("parked", AgentState::Paused)]);
let planned = plan(
&declared,
&set(&["parked"]),
&intents(&[("parked", Some(Wanted::Offline))]),
&paused(&[("parked", false)]),
);
assert_eq!(planned.start, vec!["parked".to_owned()]);
assert_eq!(planned.pause, vec!["parked".to_owned()]);
}
/// A declared-`Up` agent whose marker is still set from an earlier
/// `Paused` declaration gets resumed.
#[test]
fn an_up_agent_still_carrying_the_marker_is_resumed() {
let declared = declaration(&[("parked", AgentState::Up)]);
let planned = plan(
&declared,
&set(&["parked"]),
&intents(&[("parked", Some(Wanted::Up))]),
&paused(&[("parked", true)]),
);
assert_eq!(planned.resume, vec!["parked".to_owned()]);
}
#[test]
fn a_declared_agent_this_hive_does_not_have_is_deployed() {
assert_eq!(decide(AgentState::Up, false, None), Converge::Deploy);
// Even holding an intent: a row can outlive its container.
assert_eq!(
decide(AgentState::Up, false, Some(Wanted::Up)),
Converge::Deploy
);
}
#[test]
fn a_hive_that_already_agrees_queues_nothing() {
assert_eq!(
decide(AgentState::Up, true, Some(Wanted::Up)),
Converge::Nothing
);
assert_eq!(
decide(AgentState::Offline, true, Some(Wanted::Offline)),
Converge::Nothing
);
}
#[test]
fn a_disagreeing_intent_is_converged_both_ways() {
assert_eq!(
decide(AgentState::Up, true, Some(Wanted::Offline)),
Converge::Start
);
assert_eq!(
decide(AgentState::Offline, true, Some(Wanted::Up)),
Converge::Stop
);
}
/// No row is not agreement. An agent that predates the power store is
/// converged to what it was declared as, not left on whatever it happens
/// to be doing.
#[test]
fn an_agent_with_no_intent_row_is_converged() {
assert_eq!(decide(AgentState::Up, true, None), Converge::Start);
assert_eq!(decide(AgentState::Offline, true, None), Converge::Stop);
}
/// Declared offline, no container: the one absent case that must not
/// deploy. Its control is the `Up` case above, which must.
#[test]
fn an_absent_agent_declared_offline_is_left_absent() {
assert_eq!(decide(AgentState::Offline, false, None), Converge::Nothing);
}
/// A present agent declared destroyed gets torn down — the whole point of
/// this state. Intent is irrelevant, unlike `Up`/`Offline`: even an
/// agent whose power intent is `Up` still gets destroyed, since a
/// terminal declaration outranks a steady-state one.
#[test]
fn a_present_agent_declared_destroyed_is_destroyed() {
assert_eq!(decide(AgentState::Destroyed, true, None), Converge::Destroy);
assert_eq!(
decide(AgentState::Destroyed, true, Some(Wanted::Up)),
Converge::Destroy
);
}
/// Already gone: destroying an absent agent would be acting on nothing.
/// Its control is the present case above, which must destroy.
#[test]
fn an_absent_agent_declared_destroyed_is_left_absent() {
assert_eq!(
decide(AgentState::Destroyed, false, None),
Converge::Nothing
);
}
/// `Paused` shares every `Up` arm on the power axis — same deploy-when-
/// absent, same start-when-stopped behaviour.
#[test]
fn paused_converges_on_the_power_axis_exactly_like_up() {
for present in [true, false] {
for intent in [None, Some(Wanted::Up), Some(Wanted::Offline)] {
assert_eq!(
decide(AgentState::Paused, present, intent),
decide(AgentState::Up, present, intent),
"present={present:?} intent={intent:?}"
);
}
}
}
#[test]
fn decide_pause_sets_the_marker_for_a_present_unpaused_paused_declaration() {
assert_eq!(
decide_pause(AgentState::Paused, true, false),
PauseConverge::Pause
);
}
#[test]
fn decide_pause_is_idempotent_once_the_marker_already_matches() {
assert_eq!(
decide_pause(AgentState::Paused, true, true),
PauseConverge::Nothing
);
assert_eq!(
decide_pause(AgentState::Up, true, false),
PauseConverge::Nothing
);
}
#[test]
fn decide_pause_clears_the_marker_for_a_declared_up_agent() {
assert_eq!(
decide_pause(AgentState::Up, true, true),
PauseConverge::Resume
);
}
/// No container, nothing to write a marker into — the same present-gate
/// [`decide`] applies to `Offline`/`Destroyed`, mirrored here for the
/// pause axis regardless of declared state.
#[test]
fn decide_pause_does_nothing_for_an_absent_agent() {
for state in [
AgentState::Up,
AgentState::Paused,
AgentState::Offline,
AgentState::Destroyed,
] {
for currently_paused in [true, false] {
assert_eq!(
decide_pause(state, false, currently_paused),
PauseConverge::Nothing,
"state={state:?} currently_paused={currently_paused:?}"
);
}
}
}
/// `Offline`/`Destroyed` never touch the marker either way — it is left
/// exactly as found, whatever that is.
#[test]
fn decide_pause_leaves_offline_and_destroyed_alone() {
for state in [AgentState::Offline, AgentState::Destroyed] {
for currently_paused in [true, false] {
assert_eq!(
decide_pause(state, true, currently_paused),
PauseConverge::Nothing,
"state={state:?} currently_paused={currently_paused:?}"
);
}
}
}
/// The watch's half of "absence is not a deletion order". `Put` is the
/// control: without it this would pass on a function that refused
/// everything, which would silently stop the fast path converging at all.
#[test]
fn only_a_put_carries_a_declaration_to_converge_to() {
use async_nats::jetstream::kv::Operation;
assert!(carries_a_declaration(Operation::Put));
for withdrawn in [Operation::Delete, Operation::Purge] {
assert!(
!carries_a_declaration(withdrawn),
"{withdrawn:?} must not converge"
);
}
}
/// Version skew is handled one layer up, at the decode: `AgentState` is
/// closed, so an unknown value never reaches [`decide`] — it fails the
/// whole declaration in `swarm-queue-client`, which owns that test
/// (`an_unknown_state_fails_the_whole_declaration`). There is deliberately
/// no case for it here: a test asserting something unrepresentable would
/// pass forever without measuring anything.
#[test]
fn every_state_this_build_knows_is_covered_above() {
for state in [
AgentState::Up,
AgentState::Offline,
AgentState::Paused,
AgentState::Destroyed,
] {
let seen = [true, false].iter().any(|present| {
decide(state, *present, None) != Converge::Nothing
|| decide(state, *present, Some(Wanted::Up)) != Converge::Nothing
});
assert!(
seen,
"{state:?} produces no power action in any combination"
);
}
}
/// The pause-axis equivalent of the test above: every state this build
/// knows produces a non-`Nothing` pause decision in at least one
/// combination, so a state that silently fell out of `decide_pause`
/// would be caught here rather than passing forever unmeasured. `Up` and
/// `Paused` are the two expected to; `Offline`/`Destroyed` are asserted
/// separately above as deliberately inert on this axis.
#[test]
fn up_and_paused_both_produce_a_pause_action_in_some_combination() {
for state in [AgentState::Up, AgentState::Paused] {
let seen = [true, false].iter().any(|currently_paused| {
decide_pause(state, true, *currently_paused) != PauseConverge::Nothing
});
assert!(
seen,
"{state:?} produces no pause action in any combination"
);
}
}
}