Fix EventStreamUnmarshallerGeneratorTest

smithy-lang · Oct 17, 2022 · 6e3e165 · 6e3e165
1 parent 64c0280
commit 6e3e165
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...smithy/rust/codegen/server/smithy/protocols/parse/EventStreamUnmarshallerGeneratorTest.kt b/...smithy/rust/codegen/server/smithy/protocols/parse/EventStreamUnmarshallerGeneratorTest.kt
@@ -275,7 +275,9 @@ class EventStreamUnmarshallerGeneratorTest {
  assert!(result.is_ok(), "expected ok, got: {:?}", result);
  match expect_error(result.unwrap())$kindSuffix {
  TestStreamErrorKind::Unhandled(err) => {
- assert!(format!("{}", err).contains("message: \"unmodeled error\""));
+ let message = format!("{}", aws_smithy_types::error::display::DisplayErrorContext(&err));
+ let expected = "message: \"unmodeled error\"";
+ assert!(message.contains(expected), "Expected '{message}' to contain '{expected}'");
  }
  kind => panic!("expected generic error, but got {:?}", kind),
  }