rename to planetwars-server
This commit is contained in:
parent
71ee6c99e9
commit
3edf5d60f5
25 changed files with 3 additions and 221 deletions
|
@ -1,98 +0,0 @@
|
||||||
#![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
|
|
||||||
}
|
|
|
@ -1,61 +0,0 @@
|
||||||
#![feature(async_closure)]
|
|
||||||
extern crate mozaic4_backend;
|
|
||||||
|
|
||||||
use rocket::http::{ContentType, Status};
|
|
||||||
|
|
||||||
mod util;
|
|
||||||
use util::run_test;
|
|
||||||
|
|
||||||
#[rocket::async_test]
|
|
||||||
async fn test_registration() {
|
|
||||||
run_test(async move |client, _conn| {
|
|
||||||
let response = client
|
|
||||||
.post("/register")
|
|
||||||
.header(ContentType::JSON)
|
|
||||||
.body(r#"{"username": "piepkonijn", "password": "geheim123"}"#)
|
|
||||||
.dispatch()
|
|
||||||
.await;
|
|
||||||
|
|
||||||
assert_eq!(response.status(), Status::Ok);
|
|
||||||
assert_eq!(response.content_type(), Some(ContentType::JSON));
|
|
||||||
|
|
||||||
let response = client
|
|
||||||
.post("/login")
|
|
||||||
.header(ContentType::JSON)
|
|
||||||
.body(r#"{"username": "piepkonijn", "password": "geheim123"}"#)
|
|
||||||
.dispatch()
|
|
||||||
.await;
|
|
||||||
|
|
||||||
assert_eq!(response.status(), Status::Ok);
|
|
||||||
let token = response.into_string().await.unwrap();
|
|
||||||
|
|
||||||
let response = client
|
|
||||||
.get("/users/me")
|
|
||||||
.header(util::BearerAuth::new(token))
|
|
||||||
.dispatch()
|
|
||||||
.await;
|
|
||||||
|
|
||||||
assert_eq!(response.status(), Status::Ok);
|
|
||||||
assert_eq!(response.content_type(), Some(ContentType::JSON));
|
|
||||||
let resp = response.into_string().await.unwrap();
|
|
||||||
let json: serde_json::Value = serde_json::from_str(&resp).unwrap();
|
|
||||||
assert_eq!(json["username"], "piepkonijn");
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
|
|
||||||
#[rocket::async_test]
|
|
||||||
async fn test_reject_invalid_credentials() {
|
|
||||||
run_test(async move |client, _conn| {
|
|
||||||
let response = client
|
|
||||||
.post("/login")
|
|
||||||
.header(ContentType::JSON)
|
|
||||||
.body(r#"{"username": "piepkonijn", "password": "letmeinplease"}"#)
|
|
||||||
.dispatch()
|
|
||||||
.await;
|
|
||||||
|
|
||||||
assert_eq!(response.status(), Status::Forbidden);
|
|
||||||
// assert_eq!(response.content_type(), Some(ContentType::JSON));
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
}
|
|
|
@ -1,59 +0,0 @@
|
||||||
use std::future::Future;
|
|
||||||
|
|
||||||
use diesel::RunQueryDsl;
|
|
||||||
use mozaic4_backend::DbConn;
|
|
||||||
use rocket::{http::Header, local::asynchronous::Client};
|
|
||||||
|
|
||||||
// We use a lock to synchronize between tests so DB operations don't collide.
|
|
||||||
// For now. In the future, we'll have a nice way to run each test in a DB
|
|
||||||
// transaction so we can regain concurrency.
|
|
||||||
static DB_LOCK: parking_lot::Mutex<()> = parking_lot::const_mutex(());
|
|
||||||
|
|
||||||
async fn reset_db(db: &DbConn) {
|
|
||||||
db.run(|conn| {
|
|
||||||
diesel::sql_query(
|
|
||||||
r#"
|
|
||||||
TRUNCATE TABLE users, sessions,
|
|
||||||
bots, code_bundles"#,
|
|
||||||
)
|
|
||||||
.execute(conn)
|
|
||||||
.expect("drop all tables");
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn run_test<F, R>(test_closure: F)
|
|
||||||
where
|
|
||||||
F: FnOnce(Client, DbConn) -> R,
|
|
||||||
R: Future<Output = ()>,
|
|
||||||
{
|
|
||||||
let _lock = DB_LOCK.lock();
|
|
||||||
|
|
||||||
let client = Client::untracked(mozaic4_backend::rocket())
|
|
||||||
.await
|
|
||||||
.expect("failed to create test client");
|
|
||||||
let db = mozaic4_backend::DbConn::get_one(client.rocket())
|
|
||||||
.await
|
|
||||||
.expect("failed to get db connection");
|
|
||||||
|
|
||||||
// make sure we start with a clean DB
|
|
||||||
reset_db(&db).await;
|
|
||||||
|
|
||||||
test_closure(client, db).await;
|
|
||||||
}
|
|
||||||
|
|
||||||
pub struct BearerAuth {
|
|
||||||
token: String,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl BearerAuth {
|
|
||||||
pub fn new(token: String) -> Self {
|
|
||||||
Self { token }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a> Into<Header<'a>> for BearerAuth {
|
|
||||||
fn into(self) -> Header<'a> {
|
|
||||||
Header::new("Authorization", format!("Bearer {}", self.token))
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,5 +1,5 @@
|
||||||
[package]
|
[package]
|
||||||
name = "mozaic4-backend"
|
name = "planetwars-server"
|
||||||
version = "0.0.0"
|
version = "0.0.0"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
|
|
|
@ -1,11 +1,11 @@
|
||||||
use std::net::SocketAddr;
|
use std::net::SocketAddr;
|
||||||
|
|
||||||
extern crate mozaic4_backend;
|
extern crate planetwars_server;
|
||||||
extern crate tokio;
|
extern crate tokio;
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() {
|
async fn main() {
|
||||||
let app = mozaic4_backend::app().await;
|
let app = planetwars_server::app().await;
|
||||||
|
|
||||||
let addr = SocketAddr::from(([127, 0, 0, 1], 9000));
|
let addr = SocketAddr::from(([127, 0, 0, 1], 9000));
|
||||||
|
|
Loading…
Reference in a new issue