mirror of
https://github.com/GothenburgBitFactory/taskchampion-sync-server.git
synced 2025-08-02 14:38:36 +02:00
use a struct for arguments to WebServer::new
This commit is contained in:
parent
3a794341ce
commit
0a71cce2d1
7 changed files with 81 additions and 49 deletions
|
@ -55,11 +55,11 @@ pub(crate) async fn service(
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use crate::api::CLIENT_ID_HEADER;
|
|
||||||
use crate::WebServer;
|
use crate::WebServer;
|
||||||
|
use crate::{api::CLIENT_ID_HEADER, WebConfig};
|
||||||
use actix_web::{http::StatusCode, test, App};
|
use actix_web::{http::StatusCode, test, App};
|
||||||
use pretty_assertions::assert_eq;
|
use pretty_assertions::assert_eq;
|
||||||
use taskchampion_sync_server_core::{InMemoryStorage, Storage, NIL_VERSION_ID};
|
use taskchampion_sync_server_core::{InMemoryStorage, ServerConfig, Storage, NIL_VERSION_ID};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
|
@ -76,7 +76,7 @@ mod test {
|
||||||
txn.commit()?;
|
txn.commit()?;
|
||||||
}
|
}
|
||||||
|
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
@ -119,7 +119,7 @@ mod test {
|
||||||
txn.commit().unwrap();
|
txn.commit().unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
@ -149,7 +149,7 @@ mod test {
|
||||||
let client_id = Uuid::new_v4();
|
let client_id = Uuid::new_v4();
|
||||||
let version_id = Uuid::new_v4();
|
let version_id = Uuid::new_v4();
|
||||||
let storage = InMemoryStorage::new();
|
let storage = InMemoryStorage::new();
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
@ -169,7 +169,7 @@ mod test {
|
||||||
let client_id = Uuid::new_v4();
|
let client_id = Uuid::new_v4();
|
||||||
let version_id = Uuid::new_v4();
|
let version_id = Uuid::new_v4();
|
||||||
let storage = InMemoryStorage::new();
|
let storage = InMemoryStorage::new();
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
|
|
@ -80,7 +80,7 @@ pub(crate) async fn service(
|
||||||
rb.append_header((PARENT_VERSION_ID_HEADER, parent_version_id.to_string()));
|
rb.append_header((PARENT_VERSION_ID_HEADER, parent_version_id.to_string()));
|
||||||
Ok(rb.finish())
|
Ok(rb.finish())
|
||||||
}
|
}
|
||||||
Err(ServerError::NoSuchClient) if server_state.create_clients => {
|
Err(ServerError::NoSuchClient) if server_state.web_config.create_clients => {
|
||||||
// Create a new client and repeat the `add_version` call.
|
// Create a new client and repeat the `add_version` call.
|
||||||
let mut txn = server_state
|
let mut txn = server_state
|
||||||
.server
|
.server
|
||||||
|
@ -97,11 +97,11 @@ pub(crate) async fn service(
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use crate::api::CLIENT_ID_HEADER;
|
|
||||||
use crate::WebServer;
|
use crate::WebServer;
|
||||||
|
use crate::{api::CLIENT_ID_HEADER, WebConfig};
|
||||||
use actix_web::{http::StatusCode, test, App};
|
use actix_web::{http::StatusCode, test, App};
|
||||||
use pretty_assertions::assert_eq;
|
use pretty_assertions::assert_eq;
|
||||||
use taskchampion_sync_server_core::{InMemoryStorage, Storage};
|
use taskchampion_sync_server_core::{InMemoryStorage, ServerConfig, Storage};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
|
@ -118,7 +118,7 @@ mod test {
|
||||||
txn.commit().unwrap();
|
txn.commit().unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
@ -152,7 +152,11 @@ mod test {
|
||||||
let client_id = Uuid::new_v4();
|
let client_id = Uuid::new_v4();
|
||||||
let version_id = Uuid::new_v4();
|
let version_id = Uuid::new_v4();
|
||||||
let parent_version_id = Uuid::new_v4();
|
let parent_version_id = Uuid::new_v4();
|
||||||
let server = WebServer::new(Default::default(), None, true, InMemoryStorage::new());
|
let server = WebServer::new(
|
||||||
|
ServerConfig::default(),
|
||||||
|
WebConfig::default(),
|
||||||
|
InMemoryStorage::new(),
|
||||||
|
);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
@ -195,9 +199,11 @@ mod test {
|
||||||
let client_id = Uuid::new_v4();
|
let client_id = Uuid::new_v4();
|
||||||
let parent_version_id = Uuid::new_v4();
|
let parent_version_id = Uuid::new_v4();
|
||||||
let server = WebServer::new(
|
let server = WebServer::new(
|
||||||
Default::default(),
|
ServerConfig::default(),
|
||||||
None,
|
WebConfig {
|
||||||
/*create_clients=*/ false,
|
create_clients: false,
|
||||||
|
..WebConfig::default()
|
||||||
|
},
|
||||||
InMemoryStorage::new(),
|
InMemoryStorage::new(),
|
||||||
);
|
);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
|
@ -232,7 +238,7 @@ mod test {
|
||||||
txn.commit().unwrap();
|
txn.commit().unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
@ -260,7 +266,7 @@ mod test {
|
||||||
let client_id = Uuid::new_v4();
|
let client_id = Uuid::new_v4();
|
||||||
let parent_version_id = Uuid::new_v4();
|
let parent_version_id = Uuid::new_v4();
|
||||||
let storage = InMemoryStorage::new();
|
let storage = InMemoryStorage::new();
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
@ -280,7 +286,7 @@ mod test {
|
||||||
let client_id = Uuid::new_v4();
|
let client_id = Uuid::new_v4();
|
||||||
let parent_version_id = Uuid::new_v4();
|
let parent_version_id = Uuid::new_v4();
|
||||||
let storage = InMemoryStorage::new();
|
let storage = InMemoryStorage::new();
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
|
|
@ -48,11 +48,11 @@ pub(crate) async fn service(
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use crate::api::CLIENT_ID_HEADER;
|
|
||||||
use crate::WebServer;
|
use crate::WebServer;
|
||||||
|
use crate::{api::CLIENT_ID_HEADER, WebConfig};
|
||||||
use actix_web::{http::StatusCode, test, App};
|
use actix_web::{http::StatusCode, test, App};
|
||||||
use pretty_assertions::assert_eq;
|
use pretty_assertions::assert_eq;
|
||||||
use taskchampion_sync_server_core::{InMemoryStorage, Storage, NIL_VERSION_ID};
|
use taskchampion_sync_server_core::{InMemoryStorage, ServerConfig, Storage, NIL_VERSION_ID};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
|
@ -71,11 +71,11 @@ mod test {
|
||||||
txn.commit().unwrap();
|
txn.commit().unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
let uri = format!("/v1/client/get-child-version/{}", parent_version_id);
|
let uri = format!("/v1/client/get-child-version/{parent_version_id}");
|
||||||
let req = test::TestRequest::get()
|
let req = test::TestRequest::get()
|
||||||
.uri(&uri)
|
.uri(&uri)
|
||||||
.append_header((CLIENT_ID_HEADER, client_id.to_string()))
|
.append_header((CLIENT_ID_HEADER, client_id.to_string()))
|
||||||
|
@ -105,11 +105,11 @@ mod test {
|
||||||
let client_id = Uuid::new_v4();
|
let client_id = Uuid::new_v4();
|
||||||
let parent_version_id = Uuid::new_v4();
|
let parent_version_id = Uuid::new_v4();
|
||||||
let storage = InMemoryStorage::new();
|
let storage = InMemoryStorage::new();
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
let uri = format!("/v1/client/get-child-version/{}", parent_version_id);
|
let uri = format!("/v1/client/get-child-version/{parent_version_id}");
|
||||||
let req = test::TestRequest::get()
|
let req = test::TestRequest::get()
|
||||||
.uri(&uri)
|
.uri(&uri)
|
||||||
.append_header((CLIENT_ID_HEADER, client_id.to_string()))
|
.append_header((CLIENT_ID_HEADER, client_id.to_string()))
|
||||||
|
@ -134,12 +134,12 @@ mod test {
|
||||||
.unwrap();
|
.unwrap();
|
||||||
txn.commit().unwrap();
|
txn.commit().unwrap();
|
||||||
}
|
}
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
// the child of the nil version is the added version
|
// the child of the nil version is the added version
|
||||||
let uri = format!("/v1/client/get-child-version/{}", NIL_VERSION_ID);
|
let uri = format!("/v1/client/get-child-version/{NIL_VERSION_ID}");
|
||||||
let req = test::TestRequest::get()
|
let req = test::TestRequest::get()
|
||||||
.uri(&uri)
|
.uri(&uri)
|
||||||
.append_header((CLIENT_ID_HEADER, client_id.to_string()))
|
.append_header((CLIENT_ID_HEADER, client_id.to_string()))
|
||||||
|
@ -168,7 +168,7 @@ mod test {
|
||||||
|
|
||||||
// The child of the latest version is NOT_FOUND. The tests in crate::server test more
|
// The child of the latest version is NOT_FOUND. The tests in crate::server test more
|
||||||
// corner cases.
|
// corner cases.
|
||||||
let uri = format!("/v1/client/get-child-version/{}", test_version_id);
|
let uri = format!("/v1/client/get-child-version/{test_version_id}");
|
||||||
let req = test::TestRequest::get()
|
let req = test::TestRequest::get()
|
||||||
.uri(&uri)
|
.uri(&uri)
|
||||||
.append_header((CLIENT_ID_HEADER, client_id.to_string()))
|
.append_header((CLIENT_ID_HEADER, client_id.to_string()))
|
||||||
|
|
|
@ -33,12 +33,12 @@ pub(crate) async fn service(
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use crate::api::CLIENT_ID_HEADER;
|
|
||||||
use crate::WebServer;
|
use crate::WebServer;
|
||||||
|
use crate::{api::CLIENT_ID_HEADER, WebConfig};
|
||||||
use actix_web::{http::StatusCode, test, App};
|
use actix_web::{http::StatusCode, test, App};
|
||||||
use chrono::{TimeZone, Utc};
|
use chrono::{TimeZone, Utc};
|
||||||
use pretty_assertions::assert_eq;
|
use pretty_assertions::assert_eq;
|
||||||
use taskchampion_sync_server_core::{InMemoryStorage, Snapshot, Storage};
|
use taskchampion_sync_server_core::{InMemoryStorage, ServerConfig, Snapshot, Storage};
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
|
@ -53,7 +53,7 @@ mod test {
|
||||||
txn.commit().unwrap();
|
txn.commit().unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
@ -89,7 +89,7 @@ mod test {
|
||||||
txn.commit().unwrap();
|
txn.commit().unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
let server = WebServer::new(Default::default(), None, true, storage);
|
let server = WebServer::new(ServerConfig::default(), WebConfig::default(), storage);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
|
|
@ -1,8 +1,7 @@
|
||||||
use std::collections::HashSet;
|
|
||||||
|
|
||||||
use actix_web::{error, web, HttpRequest, Result, Scope};
|
use actix_web::{error, web, HttpRequest, Result, Scope};
|
||||||
use taskchampion_sync_server_core::{ClientId, Server, ServerError};
|
use taskchampion_sync_server_core::{ClientId, Server, ServerError};
|
||||||
use uuid::Uuid;
|
|
||||||
|
use crate::WebConfig;
|
||||||
|
|
||||||
mod add_snapshot;
|
mod add_snapshot;
|
||||||
mod add_version;
|
mod add_version;
|
||||||
|
@ -31,8 +30,7 @@ pub(crate) const SNAPSHOT_REQUEST_HEADER: &str = "X-Snapshot-Request";
|
||||||
/// The type containing a reference to the persistent state for the server
|
/// The type containing a reference to the persistent state for the server
|
||||||
pub(crate) struct ServerState {
|
pub(crate) struct ServerState {
|
||||||
pub(crate) server: Server,
|
pub(crate) server: Server,
|
||||||
pub(crate) client_id_allowlist: Option<HashSet<Uuid>>,
|
pub(crate) web_config: WebConfig,
|
||||||
pub(crate) create_clients: bool,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ServerState {
|
impl ServerState {
|
||||||
|
@ -44,7 +42,7 @@ impl ServerState {
|
||||||
if let Some(client_id_hdr) = req.headers().get(CLIENT_ID_HEADER) {
|
if let Some(client_id_hdr) = req.headers().get(CLIENT_ID_HEADER) {
|
||||||
let client_id = client_id_hdr.to_str().map_err(|_| badrequest())?;
|
let client_id = client_id_hdr.to_str().map_err(|_| badrequest())?;
|
||||||
let client_id = ClientId::parse_str(client_id).map_err(|_| badrequest())?;
|
let client_id = ClientId::parse_str(client_id).map_err(|_| badrequest())?;
|
||||||
if let Some(allow_list) = &self.client_id_allowlist {
|
if let Some(allow_list) = &self.web_config.client_id_allowlist {
|
||||||
if !allow_list.contains(&client_id) {
|
if !allow_list.contains(&client_id) {
|
||||||
return Err(error::ErrorForbidden("unknown x-client-id"));
|
return Err(error::ErrorForbidden("unknown x-client-id"));
|
||||||
}
|
}
|
||||||
|
@ -81,14 +79,17 @@ fn server_error_to_actix(err: ServerError) -> actix_web::Error {
|
||||||
mod test {
|
mod test {
|
||||||
use super::*;
|
use super::*;
|
||||||
use taskchampion_sync_server_core::InMemoryStorage;
|
use taskchampion_sync_server_core::InMemoryStorage;
|
||||||
|
use uuid::Uuid;
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn client_id_header_allow_all() {
|
fn client_id_header_allow_all() {
|
||||||
let client_id = Uuid::new_v4();
|
let client_id = Uuid::new_v4();
|
||||||
let state = ServerState {
|
let state = ServerState {
|
||||||
server: Server::new(Default::default(), InMemoryStorage::new()),
|
server: Server::new(Default::default(), InMemoryStorage::new()),
|
||||||
client_id_allowlist: None,
|
web_config: WebConfig {
|
||||||
create_clients: true,
|
client_id_allowlist: None,
|
||||||
|
create_clients: true,
|
||||||
|
},
|
||||||
};
|
};
|
||||||
let req = actix_web::test::TestRequest::default()
|
let req = actix_web::test::TestRequest::default()
|
||||||
.insert_header((CLIENT_ID_HEADER, client_id.to_string()))
|
.insert_header((CLIENT_ID_HEADER, client_id.to_string()))
|
||||||
|
@ -102,8 +103,10 @@ mod test {
|
||||||
let client_id_disallowed = Uuid::new_v4();
|
let client_id_disallowed = Uuid::new_v4();
|
||||||
let state = ServerState {
|
let state = ServerState {
|
||||||
server: Server::new(Default::default(), InMemoryStorage::new()),
|
server: Server::new(Default::default(), InMemoryStorage::new()),
|
||||||
client_id_allowlist: Some([client_id_ok].into()),
|
web_config: WebConfig {
|
||||||
create_clients: true,
|
client_id_allowlist: Some([client_id_ok].into()),
|
||||||
|
create_clients: true,
|
||||||
|
},
|
||||||
};
|
};
|
||||||
let req = actix_web::test::TestRequest::default()
|
let req = actix_web::test::TestRequest::default()
|
||||||
.insert_header((CLIENT_ID_HEADER, client_id_ok.to_string()))
|
.insert_header((CLIENT_ID_HEADER, client_id_ok.to_string()))
|
||||||
|
|
|
@ -8,7 +8,7 @@ use actix_web::{
|
||||||
};
|
};
|
||||||
use clap::{arg, builder::ValueParser, value_parser, ArgAction, Command};
|
use clap::{arg, builder::ValueParser, value_parser, ArgAction, Command};
|
||||||
use std::{collections::HashSet, ffi::OsString};
|
use std::{collections::HashSet, ffi::OsString};
|
||||||
use taskchampion_sync_server::WebServer;
|
use taskchampion_sync_server::{WebConfig, WebServer};
|
||||||
use taskchampion_sync_server_core::ServerConfig;
|
use taskchampion_sync_server_core::ServerConfig;
|
||||||
use taskchampion_sync_server_storage_sqlite::SqliteStorage;
|
use taskchampion_sync_server_storage_sqlite::SqliteStorage;
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
@ -111,8 +111,10 @@ async fn main() -> anyhow::Result<()> {
|
||||||
};
|
};
|
||||||
let server = WebServer::new(
|
let server = WebServer::new(
|
||||||
config,
|
config,
|
||||||
server_args.client_id_allowlist,
|
WebConfig {
|
||||||
server_args.create_clients,
|
client_id_allowlist: server_args.client_id_allowlist,
|
||||||
|
create_clients: server_args.create_clients,
|
||||||
|
},
|
||||||
SqliteStorage::new(server_args.data_dir)?,
|
SqliteStorage::new(server_args.data_dir)?,
|
||||||
);
|
);
|
||||||
|
|
||||||
|
@ -364,7 +366,11 @@ mod test {
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_index_get() {
|
async fn test_index_get() {
|
||||||
let server = WebServer::new(Default::default(), None, true, InMemoryStorage::new());
|
let server = WebServer::new(
|
||||||
|
ServerConfig::default(),
|
||||||
|
WebConfig::default(),
|
||||||
|
InMemoryStorage::new(),
|
||||||
|
);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = actix_web::test::init_service(app).await;
|
let app = actix_web::test::init_service(app).await;
|
||||||
|
|
||||||
|
|
|
@ -19,19 +19,32 @@ pub struct WebServer {
|
||||||
server_state: Arc<ServerState>,
|
server_state: Arc<ServerState>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Configuration for WebServer (as distinct from [`ServerConfig`]).
|
||||||
|
pub struct WebConfig {
|
||||||
|
pub client_id_allowlist: Option<HashSet<Uuid>>,
|
||||||
|
pub create_clients: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for WebConfig {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self {
|
||||||
|
client_id_allowlist: Default::default(),
|
||||||
|
create_clients: true,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl WebServer {
|
impl WebServer {
|
||||||
/// Create a new sync server with the given storage implementation.
|
/// Create a new sync server with the given storage implementation.
|
||||||
pub fn new<ST: Storage + 'static>(
|
pub fn new<ST: Storage + 'static>(
|
||||||
config: ServerConfig,
|
config: ServerConfig,
|
||||||
client_id_allowlist: Option<HashSet<Uuid>>,
|
web_config: WebConfig,
|
||||||
create_clients: bool,
|
|
||||||
storage: ST,
|
storage: ST,
|
||||||
) -> Self {
|
) -> Self {
|
||||||
Self {
|
Self {
|
||||||
server_state: Arc::new(ServerState {
|
server_state: Arc::new(ServerState {
|
||||||
server: Server::new(config, storage),
|
server: Server::new(config, storage),
|
||||||
client_id_allowlist,
|
web_config,
|
||||||
create_clients,
|
|
||||||
}),
|
}),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -59,7 +72,11 @@ mod test {
|
||||||
|
|
||||||
#[actix_rt::test]
|
#[actix_rt::test]
|
||||||
async fn test_cache_control() {
|
async fn test_cache_control() {
|
||||||
let server = WebServer::new(Default::default(), None, true, InMemoryStorage::new());
|
let server = WebServer::new(
|
||||||
|
ServerConfig::default(),
|
||||||
|
WebConfig::default(),
|
||||||
|
InMemoryStorage::new(),
|
||||||
|
);
|
||||||
let app = App::new().configure(|sc| server.config(sc));
|
let app = App::new().configure(|sc| server.config(sc));
|
||||||
let app = test::init_service(app).await;
|
let app = test::init_service(app).await;
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue