Another attempt to improve coverage of errors.rs

Finomnis · Oct 22, 2023 · e0ddd1b · e0ddd1b
1 parent 9de8b95
commit e0ddd1b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/errors.rs b/src/errors.rs
@@ -124,6 +124,7 @@ impl<ErrType: ErrTypeTraits> SubsystemError<ErrType> {
 /// [`cancel_on_shutdown()`](crate::FutureExt::cancel_on_shutdown).
 #[derive(Error, Debug, Diagnostic)]
 #[error("A shutdown request caused this task to be cancelled")]
+#[diagnostic(code(graceful_shutdown::future::cancelled_by_shutdown))]
 pub struct CancelledByShutdown;
 
 #[cfg(test)]

diff --git a/src/errors/tests.rs b/src/errors/tests.rs
@@ -7,6 +7,8 @@ fn examine_report(
 ) {
     println!("{}", error);
     println!("{:?}", error);
+    println!("{:?}", error.source());
+    println!("{}", error.code().unwrap());
     // Convert to report
     let report: miette::Report = error.into();
     println!("{}", report);