parent
52e9882658
commit
189dec7785
GPG Key ID:
90734A9E619C8A6C
@ -0,0 +1,29 @@ |
||||
mod harness; |
||||
use harness::RpcState; |
||||
|
||||
use irdest_sdk::IrdestSdk; |
||||
use irpc_sdk::error::RpcResult; |
||||
|
||||
/// A simple test that connects to an Irdest instance over RPC
|
||||
#[async_std::test] |
||||
async fn user_create() -> RpcResult<()> { |
||||
// Create a small test network with 2 RPC sockets
|
||||
let state = RpcState::new(6060, 7070).await; |
||||
|
||||
// Register a service on one of them
|
||||
let serv = harness::make_service(6060).await?; |
||||
|
||||
// Initialise Irdest SDK
|
||||
let sdk = IrdestSdk::connect(&serv)?; |
||||
|
||||
// Create a user
|
||||
let auth = sdk |
||||
.users() |
||||
.create("dont write your passwords in unit tests duh") |
||||
.await?; |
||||
|
||||
println!("User auth: {:?}", auth); |
||||
let is_auth = dbg!(sdk.users().is_authenticated(auth.clone()).await); |
||||
assert_eq!(is_auth, Ok(())); |
||||
Ok(()) |
||||
} |
||||
@ -0,0 +1,90 @@ |
||||
use irdest_core::{rpc::RpcServer, Irdest, IrdestRef}; |
||||
use irpc_broker::Broker; |
||||
use irpc_sdk::{error::RpcResult, Capabilities, RpcSocket, Service}; |
||||
use ratman_harness::{Initialize, ThreePoint}; |
||||
use std::{sync::Arc, time::Duration}; |
||||
use tracing::info; |
||||
use tracing_subscriber::{filter::LevelFilter, fmt, EnvFilter}; |
||||
|
||||
pub use async_std::future::timeout; |
||||
pub use ratman_harness::{millis, sec10, sec5}; |
||||
|
||||
fn parse_log_level() { |
||||
let filter = EnvFilter::try_from_env("QAUL_LOG") |
||||
.unwrap_or_default() |
||||
.add_directive(LevelFilter::DEBUG.into()) |
||||
.add_directive("async_std=error".parse().unwrap()) |
||||
.add_directive("async_io=error".parse().unwrap()) |
||||
.add_directive("polling=error".parse().unwrap()) |
||||
.add_directive("mio=error".parse().unwrap()); |
||||
|
||||
// Initialise the logger
|
||||
fmt().with_env_filter(filter).init(); |
||||
info!("Initialised logger: welcome to qaul-hubd!"); |
||||
} |
||||
|
||||
pub struct TestServer { |
||||
inner: Arc<RpcServer>, |
||||
port: u16, |
||||
} |
||||
|
||||
impl TestServer { |
||||
/// Create an RPC server with a random binding
|
||||
pub async fn new(ir: IrdestRef, port: u16) -> Self { |
||||
Self { |
||||
port, |
||||
inner: RpcServer::new(ir, "127.0.0.1", port).await.unwrap(), |
||||
} |
||||
} |
||||
} |
||||
|
||||
#[allow(unused)] |
||||
pub struct RpcState { |
||||
pub tp: ThreePoint<Arc<Irdest>>, |
||||
// Node A state
|
||||
rpc_a: TestServer, |
||||
broker_a: Arc<Broker>, |
||||
// Node B state
|
||||
// rpc_b: TestServer,
|
||||
// broker_b: Arc<Broker>,
|
||||
} |
||||
|
||||
impl RpcState { |
||||
pub async fn new(a: u16, b: u16) -> Self { |
||||
// parse_log_level(); // If something doesn't work, enable this line!
|
||||
let tp = init().await; |
||||
|
||||
let broker_a = Broker::bind("127.0.0.1", a).await.unwrap(); |
||||
let rpc_a = TestServer::new(Arc::clone(&tp.a.1.as_ref().unwrap()), a).await; |
||||
|
||||
// let broker_b = Broker::bind("127.0.0.1", b).await.unwrap();
|
||||
// let rpc_b = TestServer::new(Arc::clone(&tp.b.1.as_ref().unwrap()), b).await;
|
||||
|
||||
Self { |
||||
tp, |
||||
rpc_a, |
||||
broker_a, |
||||
// rpc_b,
|
||||
// broker_b,
|
||||
} |
||||
} |
||||
} |
||||
|
||||
pub async fn zzz(dur: Duration) { |
||||
async_std::task::sleep(dur).await |
||||
} |
||||
|
||||
pub async fn make_service(port: u16) -> RpcResult<Service> { |
||||
let socket = RpcSocket::connect("127.0.0.1", port).await?; |
||||
let mut service = Service::new("test", 1, "A test service"); |
||||
service |
||||
.register(&socket, Capabilities::basic_json()) |
||||
.await?; |
||||
Ok(service) |
||||
} |
||||
|
||||
async fn init() -> ThreePoint<Arc<Irdest>> { |
||||
let mut tp = ThreePoint::new().await; |
||||
tp.init_with(|_, arc| Irdest::new(arc)); |
||||
tp |
||||
} |
||||
Loading…
Reference in new issue