openai/codex
1
0
Fork 0
mirror of https://github.com/openai/codex.git synced 2026-10-20 17:38:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

[codex-api] keep websocket auth hook fixtures generic [ci changed_files]

Browse Source
This commit is contained in:
Cooper Gamble
2026-06-03 10:12:49 +00:00
parent 7bf64be075
commit 51604c749f
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
codex-rs/codex-api/src/endpoint/responses_websocket.rs
+9 -9
View File
@@ -916,9 +916,9 @@ mod tests {
use tokio::net::TcpListener;
use tokio_tungstenite::accept_hdr_async_with_config;
const SENT_STATE: &str = "ois1.sent.nonce.ciphertext";
const HANDSHAKE_STATE: &str = "ois1.handshake.nonce.ciphertext";
const WRAPPED_ERROR_STATE: &str = "ois1.error.nonce.ciphertext";
const SENT_STATE: &str = "sent-state";
const HANDSHAKE_STATE: &str = "handshake-state";
const WRAPPED_ERROR_STATE: &str = "wrapped-error-state";
struct RecordingAuthProvider {
state: StdMutex<String>,
@@ -943,7 +943,7 @@ mod tests {
.lock()
.expect("state lock should not be poisoned");
headers.insert(
"x-oai-is",
"x-test-state",
HeaderValue::from_str(&state).expect("state should be a valid header value"),
);
}
@@ -955,13 +955,13 @@ mod tests {
response_headers: &HeaderMap,
) {
let Some(sent_state) = request_headers
.get("x-oai-is")
.get("x-test-state")
.and_then(|value| value.to_str().ok())
else {
return;
};
let Some(update_state) = response_headers
.get("x-oai-is-update")
.get("x-test-state-update")
.and_then(|value| value.to_str().ok())
else {
return;
@@ -999,11 +999,11 @@ mod tests {
.lock()
.expect("request state lock should not be poisoned") = request
.headers()
.get("x-oai-is")
.get("x-test-state")
.and_then(|value| value.to_str().ok())
.map(ToString::to_string);
response.headers_mut().insert(
"x-oai-is-update",
"x-test-state-update",
HeaderValue::from_static(HANDSHAKE_STATE),
);
Ok(response)
@@ -1028,7 +1028,7 @@ mod tests {
"message": WEBSOCKET_CONNECTION_LIMIT_REACHED_MESSAGE,
},
"headers": {
"x-oai-is-update": WRAPPED_ERROR_STATE,
"x-test-state-update": WRAPPED_ERROR_STATE,
},
})
.to_string()