use crate::assertions::AssertErrorHasSource;
use crate::colored::{mark_missing, mark_missing_string, mark_unexpected, mark_unexpected_string};
use crate::expectations::{
error_has_source, error_has_source_message, not, ErrorHasSource, ErrorHasSourceMessage,
};
use crate::spec::{DiffFormat, Expectation, Expression, FailingStrategy, Invertible, Spec};
use crate::std::error::Error;
use crate::std::format;
use crate::std::string::{String, ToString};
impl<'a, S, R> AssertErrorHasSource<'a, R> for Spec<'a, S, R>
where
S: Error,
R: FailingStrategy,
{
fn has_no_source(self) -> Self {
self.expecting(not(error_has_source()))
}
fn has_source(self) -> Self {
self.expecting(error_has_source())
}
fn has_source_message(
self,
expected_source_message: impl Into<String>,
) -> Spec<'a, Option<String>, R> {
let expected_source_message = expected_source_message.into();
self.expecting(error_has_source_message(expected_source_message))
.mapping(|err| err.source().map(ToString::to_string))
}
}
impl<S> Expectation<S> for ErrorHasSource
where
S: Error,
{
fn test(&mut self, subject: &S) -> bool {
subject.source().is_some()
}
fn message(
&self,
expression: &Expression<'_>,
actual: &S,
inverted: bool,
format: &DiffFormat,
) -> String {
let (a, expected) = if inverted {
("no", "<error with no source>")
} else {
("a", "<error with some source>")
};
let marked_actual = mark_unexpected(actual, format);
let marked_expected = mark_missing_string(expected, format);
format!("expected {expression} to have {a} source\n but was: {marked_actual}\n expected: {marked_expected}")
}
}
impl Invertible for ErrorHasSource {}
impl<S> Expectation<S> for ErrorHasSourceMessage
where
S: Error,
{
fn test(&mut self, subject: &S) -> bool {
subject
.source()
.is_some_and(|msg| msg.to_string() == self.expected_source_message)
}
fn message(
&self,
expression: &Expression<'_>,
actual: &S,
inverted: bool,
format: &DiffFormat,
) -> String {
let not = if inverted { "not " } else { "" };
let expected = &self.expected_source_message;
if let Some(actual_source) = actual.source() {
let marked_actual = mark_unexpected_string(&actual_source.to_string(), format);
let marked_expected = mark_missing_string(expected, format);
format!("expected {expression} to have a source message {not}equal to \"{expected}\"\n but was: \"{marked_actual}\"\n expected: \"{marked_expected}\"")
} else {
let mut marked_actual = mark_unexpected(actual, format);
marked_actual.push_str(" - which has no source");
let marked_expected = mark_missing(expected, format);
format!("expected {expression} to have a source message {not}equal to \"{expected}\"\n but was: {marked_actual}\n expected: {not}{marked_expected}")
}
}
}
impl Invertible for ErrorHasSourceMessage {}
#[cfg(test)]
mod tests;