Update skills budget tests for extension API changes (#34744)

## What changed

- Initialize the thread lifecycle input without an MCP resource client.
- Call `contribute_thread_context` without the removed step-scoped data argument.

GitOrigin-RevId: 689a982ba49c2788dde6bc2cbe0d6d7cae6c0a73
This commit is contained in:
felixxia-oai
2026-07-22 12:22:40 +00:00
committed by copyberry
parent bd9a28a839
commit f21f98936c

View File

@@ -434,13 +434,14 @@ async fn moderate_budget_pressure_keeps_every_catalog_entry() -> TestResult {
session_source: &session_source,
persistent_thread_state_available: true,
environments: &[],
mcp_resource_client: None,
session_store: &session_store,
thread_store: &thread_store,
})
.await;
let fragments = registry.context_contributors()[0]
.contribute_thread_context(&session_store, &thread_store, &ExtensionData::new("step"))
.contribute_thread_context(&session_store, &thread_store)
.await;
assert_eq!(1, fragments.len());
let rendered = fragments[0].text();
@@ -515,13 +516,14 @@ async fn extreme_budget_pressure_removes_descriptions_before_omitting_entries()
session_source: &session_source,
persistent_thread_state_available: true,
environments: &[],
mcp_resource_client: None,
session_store: &session_store,
thread_store: &thread_store,
})
.await;
let fragments = registry.context_contributors()[0]
.contribute_thread_context(&session_store, &thread_store, &ExtensionData::new("step"))
.contribute_thread_context(&session_store, &thread_store)
.await;
assert_eq!(1, fragments.len());
let rendered = fragments[0].text();