Skip to content

Commit 4106b19

Browse files
committed
test: add tests for mcp server event
1 parent 58328c7 commit 4106b19

File tree

1 file changed

+45
-1
lines changed

1 file changed

+45
-1
lines changed

crates/apollo-mcp-server/src/event.rs

Lines changed: 45 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ use apollo_mcp_registry::platform_api::operation_collections::error::CollectionE
33
use apollo_mcp_registry::uplink::schema::event::Event as SchemaEvent;
44
use std::fmt::Debug;
55
use std::fmt::Formatter;
6+
use std::fmt::Result;
67
use std::io;
78

89
/// MCP Server events
@@ -24,7 +25,7 @@ pub enum Event {
2425
}
2526

2627
impl Debug for Event {
27-
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
28+
fn fmt(&self, f: &mut Formatter) -> Result {
2829
match self {
2930
Event::SchemaUpdated(event) => {
3031
write!(f, "SchemaUpdated({event:?})")
@@ -44,3 +45,46 @@ impl Debug for Event {
4445
}
4546
}
4647
}
48+
49+
#[cfg(test)]
50+
mod tests {
51+
use super::*;
52+
53+
#[test]
54+
fn test_debug_event_schema_updated() {
55+
let event = Event::SchemaUpdated(SchemaEvent::NoMoreSchema);
56+
let output = format!("{:?}", event);
57+
assert_eq!(output, "SchemaUpdated(NoMoreSchema)");
58+
}
59+
60+
#[test]
61+
fn test_debug_event_operations_updated() {
62+
let event = Event::OperationsUpdated(vec![]);
63+
let output = format!("{:?}", event);
64+
assert_eq!(output, "OperationsChanged([])");
65+
}
66+
67+
#[test]
68+
fn test_debug_event_operation_error() {
69+
let event = Event::OperationError(std::io::Error::other("TEST"), None);
70+
let output = format!("{:?}", event);
71+
assert_eq!(
72+
output,
73+
r#"OperationError(Custom { kind: Other, error: "TEST" }, None)"#
74+
);
75+
}
76+
77+
#[test]
78+
fn test_debug_event_collection_error() {
79+
let event = Event::CollectionError(CollectionError::Response("TEST".to_string()));
80+
let output = format!("{:?}", event);
81+
assert_eq!(output, r#"OperationError(Response("TEST"))"#);
82+
}
83+
84+
#[test]
85+
fn test_debug_event_shutdown() {
86+
let event = Event::Shutdown;
87+
let output = format!("{:?}", event);
88+
assert_eq!(output, "Shutdown");
89+
}
90+
}

0 commit comments

Comments
 (0)