survey/src/pages/errors.rs

121 lines
3.3 KiB
Rust
Raw Normal View History

2021-10-04 21:21:10 +05:30
/*
* Copyright (C) 2021 Aravinth Manivannan <realaravinth@batsense.net>
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
use actix_web::{web, HttpResponse, Responder};
use lazy_static::lazy_static;
use sailfish::TemplateOnce;
use crate::errors::PageError;
#[derive(Clone, TemplateOnce)]
#[template(path = "errors/index.html")]
pub struct ErrorPage<'a> {
pub title: &'a str,
pub message: &'a str,
}
const PAGE: &str = "Error";
impl<'a> ErrorPage<'a> {
pub fn new(title: &'a str, message: &'a str) -> Self {
ErrorPage { title, message }
}
}
lazy_static! {
static ref INTERNAL_SERVER_ERROR_BODY: String = ErrorPage::new(
"Internal Server Error",
&format!("{}", PageError::InternalServerError),
)
.render_once()
.unwrap();
static ref UNKNOWN_ERROR_BODY: String = ErrorPage::new(
"Something went wrong",
&format!("{}", PageError::InternalServerError),
)
.render_once()
.unwrap();
}
const ERROR_ROUTE: &str = "/error/{id}";
#[my_codegen::get(path = "ERROR_ROUTE")]
async fn error(path: web::Path<usize>) -> impl Responder {
let resp = match path.into_inner() {
500 => HttpResponse::InternalServerError()
.content_type("text/html; charset=utf-8")
.body(&*INTERNAL_SERVER_ERROR_BODY),
_ => HttpResponse::InternalServerError()
.content_type("text/html; charset=utf-8")
.body(&*UNKNOWN_ERROR_BODY),
};
resp
}
pub fn services(cfg: &mut web::ServiceConfig) {
cfg.service(error);
}
pub mod routes {
pub struct Errors {
pub internal_server_error: &'static str,
pub unknown_error: &'static str,
}
impl Errors {
pub const fn new() -> Self {
Errors {
internal_server_error: "/error/500",
unknown_error: "/error/007",
}
}
}
}
#[cfg(test)]
mod tests {
use actix_web::{http::StatusCode, test, App};
use super::*;
use crate::PAGES;
#[actix_rt::test]
async fn error_pages_work() {
let app = test::init_service(App::new().configure(services)).await;
let resp = test::call_service(
&app,
test::TestRequest::get()
.uri(PAGES.errors.internal_server_error)
.to_request(),
)
.await;
assert_eq!(resp.status(), StatusCode::INTERNAL_SERVER_ERROR);
let resp = test::call_service(
&app,
test::TestRequest::get()
.uri(PAGES.errors.unknown_error)
.to_request(),
)
.await;
assert_eq!(resp.status(), StatusCode::INTERNAL_SERVER_ERROR);
}
}