mod.rs

use crate::assertions::AssertErrorHasSource;
use crate::colored::{mark_missing, mark_missing_string, mark_unexpected, mark_unexpected_string};
use crate::expectations::{
    ErrorHasSource, ErrorHasSourceMessage, error_has_source, error_has_source_message, not,
};
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 for Spec<'a, S, R>
where
    S: Error,
    R: FailingStrategy,
{
    type SourceMessage = Spec<'a, Option<String>, R>;

    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>) -> Self::SourceMessage {
        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;