Skip to content

Commit cf55c7e

Browse files
committed
fmt-ed
1 parent 2c620db commit cf55c7e

1 file changed

Lines changed: 36 additions & 10 deletions

File tree

src/data/poller.rs

Lines changed: 36 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -228,8 +228,11 @@ pub fn poll_fast() -> anyhow::Result<PollSnapshot> {
228228

229229
pub fn poll_full() -> anyhow::Result<PollSnapshot> {
230230
let mut snapshot = poll_fast()?;
231-
let mut seen: std::collections::HashSet<String> =
232-
snapshot.sessions.iter().map(|s| s.session_id.clone()).collect();
231+
let mut seen: std::collections::HashSet<String> = snapshot
232+
.sessions
233+
.iter()
234+
.map(|s| s.session_id.clone())
235+
.collect();
233236

234237
let reader = DbReader::open_default()?;
235238
let projects = reader.get_projects()?;
@@ -451,9 +454,7 @@ fn hydrate_session(
451454
status,
452455
process_pid,
453456
model: reader.get_session_model(session_id)?,
454-
preview: reader
455-
.get_last_message_preview(session_id)?
456-
.map(|p| p.text),
457+
preview: reader.get_last_message_preview(session_id)?.map(|p| p.text),
457458
time_updated: Some(session.time_updated),
458459
has_children: reader.has_child_sessions(session_id)?,
459460
children: collect_children(reader, session_id, 2)?,
@@ -565,8 +566,16 @@ mod tests {
565566
let db_path = temp_db_path("hydrate-miss");
566567
let _conn = init_db(&db_path);
567568
let reader = DbReader::open(&db_path).unwrap();
568-
let result =
569-
hydrate_session(&reader, "nonexistent", None, None, DiscoverySource::Serve, None, None).unwrap();
569+
let result = hydrate_session(
570+
&reader,
571+
"nonexistent",
572+
None,
573+
None,
574+
DiscoverySource::Serve,
575+
None,
576+
None,
577+
)
578+
.unwrap();
570579
assert!(result.is_none());
571580
}
572581

@@ -586,7 +595,16 @@ mod tests {
586595
.unwrap();
587596

588597
let reader = DbReader::open(&db_path).unwrap();
589-
let result = hydrate_session(&reader, "sess1", None, None, DiscoverySource::Serve, None, None).unwrap();
598+
let result = hydrate_session(
599+
&reader,
600+
"sess1",
601+
None,
602+
None,
603+
DiscoverySource::Serve,
604+
None,
605+
None,
606+
)
607+
.unwrap();
590608
assert!(result.is_none());
591609
}
592610

@@ -657,11 +675,19 @@ mod tests {
657675
let merged = merge_cached_serve_sessions(fast, &cached);
658676
assert_eq!(merged.sessions.len(), 3);
659677
// Fast B should win over cached B
660-
let b = merged.sessions.iter().find(|s| s.session_id == "sess_b").unwrap();
678+
let b = merged
679+
.sessions
680+
.iter()
681+
.find(|s| s.session_id == "sess_b")
682+
.unwrap();
661683
assert_eq!(b.title, "B");
662684
assert_eq!(b.source, DiscoverySource::TuiExplicit);
663685
// C should be added from cache
664-
let c = merged.sessions.iter().find(|s| s.session_id == "sess_c").unwrap();
686+
let c = merged
687+
.sessions
688+
.iter()
689+
.find(|s| s.session_id == "sess_c")
690+
.unwrap();
665691
assert_eq!(c.title, "C");
666692
assert_eq!(c.source, DiscoverySource::Serve);
667693
}

0 commit comments

Comments
 (0)