actix-admin/tests/test_setup/helper.rs

93 lines
2.7 KiB
Rust
Raw Normal View History

2022-08-27 14:41:08 +02:00
use sea_orm::{ConnectOptions, DatabaseConnection};
use actix_admin::prelude::*;
use actix_web::Error;
use actix_session::Session;
use actix_web::HttpResponse;
use actix_web::{web};
2022-08-27 14:41:08 +02:00
use super::{Post, Comment, create_tables};
pub async fn create_tables_and_get_connection() -> DatabaseConnection {
let opt = ConnectOptions::new("sqlite::memory:".to_owned());
let conn = sea_orm::Database::connect(opt).await.unwrap();
let _ = create_tables(&conn).await;
conn
}
#[derive(Clone)]
pub struct AppState {
pub db: DatabaseConnection,
pub actix_admin: ActixAdmin,
}
impl ActixAdminAppDataTrait for AppState {
fn get_db(&self) -> &DatabaseConnection {
&self.db
}
fn get_actix_admin(&self) -> &ActixAdmin {
&self.actix_admin
}
}
pub fn create_actix_admin_builder() -> ActixAdminBuilder {
let post_view_model = ActixAdminViewModel::from(Post);
let comment_view_model = ActixAdminViewModel::from(Comment);
let configuration = ActixAdminConfiguration {
enable_auth: false,
user_is_logged_in: None,
login_link: None,
logout_link: None,
};
let mut admin_builder = ActixAdminBuilder::new(configuration);
admin_builder.add_entity::<AppState, Post>(&post_view_model);
admin_builder.add_entity::<AppState, Comment>(&comment_view_model);
admin_builder.add_custom_handler_for_entity::<AppState, Comment>(
2022-08-28 20:26:10 +02:00
"/create_post_from_plaintext",
web::post().to(create_post_from_plaintext::<AppState, Comment>));
admin_builder.add_custom_handler_for_entity::<AppState, Post>(
"/create_post_from_plaintext",
web::post().to(create_post_from_plaintext::<AppState, Post>));
admin_builder.add_custom_handler_for_entity::<AppState, Post>(
"/edit_post_from_plaintext/{id}",
web::post().to(edit_post_from_plaintext::<AppState, Post>));
admin_builder.add_custom_handler_for_entity::<AppState, Comment>(
2022-08-28 20:26:10 +02:00
"/edit_post_from_plaintext/{id}",
web::post().to(edit_post_from_plaintext::<AppState, Comment>));
2022-08-27 14:41:08 +02:00
admin_builder
}
async fn create_post_from_plaintext<
T: ActixAdminAppDataTrait,
E: ActixAdminViewModelTrait,
>(
session: Session,
data: web::Data<T>,
text: String,
) -> Result<HttpResponse, Error> {
let model = ActixAdminModel::from(text);
create_or_edit_post::<T, E>(&session, &data, model, None).await
}
async fn edit_post_from_plaintext<
T: ActixAdminAppDataTrait,
E: ActixAdminViewModelTrait,
>(
session: Session,
data: web::Data<T>,
text: String,
id: web::Path<i32>,
) -> Result<HttpResponse, Error> {
println!("ok");
let model = ActixAdminModel::from(text);
create_or_edit_post::<T, E>(&session, &data, model, Some(id.into_inner())).await
}