90 lines
1.8 KiB
Rust
90 lines
1.8 KiB
Rust
use std::{error::Error, fmt};
|
|
|
|
#[derive(Debug)]
|
|
pub enum ErrorCode {
|
|
// General
|
|
Undefined(String),
|
|
}
|
|
|
|
impl fmt::Display for ErrorCode {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
match self {
|
|
ErrorCode::Undefined(msg) => write!(f, "{}", msg),
|
|
}
|
|
}
|
|
}
|
|
|
|
mod errorcodes {
|
|
use super::*;
|
|
|
|
const ITEMS: [&str; 2] = ["one", "two"];
|
|
|
|
#[test]
|
|
fn undefined_display() {
|
|
for item in ITEMS {
|
|
let err = ErrorCode::Undefined(item.to_string());
|
|
assert_eq!(err.to_string(), item);
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub struct MTTError {
|
|
code: ErrorCode,
|
|
}
|
|
|
|
impl MTTError {
|
|
fn new<S>(msg: S) -> Self
|
|
where
|
|
S: Into<String>,
|
|
{
|
|
let text = msg.into();
|
|
Self {
|
|
code: ErrorCode::Undefined(text),
|
|
}
|
|
}
|
|
|
|
fn from_code(code: ErrorCode) -> Self {
|
|
Self { code: code }
|
|
}
|
|
}
|
|
|
|
impl Error for MTTError {}
|
|
|
|
impl fmt::Display for MTTError {
|
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
|
write!(f, "{}", self.code)
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod errors {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn create_with_str() {
|
|
let msgs = ["one", "two"];
|
|
for msg in msgs {
|
|
let err = MTTError::new(msg);
|
|
assert_eq!(err.to_string(), msg);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn create_with_string() {
|
|
let msg = "three";
|
|
let err = MTTError::new(msg.to_string());
|
|
assert_eq!(err.to_string(), msg);
|
|
}
|
|
|
|
#[test]
|
|
fn create_from_code() {
|
|
let code = ErrorCode::Undefined("oops".to_string());
|
|
let err = MTTError::from_code(code);
|
|
match err.code {
|
|
ErrorCode::Undefined(_) => (),
|
|
_ => assert!(false, "{:?} is not undefined", err.code),
|
|
}
|
|
}
|
|
}
|