mod common;

use hegel::backend::DataSourceError;

#[test]
fn test_data_source_error_display() {
    let stop = DataSourceError::StopTest;
    let assume = DataSourceError::Assume;
    let server = DataSourceError::ServerError("something went wrong".to_string());

    assert!(stop.to_string().contains("StopTest"));
    assert!(assume.to_string().contains("Assume"));
    assert_eq!(server.to_string(), "something went wrong");
}
