From e7575e2c57db57b5b7df02fbe9175a0d7e386e4a Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Tue, 21 Oct 2025 18:55:53 -0400 Subject: [PATCH] why does pretty formatting matter --- .../messages/websocket_server_message.rs | 15 +++++++++++ .../websocket_messages/wire_formatted.json | 25 +++++++++++++++++++ 2 files changed, 40 insertions(+) create mode 100644 crates/api_clients/grok_client/test_data/websocket_messages/wire_formatted.json diff --git a/crates/api_clients/grok_client/src/requests/listen_websocket/messages/websocket_server_message.rs b/crates/api_clients/grok_client/src/requests/listen_websocket/messages/websocket_server_message.rs index b7c204b84f..046a1f48c3 100644 --- a/crates/api_clients/grok_client/src/requests/listen_websocket/messages/websocket_server_message.rs +++ b/crates/api_clients/grok_client/src/requests/listen_websocket/messages/websocket_server_message.rs @@ -151,6 +151,21 @@ mod tests { Ok(()) } + #[test] + fn test_wire_formatted() -> anyhow::Result<()> { + let path = "/Users/bt/dev/storyteller/storyteller-rust/crates/api_clients/grok_client/test_data/websocket_messages/wire_formatted.json"; + let data = std::fs::read_to_string(path)?; + //let message: WebsocketServerMessage = serde_json::from_str(&data)?; + let message = WebsocketServerMessage::from_json_str(&data)?; + + match message { + WebsocketServerMessage::Unknown(_value)=> {}, + _ => panic!("Expected Unknown message"), + } + + Ok(()) + } + #[test] fn test_empty() -> anyhow::Result<()> { let path = "/Users/bt/dev/storyteller/storyteller-rust/crates/api_clients/grok_client/test_data/websocket_messages/empty.json"; diff --git a/crates/api_clients/grok_client/test_data/websocket_messages/wire_formatted.json b/crates/api_clients/grok_client/test_data/websocket_messages/wire_formatted.json new file mode 100644 index 0000000000..2f458d5128 --- /dev/null +++ b/crates/api_clients/grok_client/test_data/websocket_messages/wire_formatted.json @@ -0,0 +1,25 @@ +{ + "type": "json", + "current_status": "start_stage", + "percentage_complete": 0, + "job_id": "0080eaac-f7f7-4369-90a2-346491939c1b", + "elapsed": null, + "prompt": "a dog riding a motorcycle", + "confirmation": "Drew a dog riding a motorcycle", + "full_prompt": "a dog riding a motorcycle", + "flags": null, + "is_alteration": false, + "section_id": 0, + "input_type": "text", + "request_id": "bbc15402-3257-4415-922e-b156c151f3c2", + "asr_transcript": null, + "asr_accepted": null, + "seed": null, + "image_reference": null, + "is_image_edit": null, + "moderated": false, + "r_rated": null, + "order": 0, + "model_name": "imagine_h_1", + "image_id": null +}