mirror of
https://github.com/openai/codex.git
synced 2026-10-20 08:19:50 +00:00
Parameterize the turn-start originator header test (#47918)
Accept `service_name` and the expected originator as test parameters, passing `service_name` through `ThreadStartParams`. Retain the existing case that expects `TEST_ORIGINATOR` when no service name is provided. GitOrigin-RevId: 71097dcfe45496216cf3e765999ed0abd0a0cd71
This commit is contained in:
@@ -844,8 +844,12 @@ async fn turn_start_additional_context_flows_to_model_input() -> Result<()> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[test_case::test_case(None, TEST_ORIGINATOR)]
|
||||
#[tokio::test]
|
||||
async fn turn_start_sends_originator_header() -> Result<()> {
|
||||
async fn turn_start_sends_originator_header(
|
||||
service_name: Option<&str>,
|
||||
expected_originator: &str,
|
||||
) -> Result<()> {
|
||||
let responses = vec![create_final_assistant_message_sse_response("Done")?];
|
||||
let server = create_mock_responses_server_sequence_unchecked(responses).await;
|
||||
|
||||
@@ -870,6 +874,7 @@ async fn turn_start_sends_originator_header() -> Result<()> {
|
||||
.start_thread(ThreadStartParams {
|
||||
model: Some("mock-model".to_string()),
|
||||
thread_source: Some(ThreadSource::User),
|
||||
service_name: service_name.map(str::to_string),
|
||||
..Default::default()
|
||||
})
|
||||
.await?;
|
||||
@@ -905,7 +910,7 @@ async fn turn_start_sends_originator_header() -> Result<()> {
|
||||
.headers
|
||||
.get("originator")
|
||||
.expect("originator header missing");
|
||||
assert_eq!(originator.to_str()?, TEST_ORIGINATOR);
|
||||
assert_eq!(originator.to_str()?, expected_originator);
|
||||
}
|
||||
|
||||
Ok(())
|
||||
|
||||
Reference in New Issue
Block a user