tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

chore: turn clippy on for test code (#4817)

Browse Source
This commit is contained in:
Jack Amadeo
2025-09-26 00:06:07 -04:00
committed by GitHub
parent f5877a9996
commit 757ceb6109
38 changed files with 660 additions and 849 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/context_mgmt/auto_compact.rs
+15 -6
View File
@@ -558,11 +558,14 @@ mod tests {
];
// Create session metadata with specific token counts
#[allow(clippy::field_reassign_with_default)]
let mut session_metadata = SessionMetadata::default();
session_metadata.total_tokens = Some(8000); // High token count to trigger compaction
session_metadata.accumulated_total_tokens = Some(15000); // Even higher accumulated count
session_metadata.input_tokens = Some(5000);
session_metadata.output_tokens = Some(3000);
{
session_metadata.total_tokens = Some(8000); // High token count to trigger compaction
session_metadata.accumulated_total_tokens = Some(15000); // Even higher accumulated count
session_metadata.input_tokens = Some(5000);
session_metadata.output_tokens = Some(3000);
}
// Test with session metadata - should use total_tokens for compaction (not accumulated)
let result_with_metadata = check_compaction_needed(
@@ -594,7 +597,10 @@ mod tests {
// Test with metadata that has only accumulated tokens (no total_tokens)
let mut session_metadata_no_total = SessionMetadata::default();
session_metadata_no_total.accumulated_total_tokens = Some(7500);
#[allow(clippy::field_reassign_with_default)]
{
session_metadata_no_total.accumulated_total_tokens = Some(7500);
}
let result_with_no_total = check_compaction_needed(
&agent,
@@ -650,7 +656,10 @@ mod tests {
// Create session metadata with high token count to trigger compaction
let mut session_metadata = SessionMetadata::default();
session_metadata.total_tokens = Some(9000); // High enough to trigger compaction
#[allow(clippy::field_reassign_with_default)]
{
session_metadata.total_tokens = Some(9000); // High enough to trigger compaction
}
// Test full compaction flow with session metadata
let result = check_and_compact_messages(
crates/goose/src/context_mgmt/truncate.rs
+5 -5
View File
@@ -506,7 +506,7 @@ mod tests {
let context_limit = 25;
let result = truncate_messages(
&messages.messages(),
messages.messages(),
&token_counts,
context_limit,
&OldestFirstTruncation,
@@ -590,7 +590,7 @@ mod tests {
let context_limit = 100; // Exactly matches total tokens
let result = truncate_messages(
&messages.messages(),
messages.messages(),
&token_counts,
context_limit,
&OldestFirstTruncation,
@@ -605,7 +605,7 @@ mod tests {
token_counts.push(1);
let result = truncate_messages(
&messages.messages(),
messages.messages(),
&token_counts,
context_limit,
&OldestFirstTruncation,
@@ -698,7 +698,7 @@ mod tests {
let (messages, token_counts) = result;
// Verify the conversation still makes sense
assert!(messages.len() >= 1);
assert!(!messages.is_empty());
assert!(messages.last().unwrap().role == Role::User);
assert!(token_counts.iter().sum::<usize>() <= context_limit);
@@ -710,7 +710,7 @@ mod tests {
// Test impossibly small context window
let (messages, token_counts) = create_messages_with_counts(1, 10, false);
let result = truncate_messages(
&messages.messages(),
messages.messages(),
&token_counts,
5, // Impossibly small context
&OldestFirstTruncation,