MoreThanText/morethantext
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Made request pass session to message.

Browse Source
This commit is contained in:
Jeff Baskin 2025-04-08 09:30:04 -04:00
parent f9f64dae55
commit 7792e7ca82
4 changed files with 91 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
src/client.rs
View File

@ -1,4 +1,5 @@
use crate::{ use crate::{
field::Field,
queue::{Message, MsgType, Queue}, queue::{Message, MsgType, Queue},
utils::GenID, utils::GenID,
}; };
@ -14,20 +15,46 @@ use uuid::Uuid;
const RESPONS_TO: [MsgType; 1] = [MsgType::Session]; const RESPONS_TO: [MsgType; 1] = [MsgType::Session];
pub struct Request; pub struct Request {
pub session: Option<Field>,
}
impl Request { impl Request {
pub fn new() -> Self { pub fn new(session: Option<Field>) -> Self {
Self {} Self { session: session }
} }
} }
#[cfg(test)] #[cfg(test)]
mod requests { pub mod requests {
use super::*; use super::*;
pub fn create_request() -> Request { pub fn get_root_document() -> Request {
Request::new() Request::new(None)
}
pub fn get_root_document_eith_session<F>(id: F) -> Request
where
F: Into<Field>,
{
Request::new(Some(id.into()))
}
#[test]
fn new_request_no_session() {
let sess: Option<Field> = None;
let req = Request::new(sess);
assert!(req.session.is_none(), "should not have a session")
}
#[test]
fn new_request_with_session() {
let id = Uuid::new_v4();
let req = Request::new(Some(id.into()));
match req.session {
Some(result) => assert_eq!(result.to_uuid().unwrap(), id),
None => unreachable!("should contain a session"),
}
} }
} }
@ -193,7 +220,7 @@ mod clientlinks {
let (tx, rx) = channel(); let (tx, rx) = channel();
let mut registry = ClientRegistry::new(); let mut registry = ClientRegistry::new();
let mut link = ClientLink::new(tx, registry.clone()); let mut link = ClientLink::new(tx, registry.clone());
let req = Request::new(); let req = Request::new(None);
let rx_client = link.send(req); let rx_client = link.send(req);
let msg = rx.recv_timeout(TIMEOUT).unwrap(); let msg = rx.recv_timeout(TIMEOUT).unwrap();
match msg.get_class() { match msg.get_class() {
@ -266,7 +293,7 @@ impl Client {
#[cfg(test)] #[cfg(test)]
mod clients { mod clients {
use super::*; use super::*;
use requests::create_request; use requests::get_root_document;
use std::time::Duration; use std::time::Duration;
static TIMEOUT: Duration = Duration::from_millis(500); static TIMEOUT: Duration = Duration::from_millis(500);
@ -277,7 +304,7 @@ mod clients {
let queue = Queue::new(); let queue = Queue::new();
queue.add(tx, [MsgType::SessionValidate].to_vec()); queue.add(tx, [MsgType::SessionValidate].to_vec());
let mut link = Client::start(queue.clone()); let mut link = Client::start(queue.clone());
let req = create_request(); let req = get_root_document();
let reply_rx = link.send(req); let reply_rx = link.send(req);
let sess = rx.recv_timeout(TIMEOUT).unwrap(); let sess = rx.recv_timeout(TIMEOUT).unwrap();
match sess.get_class() { match sess.get_class() {

2
src/lib.rs
View File

@ -27,7 +27,7 @@ impl MoreThanText {
where where
F: Into<Field>, F: Into<Field>,
{ {
let req = Request::new(); let req = Request::new(None);
let rx = self.client_link.send(req); let rx = self.client_link.send(req);
rx.recv().unwrap() rx.recv().unwrap()
} }

26
src/main.rs
View File

@ -67,7 +67,7 @@ mod servers {
body::Body, body::Body,
http::{Request, StatusCode}, http::{Request, StatusCode},
}; };
use tower::ServiceExt; use tower::{Service, ServiceExt};
#[tokio::test] #[tokio::test]
async fn get_home_page() { async fn get_home_page() {
@ -100,4 +100,28 @@ mod servers {
holder.push(sessid); holder.push(sessid);
} }
} }
// Next test to complete.
//#[tokio::test]
async fn session_only_sent_once() {
let mut app = mtt_conn.with_state(MoreThanText::new());
let request = Request::builder().uri("/").body(Body::empty()).unwrap();
let response = ServiceExt::<Request<Body>>::ready(&mut app)
.await
.unwrap()
.call(request)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let request = Request::builder().uri("/").body(Body::empty()).unwrap();
let response = ServiceExt::<Request<Body>>::ready(&mut app)
.await
.unwrap()
.call(request)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let session = response.headers().get("set-cookie");
assert!(session.is_none(), "should only send the session once");
}
} }

32
src/queue.rs
View File

@ -58,15 +58,20 @@ impl Message {
} }
impl From<Request> for Message { impl From<Request> for Message {
fn from(_value: Request) -> Self { fn from(value: Request) -> Self {
let msg = Message::new(MsgType::ClientRequest); let mut msg = Message::new(MsgType::ClientRequest);
msg.reply(MsgType::ClientRequest) match value.session {
Some(id) => msg.add_data("sess_id", id),
None => {}
}
msg
} }
} }
#[cfg(test)] #[cfg(test)]
mod messages { mod messages {
use super::*; use super::*;
use crate::client::requests::{get_root_document, get_root_document_eith_session};
#[test] #[test]
fn new_message() { fn new_message() {
@ -130,6 +135,27 @@ mod messages {
let msg = Message::new(MsgType::Session); let msg = Message::new(MsgType::Session);
assert_eq!(msg.get_id(), msg.id); assert_eq!(msg.get_id(), msg.id);
} }
#[test]
fn from_request_no_session() {
let req = get_root_document();
let msg: Message = req.into();
assert!(
msg.get_data("sess_id").is_none(),
"should not have a session id"
)
}
#[test]
fn from_request_with_session() {
let id = Uuid::new_v4();
let req = get_root_document_eith_session(id.clone());
let msg: Message = req.into();
match msg.get_data("sess_id") {
Some(result) => assert_eq!(result.to_uuid().unwrap(), id),
None => unreachable!("should return an id"),
}
}
} }
#[derive(Clone)] #[derive(Clone)]