Skip to content

Commit c00fac2

Browse files
test: cover sparsekernel run ledger audits
1 parent 233935a commit c00fac2

1 file changed

Lines changed: 22 additions & 0 deletions

File tree

  • crates/sparsekernel-cli/src

crates/sparsekernel-cli/src/lib.rs

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2176,6 +2176,10 @@ mod tests {
21762176
let task = json_call(&mut db, "POST", "/tasks/enqueue", json!({ "kind": "demo" }));
21772177
let task_id = task["id"].as_str().unwrap().to_string();
21782178
assert_eq!(task["priority"], 0);
2179+
let audit = db.list_audit(1).unwrap();
2180+
assert_eq!(audit[0].action, "task.created");
2181+
assert_eq!(audit[0].object_id.as_deref(), Some(task_id.as_str()));
2182+
assert_eq!(audit[0].payload.as_ref().unwrap()["kind"], "demo");
21792183

21802184
let claimed = json_call(
21812185
&mut db,
@@ -2185,6 +2189,10 @@ mod tests {
21852189
);
21862190
assert_eq!(claimed["id"], task_id);
21872191
assert_eq!(claimed["lease_owner"], "worker-a");
2192+
let audit = db.list_audit(1).unwrap();
2193+
assert_eq!(audit[0].action, "task.claimed");
2194+
assert_eq!(audit[0].actor_id.as_deref(), Some("worker-a"));
2195+
assert_eq!(audit[0].object_id.as_deref(), Some(task_id.as_str()));
21882196

21892197
let empty_claim = json_call(
21902198
&mut db,
@@ -2201,6 +2209,10 @@ mod tests {
22012209
json!({ "task_id": task_id, "worker_id": "worker-a" }),
22022210
);
22032211
assert_eq!(completed["ok"], true);
2212+
let audit = db.list_audit(1).unwrap();
2213+
assert_eq!(audit[0].action, "task.completed");
2214+
assert_eq!(audit[0].actor_id.as_deref(), Some("worker-a"));
2215+
assert_eq!(audit[0].object_id.as_deref(), Some(task_id.as_str()));
22042216
}
22052217

22062218
#[test]
@@ -2288,6 +2300,10 @@ mod tests {
22882300
assert_eq!(session["id"], "session-a");
22892301
assert_eq!(session["agent_id"], "agent-a");
22902302
assert_eq!(session["current_token_count"], 12);
2303+
let audit = db.list_audit(1).unwrap();
2304+
assert_eq!(audit[0].action, "session.upserted");
2305+
assert_eq!(audit[0].object_id.as_deref(), Some("session-a"));
2306+
assert_eq!(audit[0].payload.as_ref().unwrap()["agentId"], "agent-a");
22912307

22922308
let sessions = handle_api_request(&mut db, "GET", "/sessions", &[])
22932309
.unwrap()
@@ -2306,6 +2322,12 @@ mod tests {
23062322
}),
23072323
);
23082324
assert_eq!(event["seq"], 1);
2325+
let event_id = event["id"].as_i64().unwrap().to_string();
2326+
let audit = db.list_audit(1).unwrap();
2327+
assert_eq!(audit[0].action, "transcript_event.appended");
2328+
assert_eq!(audit[0].object_id.as_deref(), Some(event_id.as_str()));
2329+
assert_eq!(audit[0].payload.as_ref().unwrap()["sessionId"], "session-a");
2330+
assert_eq!(audit[0].payload.as_ref().unwrap()["seq"], 1);
23092331
let events = json_call(
23102332
&mut db,
23112333
"POST",

0 commit comments

Comments
 (0)