98 lines
2.7 KiB
Rust
98 lines
2.7 KiB
Rust
#![feature(async_closure)]
|
|
extern crate mozaic4_backend;
|
|
extern crate zip;
|
|
|
|
use rocket::http::{ContentType, Status};
|
|
|
|
mod util;
|
|
use mozaic4_backend::db::{bots, sessions, users};
|
|
use mozaic4_backend::DbConn;
|
|
use sessions::Session;
|
|
use users::{Credentials, User};
|
|
use util::{run_test, BearerAuth};
|
|
|
|
async fn user_with_session(conn: &DbConn) -> (User, Session) {
|
|
conn.run(|conn| {
|
|
let credentials = Credentials {
|
|
username: "piepkonijn",
|
|
password: "geheim123",
|
|
};
|
|
let user = users::create_user(&credentials, conn).unwrap();
|
|
let session = sessions::create_session(&user, conn);
|
|
(user, session)
|
|
})
|
|
.await
|
|
}
|
|
|
|
#[rocket::async_test]
|
|
async fn test_bot_create() {
|
|
run_test(async move |client, conn| {
|
|
let (user, session) = user_with_session(&conn).await;
|
|
|
|
let response = client
|
|
.post("/bots")
|
|
.header(BearerAuth::new(session.token.clone()))
|
|
.header(ContentType::JSON)
|
|
.body(
|
|
r#"{
|
|
"name": "testbot"
|
|
}"#,
|
|
)
|
|
.dispatch()
|
|
.await;
|
|
|
|
assert_eq!(response.status(), Status::Created);
|
|
assert_eq!(response.content_type(), Some(ContentType::JSON));
|
|
|
|
let resp_text = response.into_string().await.unwrap();
|
|
let json: serde_json::Value = serde_json::from_str(&resp_text).unwrap();
|
|
assert_eq!(json["name"], "testbot");
|
|
assert_eq!(json["owner_id"], user.id);
|
|
})
|
|
.await
|
|
}
|
|
|
|
// create an example zipfile for bot upload
|
|
fn create_zip() -> std::io::Result<Vec<u8>> {
|
|
use std::io::Write;
|
|
use zip::write::FileOptions;
|
|
|
|
let cursor = std::io::Cursor::new(Vec::new());
|
|
let mut zip = zip::ZipWriter::new(cursor);
|
|
|
|
zip.start_file("test.txt", FileOptions::default())?;
|
|
zip.write_all(b"sup brudi")?;
|
|
let buf = zip.finish()?;
|
|
Ok(buf.into_inner())
|
|
}
|
|
|
|
#[rocket::async_test]
|
|
async fn test_bot_upload() {
|
|
run_test(async move |client, conn| {
|
|
let (user, session) = user_with_session(&conn).await;
|
|
|
|
let owner_id = user.id;
|
|
let bot = conn
|
|
.run(move |conn| {
|
|
let new_bot = bots::NewBot {
|
|
name: "testbot",
|
|
owner_id: owner_id,
|
|
};
|
|
bots::create_bot(&new_bot, conn).unwrap()
|
|
})
|
|
.await;
|
|
|
|
let zip_file = create_zip().unwrap();
|
|
|
|
let response = client
|
|
.post(format!("/bots/{}/upload", bot.id))
|
|
.header(BearerAuth::new(session.token.clone()))
|
|
.header(ContentType::JSON)
|
|
.body(zip_file)
|
|
.dispatch()
|
|
.await;
|
|
|
|
assert_eq!(response.status(), Status::Created);
|
|
})
|
|
.await
|
|
}
|