tests: Extend self_test

duesee · Dec 21, 2023 · ad39866 · ad39866
1 parent 4972ad6
commit ad39866
Showing 1 changed file with 47 additions and 5 deletions.
diff --git a/tests/tests.rs b/tests/tests.rs
@@ -1,7 +1,12 @@
-use imap_codec::imap_types::response::Greeting;
+use imap_codec::imap_types::{
+    auth::AuthMechanism,
+    command::{Command, CommandBody},
+    core::Tag,
+    response::{Greeting, Status},
+};
 use imap_flow::{
-    client::{ClientFlow, ClientFlowOptions},
-    server::{ServerFlow, ServerFlowOptions},
+    client::{ClientFlow, ClientFlowEvent, ClientFlowOptions},
+    server::{ServerFlow, ServerFlowEvent, ServerFlowOptions},
     stream::AnyStream,
 };
 use tokio::net::{TcpListener, TcpStream};
@@ -20,24 +25,61 @@ async fn self_test() {
         async move {
             let (stream, _) = listener.accept().await.unwrap();
 
-            ServerFlow::send_greeting(
+            let (mut server, _) = ServerFlow::send_greeting(
                 AnyStream::new(stream),
                 ServerFlowOptions::default(),
                 greeting.clone(),
             )
             .await
             .unwrap();
+
+            loop {
+                match server.progress().await.unwrap() {
+                    ServerFlowEvent::CommandReceived { command } => {
+                        let no = Status::no(Some(command.tag), None, "...").unwrap();
+                        server.enqueue_status(no);
+                    }
+                    ServerFlowEvent::CommandAuthenticateReceived {
+                        command_authenticate,
+                    } => {
+                        let no = Status::no(Some(command_authenticate.tag), None, "...").unwrap();
+                        server.enqueue_status(no);
+                    }
+                    _ => {}
+                }
+            }
         }
     };
 
     let _ = tokio::task::spawn(server);
 
-    let (_, received_greeting) = {
+    let (mut client, received_greeting) = {
         let stream = TcpStream::connect(("127.0.0.1", port)).await.unwrap();
         ClientFlow::receive_greeting(AnyStream::new(stream), ClientFlowOptions::default())
             .await
             .unwrap()
     };
 
     assert_eq!(greeting, received_greeting);
+
+    client.enqueue_command(Command::new(Tag::unvalidated("A1"), CommandBody::Capability).unwrap());
+
+    loop {
+        match client.progress().await.unwrap() {
+            ClientFlowEvent::StatusReceived { .. } => {
+                client.enqueue_command(
+                    Command::new(
+                        Tag::unvalidated("A2"),
+                        CommandBody::Authenticate {
+                            mechanism: AuthMechanism::Plain,
+                            initial_response: None,
+                        },
+                    )
+                    .unwrap(),
+                );
+            }
+            ClientFlowEvent::AuthenticateRejected { .. } => break,
+            _ => {}
+        }
+    }
 }