diff options
author | Quentin Dufour <quentin@deuxfleurs.fr> | 2024-03-08 10:20:45 +0100 |
---|---|---|
committer | Quentin Dufour <quentin@deuxfleurs.fr> | 2024-03-08 10:20:45 +0100 |
commit | b9f32d720ae5ec60cadeb492af781ade48cd6cbf (patch) | |
tree | 5b4d01700bddc66c77809ef2cc3183f88ccaf229 /tests | |
parent | 11462f80c4ae25696c7436ed7aacb92074d7e911 (diff) | |
download | aerogramme-b9f32d720ae5ec60cadeb492af781ade48cd6cbf.tar.gz aerogramme-b9f32d720ae5ec60cadeb492af781ade48cd6cbf.zip |
Finalize Aerogramme's refactor
Diffstat (limited to 'tests')
-rw-r--r-- | tests/behavior.rs | 357 | ||||
-rw-r--r-- | tests/common/constants.rs | 54 | ||||
-rw-r--r-- | tests/common/fragments.rs | 570 | ||||
-rw-r--r-- | tests/common/mod.rs | 99 |
4 files changed, 0 insertions, 1080 deletions
diff --git a/tests/behavior.rs b/tests/behavior.rs deleted file mode 100644 index 13baf0e..0000000 --- a/tests/behavior.rs +++ /dev/null @@ -1,357 +0,0 @@ -use anyhow::Context; - -mod common; -use crate::common::constants::*; -use crate::common::fragments::*; - -fn main() { - rfc3501_imap4rev1_base(); - rfc6851_imapext_move(); - rfc4551_imapext_condstore(); - rfc2177_imapext_idle(); - rfc5161_imapext_enable(); // 1 - rfc3691_imapext_unselect(); // 2 - rfc7888_imapext_literal(); // 3 - rfc4315_imapext_uidplus(); // 4 - rfc5819_imapext_liststatus(); // 5 - println!("✅ SUCCESS 🌟🚀🥳🙏🥹"); -} - -fn rfc3501_imap4rev1_base() { - println!("🧪 rfc3501_imap4rev1_base"); - common::aerogramme_provider_daemon_dev(|imap_socket, lmtp_socket| { - connect(imap_socket).context("server says hello")?; - capability(imap_socket, Extension::None).context("check server capabilities")?; - login(imap_socket, Account::Alice).context("login test")?; - create_mailbox(imap_socket, Mailbox::Archive).context("created mailbox archive")?; - let select_res = - select(imap_socket, Mailbox::Inbox, SelectMod::None).context("select inbox")?; - assert!(select_res.contains("* 0 EXISTS")); - - check(imap_socket).context("check must run")?; - status(imap_socket, Mailbox::Archive, StatusKind::UidNext) - .context("status of archive from inbox")?; - lmtp_handshake(lmtp_socket).context("handshake lmtp done")?; - lmtp_deliver_email(lmtp_socket, Email::Multipart).context("mail delivered successfully")?; - noop_exists(imap_socket, 1).context("noop loop must detect a new email")?; - - let srv_msg = fetch( - imap_socket, - Selection::FirstId, - FetchKind::Rfc822, - FetchMod::None, - ) - .context("fetch rfc822 message, should be our first message")?; - let orig_email = std::str::from_utf8(EMAIL1)?; - assert!(srv_msg.contains(orig_email)); - - copy(imap_socket, Selection::FirstId, Mailbox::Archive) - .context("copy message to the archive mailbox")?; - append(imap_socket, Email::Basic).context("insert email in INBOX")?; - noop_exists(imap_socket, 2).context("noop loop must detect a new email")?; - search(imap_socket, SearchKind::Text("OoOoO")).expect("search should return something"); - store( - imap_socket, - Selection::FirstId, - Flag::Deleted, - StoreAction::AddFlags, - StoreMod::None, - ) - .context("should add delete flag to the email")?; - expunge(imap_socket).context("expunge emails")?; - rename_mailbox(imap_socket, Mailbox::Archive, Mailbox::Drafts) - .context("Archive mailbox is renamed Drafts")?; - delete_mailbox(imap_socket, Mailbox::Drafts).context("Drafts mailbox is deleted")?; - Ok(()) - }) - .expect("test fully run"); -} - -fn rfc3691_imapext_unselect() { - println!("🧪 rfc3691_imapext_unselect"); - common::aerogramme_provider_daemon_dev(|imap_socket, lmtp_socket| { - connect(imap_socket).context("server says hello")?; - - lmtp_handshake(lmtp_socket).context("handshake lmtp done")?; - lmtp_deliver_email(lmtp_socket, Email::Basic).context("mail delivered successfully")?; - - capability(imap_socket, Extension::Unselect).context("check server capabilities")?; - login(imap_socket, Account::Alice).context("login test")?; - let select_res = - select(imap_socket, Mailbox::Inbox, SelectMod::None).context("select inbox")?; - assert!(select_res.contains("* 0 EXISTS")); - - noop_exists(imap_socket, 1).context("noop loop must detect a new email")?; - store( - imap_socket, - Selection::FirstId, - Flag::Deleted, - StoreAction::AddFlags, - StoreMod::None, - ) - .context("add delete flags to the email")?; - unselect(imap_socket) - .context("unselect inbox while preserving email with the \\Delete flag")?; - let select_res = - select(imap_socket, Mailbox::Inbox, SelectMod::None).context("select inbox again")?; - assert!(select_res.contains("* 1 EXISTS")); - - let srv_msg = fetch( - imap_socket, - Selection::FirstId, - FetchKind::Rfc822, - FetchMod::None, - ) - .context("message is still present")?; - let orig_email = std::str::from_utf8(EMAIL2)?; - assert!(srv_msg.contains(orig_email)); - - close(imap_socket).context("close inbox and expunge message")?; - let select_res = select(imap_socket, Mailbox::Inbox, SelectMod::None) - .context("select inbox again and check it's empty")?; - assert!(select_res.contains("* 0 EXISTS")); - - Ok(()) - }) - .expect("test fully run"); -} - -fn rfc5161_imapext_enable() { - println!("🧪 rfc5161_imapext_enable"); - common::aerogramme_provider_daemon_dev(|imap_socket, _lmtp_socket| { - connect(imap_socket).context("server says hello")?; - login(imap_socket, Account::Alice).context("login test")?; - enable(imap_socket, Enable::Utf8Accept, Some(Enable::Utf8Accept))?; - enable(imap_socket, Enable::Utf8Accept, None)?; - logout(imap_socket)?; - - Ok(()) - }) - .expect("test fully run"); -} - -fn rfc6851_imapext_move() { - println!("🧪 rfc6851_imapext_move"); - common::aerogramme_provider_daemon_dev(|imap_socket, lmtp_socket| { - connect(imap_socket).context("server says hello")?; - - capability(imap_socket, Extension::Move).context("check server capabilities")?; - login(imap_socket, Account::Alice).context("login test")?; - create_mailbox(imap_socket, Mailbox::Archive).context("created mailbox archive")?; - let select_res = - select(imap_socket, Mailbox::Inbox, SelectMod::None).context("select inbox")?; - assert!(select_res.contains("* 0 EXISTS")); - - lmtp_handshake(lmtp_socket).context("handshake lmtp done")?; - lmtp_deliver_email(lmtp_socket, Email::Basic).context("mail delivered successfully")?; - - noop_exists(imap_socket, 1).context("noop loop must detect a new email")?; - r#move(imap_socket, Selection::FirstId, Mailbox::Archive) - .context("message from inbox moved to archive")?; - - unselect(imap_socket) - .context("unselect inbox while preserving email with the \\Delete flag")?; - let select_res = - select(imap_socket, Mailbox::Archive, SelectMod::None).context("select archive")?; - assert!(select_res.contains("* 1 EXISTS")); - - let srv_msg = fetch( - imap_socket, - Selection::FirstId, - FetchKind::Rfc822, - FetchMod::None, - ) - .context("check mail exists")?; - let orig_email = std::str::from_utf8(EMAIL2)?; - assert!(srv_msg.contains(orig_email)); - - logout(imap_socket).context("must quit")?; - - Ok(()) - }) - .expect("test fully run"); -} - -fn rfc7888_imapext_literal() { - println!("🧪 rfc7888_imapext_literal"); - common::aerogramme_provider_daemon_dev(|imap_socket, _lmtp_socket| { - connect(imap_socket).context("server says hello")?; - - capability(imap_socket, Extension::LiteralPlus).context("check server capabilities")?; - login_with_literal(imap_socket, Account::Alice).context("use literal to connect Alice")?; - - Ok(()) - }) - .expect("test fully run"); -} - -fn rfc4551_imapext_condstore() { - println!("🧪 rfc4551_imapext_condstore"); - common::aerogramme_provider_daemon_dev(|imap_socket, lmtp_socket| { - // Setup the test - connect(imap_socket).context("server says hello")?; - - // RFC 3.1.1 Advertising Support for CONDSTORE - capability(imap_socket, Extension::Condstore).context("check server capabilities")?; - login(imap_socket, Account::Alice).context("login test")?; - - // RFC 3.1.8. CONDSTORE Parameter to SELECT and EXAMINE - let select_res = - select(imap_socket, Mailbox::Inbox, SelectMod::Condstore).context("select inbox")?; - // RFC 3.1.2 New OK Untagged Responses for SELECT and EXAMINE - assert!(select_res.contains("[HIGHESTMODSEQ 1]")); - - // RFC 3.1.3. STORE and UID STORE Commands - lmtp_handshake(lmtp_socket).context("handshake lmtp done")?; - lmtp_deliver_email(lmtp_socket, Email::Basic).context("mail delivered successfully")?; - lmtp_deliver_email(lmtp_socket, Email::Multipart).context("mail delivered successfully")?; - noop_exists(imap_socket, 2).context("noop loop must detect a new email")?; - let store_res = store( - imap_socket, - Selection::All, - Flag::Important, - StoreAction::AddFlags, - StoreMod::UnchangedSince(1), - )?; - assert!(store_res.contains("[MODIFIED 2]")); - assert!(store_res.contains("* 1 FETCH (FLAGS (\\Important) MODSEQ (3))")); - assert!(!store_res.contains("* 2 FETCH")); - assert_eq!(store_res.lines().count(), 2); - - // RFC 3.1.4. FETCH and UID FETCH Commands - let fetch_res = fetch( - imap_socket, - Selection::All, - FetchKind::Rfc822Size, - FetchMod::ChangedSince(2), - )?; - assert!(fetch_res.contains("* 1 FETCH (RFC822.SIZE 81 MODSEQ (3))")); - assert!(!fetch_res.contains("* 2 FETCH")); - assert_eq!(store_res.lines().count(), 2); - - // RFC 3.1.5. MODSEQ Search Criterion in SEARCH - let search_res = search(imap_socket, SearchKind::ModSeq(3))?; - // RFC 3.1.6. Modified SEARCH Untagged Response - assert!(search_res.contains("* SEARCH 1 (MODSEQ 3)")); - - // RFC 3.1.7 HIGHESTMODSEQ Status Data Items - let status_res = status(imap_socket, Mailbox::Inbox, StatusKind::HighestModSeq)?; - assert!(status_res.contains("HIGHESTMODSEQ 3")); - - Ok(()) - }) - .expect("test fully run"); -} - -fn rfc2177_imapext_idle() { - println!("🧪 rfc2177_imapext_idle"); - common::aerogramme_provider_daemon_dev(|imap_socket, lmtp_socket| { - // Test setup, check capability - connect(imap_socket).context("server says hello")?; - capability(imap_socket, Extension::Idle).context("check server capabilities")?; - login(imap_socket, Account::Alice).context("login test")?; - select(imap_socket, Mailbox::Inbox, SelectMod::None).context("select inbox")?; - - // Check that new messages from LMTP are correctly detected during idling - start_idle(imap_socket).context("can't start idling")?; - lmtp_handshake(lmtp_socket).context("handshake lmtp done")?; - lmtp_deliver_email(lmtp_socket, Email::Basic).context("mail delivered successfully")?; - let srv_msg = stop_idle(imap_socket).context("stop idling")?; - assert!(srv_msg.contains("* 1 EXISTS")); - - Ok(()) - }) - .expect("test fully run"); -} - -fn rfc4315_imapext_uidplus() { - println!("🧪 rfc4315_imapext_uidplus"); - common::aerogramme_provider_daemon_dev(|imap_socket, lmtp_socket| { - // Test setup, check capability, insert 2 emails - connect(imap_socket).context("server says hello")?; - capability(imap_socket, Extension::UidPlus).context("check server capabilities")?; - login(imap_socket, Account::Alice).context("login test")?; - select(imap_socket, Mailbox::Inbox, SelectMod::None).context("select inbox")?; - lmtp_handshake(lmtp_socket).context("handshake lmtp done")?; - lmtp_deliver_email(lmtp_socket, Email::Basic).context("mail delivered successfully")?; - lmtp_deliver_email(lmtp_socket, Email::Multipart).context("mail delivered successfully")?; - noop_exists(imap_socket, 2).context("noop loop must detect a new email")?; - - // Check UID EXPUNGE seqset - store( - imap_socket, - Selection::All, - Flag::Deleted, - StoreAction::AddFlags, - StoreMod::None, - )?; - let res = uid_expunge(imap_socket, Selection::FirstId)?; - assert_eq!(res.lines().count(), 2); - assert!(res.contains("* 1 EXPUNGE")); - - // APPENDUID check UID + UID VALIDITY - // Note: 4 and not 3, as we update the UID counter when we delete an email - // it's part of our UID proof - let res = append(imap_socket, Email::Multipart)?; - assert!(res.contains("[APPENDUID 1 4]")); - - // COPYUID, check - create_mailbox(imap_socket, Mailbox::Archive).context("created mailbox archive")?; - let res = copy(imap_socket, Selection::FirstId, Mailbox::Archive)?; - assert!(res.contains("[COPYUID 1 2 1]")); - - // MOVEUID, check - let res = r#move(imap_socket, Selection::FirstId, Mailbox::Archive)?; - assert!(res.contains("[COPYUID 1 2 2]")); - - Ok(()) - }) - .expect("test fully run"); -} - -/// -/// Example -/// -/// ```text -/// 30 list "" "*" RETURN (STATUS (MESSAGES UNSEEN)) -/// * LIST (\Subscribed) "." INBOX -/// * STATUS INBOX (MESSAGES 2 UNSEEN 1) -/// 30 OK LIST completed -/// ``` -fn rfc5819_imapext_liststatus() { - println!("🧪 rfc5819_imapext_liststatus"); - common::aerogramme_provider_daemon_dev(|imap_socket, lmtp_socket| { - // Test setup, check capability, add 2 emails, read 1 - connect(imap_socket).context("server says hello")?; - capability(imap_socket, Extension::ListStatus).context("check server capabilities")?; - login(imap_socket, Account::Alice).context("login test")?; - select(imap_socket, Mailbox::Inbox, SelectMod::None).context("select inbox")?; - lmtp_handshake(lmtp_socket).context("handshake lmtp done")?; - lmtp_deliver_email(lmtp_socket, Email::Basic).context("mail delivered successfully")?; - lmtp_deliver_email(lmtp_socket, Email::Multipart).context("mail delivered successfully")?; - noop_exists(imap_socket, 2).context("noop loop must detect a new email")?; - fetch( - imap_socket, - Selection::FirstId, - FetchKind::Rfc822, - FetchMod::None, - ) - .context("read one message")?; - close(imap_socket).context("close inbox")?; - - // Test return status MESSAGES UNSEEN - let ret = list( - imap_socket, - MbxSelect::All, - ListReturn::StatusMessagesUnseen, - )?; - assert!(ret.contains("* STATUS INBOX (MESSAGES 2 UNSEEN 1)")); - - // Test that without RETURN, no status is sent - let ret = list(imap_socket, MbxSelect::All, ListReturn::None)?; - assert!(!ret.contains("* STATUS")); - - Ok(()) - }) - .expect("test fully run"); -} diff --git a/tests/common/constants.rs b/tests/common/constants.rs deleted file mode 100644 index c11a04d..0000000 --- a/tests/common/constants.rs +++ /dev/null @@ -1,54 +0,0 @@ -use std::time; - -pub static SMALL_DELAY: time::Duration = time::Duration::from_millis(200); - -pub static EMAIL1: &[u8] = b"Date: Sat, 8 Jul 2023 07:14:29 +0200\r -From: Bob Robert <bob@example.tld>\r -To: Alice Malice <alice@example.tld>\r -CC: =?ISO-8859-1?Q?Andr=E9?= Pirard <PIRARD@vm1.ulg.ac.be>\r -Subject: =?ISO-8859-1?B?SWYgeW91IGNhbiByZWFkIHRoaXMgeW8=?=\r - =?ISO-8859-2?B?dSB1bmRlcnN0YW5kIHRoZSBleGFtcGxlLg==?=\r -X-Unknown: something something\r -Bad entry\r - on multiple lines\r -Message-ID: <NTAxNzA2AC47634Y366BAMTY4ODc5MzQyODY0ODY5@www.grrrndzero.org>\r -MIME-Version: 1.0\r -Content-Type: multipart/alternative;\r - boundary=\"b1_e376dc71bafc953c0b0fdeb9983a9956\"\r -Content-Transfer-Encoding: 7bit\r -\r -This is a multi-part message in MIME format.\r -\r ---b1_e376dc71bafc953c0b0fdeb9983a9956\r -Content-Type: text/plain; charset=utf-8\r -Content-Transfer-Encoding: quoted-printable\r -\r -GZ\r -OoOoO\r -oOoOoOoOo\r -oOoOoOoOoOoOoOoOo\r -oOoOoOoOoOoOoOoOoOoOoOo\r -oOoOoOoOoOoOoOoOoOoOoOoOoOoOo\r -OoOoOoOoOoOoOoOoOoOoOoOoOoOoOoOoO\r -\r ---b1_e376dc71bafc953c0b0fdeb9983a9956\r -Content-Type: text/html; charset=us-ascii\r -\r -<div style=\"text-align: center;\"><strong>GZ</strong><br />\r -OoOoO<br />\r -oOoOoOoOo<br />\r -oOoOoOoOoOoOoOoOo<br />\r -oOoOoOoOoOoOoOoOoOoOoOo<br />\r -oOoOoOoOoOoOoOoOoOoOoOoOoOoOo<br />\r -OoOoOoOoOoOoOoOoOoOoOoOoOoOoOoOoO<br />\r -</div>\r -\r ---b1_e376dc71bafc953c0b0fdeb9983a9956--\r -"; - -pub static EMAIL2: &[u8] = b"From: alice@example.com\r -To: alice@example.tld\r -Subject: Test\r -\r -Hello world!\r -"; diff --git a/tests/common/fragments.rs b/tests/common/fragments.rs deleted file mode 100644 index 606af2b..0000000 --- a/tests/common/fragments.rs +++ /dev/null @@ -1,570 +0,0 @@ -use anyhow::{bail, Result}; -use std::io::Write; -use std::net::TcpStream; -use std::thread; - -use crate::common::constants::*; -use crate::common::*; - -/// These fragments are not a generic IMAP client -/// but specialized to our specific tests. They can't take -/// arbitrary values, only enum for which the code is known -/// to be correct. The idea is that the generated message is more -/// or less hardcoded by the developer, so its clear what's expected, -/// and not generated by a library. Also don't use vector of enum, -/// as it again introduce some kind of genericity we try so hard to avoid: -/// instead add a dedicated enum, for example "All" or anything relaevent that would -/// describe your list and then hardcode it in your fragment. -/// DON'T. TRY. TO. BE. GENERIC. HERE. - -pub fn connect(imap: &mut TcpStream) -> Result<()> { - let mut buffer: [u8; 1500] = [0; 1500]; - - let read = read_lines(imap, &mut buffer, None)?; - assert_eq!(&read[..4], &b"* OK"[..]); - - Ok(()) -} - -pub enum Account { - Alice, -} - -pub enum Extension { - None, - Unselect, - Move, - Condstore, - LiteralPlus, - Idle, - UidPlus, - ListStatus, -} - -pub enum Enable { - Utf8Accept, - CondStore, - All, -} - -pub enum Mailbox { - Inbox, - Archive, - Drafts, -} - -pub enum Flag { - Deleted, - Important, -} - -pub enum Email { - Basic, - Multipart, -} - -pub enum Selection { - FirstId, - SecondId, - All, -} - -pub enum SelectMod { - None, - Condstore, -} - -pub enum StoreAction { - AddFlags, - DelFlags, - SetFlags, - AddFlagsSilent, - DelFlagsSilent, - SetFlagsSilent, -} - -pub enum StoreMod { - None, - UnchangedSince(u64), -} - -pub enum FetchKind { - Rfc822, - Rfc822Size, -} - -pub enum FetchMod { - None, - ChangedSince(u64), -} - -pub enum SearchKind<'a> { - Text(&'a str), - ModSeq(u64), -} - -pub enum StatusKind { - UidNext, - HighestModSeq, -} - -pub enum MbxSelect { - All, -} - -pub enum ListReturn { - None, - StatusMessagesUnseen, -} - -pub fn capability(imap: &mut TcpStream, ext: Extension) -> Result<()> { - imap.write(&b"5 capability\r\n"[..])?; - - let maybe_ext = match ext { - Extension::None => None, - Extension::Unselect => Some("UNSELECT"), - Extension::Move => Some("MOVE"), - Extension::Condstore => Some("CONDSTORE"), - Extension::LiteralPlus => Some("LITERAL+"), - Extension::Idle => Some("IDLE"), - Extension::UidPlus => Some("UIDPLUS"), - Extension::ListStatus => Some("LIST-STATUS"), - }; - - let mut buffer: [u8; 6000] = [0; 6000]; - let read = read_lines(imap, &mut buffer, Some(&b"5 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - assert!(srv_msg.contains("IMAP4REV1")); - if let Some(ext) = maybe_ext { - assert!(srv_msg.contains(ext)); - } - - Ok(()) -} - -pub fn login(imap: &mut TcpStream, account: Account) -> Result<()> { - let mut buffer: [u8; 1500] = [0; 1500]; - - assert!(matches!(account, Account::Alice)); - imap.write(&b"10 login alice hunter2\r\n"[..])?; - - let read = read_lines(imap, &mut buffer, None)?; - assert_eq!(&read[..5], &b"10 OK"[..]); - - Ok(()) -} - -pub fn login_with_literal(imap: &mut TcpStream, account: Account) -> Result<()> { - let mut buffer: [u8; 1500] = [0; 1500]; - - assert!(matches!(account, Account::Alice)); - imap.write(&b"10 login {5+}\r\nalice {7+}\r\nhunter2\r\n"[..])?; - let _read = read_lines(imap, &mut buffer, Some(&b"10 OK"[..]))?; - Ok(()) -} - -pub fn create_mailbox(imap: &mut TcpStream, mbx: Mailbox) -> Result<()> { - let mut buffer: [u8; 1500] = [0; 1500]; - - let mbx_str = match mbx { - Mailbox::Inbox => "INBOX", - Mailbox::Archive => "ArchiveCustom", - Mailbox::Drafts => "DraftsCustom", - }; - - let cmd = format!("15 create {}\r\n", mbx_str); - imap.write(cmd.as_bytes())?; - let read = read_lines(imap, &mut buffer, None)?; - assert_eq!(&read[..12], &b"15 OK CREATE"[..]); - - Ok(()) -} - -pub fn list(imap: &mut TcpStream, select: MbxSelect, mod_return: ListReturn) -> Result<String> { - let mut buffer: [u8; 6000] = [0; 6000]; - - let select_str = match select { - MbxSelect::All => "%", - }; - - let mod_return_str = match mod_return { - ListReturn::None => "", - ListReturn::StatusMessagesUnseen => " RETURN (STATUS (MESSAGES UNSEEN))", - }; - - imap.write(format!("19 LIST \"\" \"{}\"{}\r\n", select_str, mod_return_str).as_bytes())?; - - let read = read_lines(imap, &mut buffer, Some(&b"19 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - Ok(srv_msg.to_string()) -} - -pub fn select(imap: &mut TcpStream, mbx: Mailbox, modifier: SelectMod) -> Result<String> { - let mut buffer: [u8; 6000] = [0; 6000]; - - let mbx_str = match mbx { - Mailbox::Inbox => "INBOX", - Mailbox::Archive => "ArchiveCustom", - Mailbox::Drafts => "DraftsCustom", - }; - - let mod_str = match modifier { - SelectMod::Condstore => " (CONDSTORE)", - SelectMod::None => "", - }; - - imap.write(format!("20 select {}{}\r\n", mbx_str, mod_str).as_bytes())?; - - let read = read_lines(imap, &mut buffer, Some(&b"20 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - - Ok(srv_msg.to_string()) -} - -pub fn unselect(imap: &mut TcpStream) -> Result<()> { - imap.write(&b"70 unselect\r\n"[..])?; - let mut buffer: [u8; 1500] = [0; 1500]; - let _read = read_lines(imap, &mut buffer, Some(&b"70 OK"[..]))?; - - Ok(()) -} - -pub fn check(imap: &mut TcpStream) -> Result<()> { - let mut buffer: [u8; 1500] = [0; 1500]; - - imap.write(&b"21 check\r\n"[..])?; - let _read = read_lines(imap, &mut buffer, Some(&b"21 OK"[..]))?; - - Ok(()) -} - -pub fn status(imap: &mut TcpStream, mbx: Mailbox, sk: StatusKind) -> Result<String> { - let mbx_str = match mbx { - Mailbox::Inbox => "INBOX", - Mailbox::Archive => "ArchiveCustom", - Mailbox::Drafts => "DraftsCustom", - }; - let sk_str = match sk { - StatusKind::UidNext => "(UIDNEXT)", - StatusKind::HighestModSeq => "(HIGHESTMODSEQ)", - }; - imap.write(format!("25 STATUS {} {}\r\n", mbx_str, sk_str).as_bytes())?; - let mut buffer: [u8; 6000] = [0; 6000]; - let read = read_lines(imap, &mut buffer, Some(&b"25 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - - Ok(srv_msg.to_string()) -} - -pub fn lmtp_handshake(lmtp: &mut TcpStream) -> Result<()> { - let mut buffer: [u8; 1500] = [0; 1500]; - - let _read = read_lines(lmtp, &mut buffer, None)?; - assert_eq!(&buffer[..4], &b"220 "[..]); - - lmtp.write(&b"LHLO example.tld\r\n"[..])?; - let _read = read_lines(lmtp, &mut buffer, Some(&b"250 "[..]))?; - - Ok(()) -} - -pub fn lmtp_deliver_email(lmtp: &mut TcpStream, email_type: Email) -> Result<()> { - let mut buffer: [u8; 1500] = [0; 1500]; - - let email = match email_type { - Email::Basic => EMAIL2, - Email::Multipart => EMAIL1, - }; - lmtp.write(&b"MAIL FROM:<bob@example.tld>\r\n"[..])?; - let _read = read_lines(lmtp, &mut buffer, Some(&b"250 2.0.0"[..]))?; - - lmtp.write(&b"RCPT TO:<alice@example.tld>\r\n"[..])?; - let _read = read_lines(lmtp, &mut buffer, Some(&b"250 2.1.5"[..]))?; - - lmtp.write(&b"DATA\r\n"[..])?; - let _read = read_lines(lmtp, &mut buffer, Some(&b"354 "[..]))?; - - lmtp.write(email)?; - lmtp.write(&b"\r\n.\r\n"[..])?; - let _read = read_lines(lmtp, &mut buffer, Some(&b"250 2.0.0"[..]))?; - - Ok(()) -} - -pub fn noop_exists(imap: &mut TcpStream, must_exists: u32) -> Result<()> { - let mut buffer: [u8; 6000] = [0; 6000]; - - let mut max_retry = 20; - loop { - max_retry -= 1; - imap.write(&b"30 NOOP\r\n"[..])?; - let read = read_lines(imap, &mut buffer, Some(&b"30 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - - for line in srv_msg.lines() { - if line.contains("EXISTS") { - let got = read_first_u32(line)?; - if got == must_exists { - // Done - return Ok(()); - } - } - } - - if max_retry <= 0 { - // Failed - bail!("no more retry"); - } - - thread::sleep(SMALL_DELAY); - } -} - -pub fn fetch( - imap: &mut TcpStream, - selection: Selection, - kind: FetchKind, - modifier: FetchMod, -) -> Result<String> { - let mut buffer: [u8; 65535] = [0; 65535]; - - let sel_str = match selection { - Selection::FirstId => "1", - Selection::SecondId => "2", - Selection::All => "1:*", - }; - - let kind_str = match kind { - FetchKind::Rfc822 => "RFC822", - FetchKind::Rfc822Size => "RFC822.SIZE", - }; - - let mod_str = match modifier { - FetchMod::None => "".into(), - FetchMod::ChangedSince(val) => format!(" (CHANGEDSINCE {})", val), - }; - - imap.write(format!("40 fetch {} {}{}\r\n", sel_str, kind_str, mod_str).as_bytes())?; - - let read = read_lines(imap, &mut buffer, Some(&b"40 OK FETCH"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - - Ok(srv_msg.to_string()) -} - -pub fn copy(imap: &mut TcpStream, selection: Selection, to: Mailbox) -> Result<String> { - let mut buffer: [u8; 65535] = [0; 65535]; - assert!(matches!(selection, Selection::FirstId)); - assert!(matches!(to, Mailbox::Archive)); - - imap.write(&b"45 copy 1 ArchiveCustom\r\n"[..])?; - let read = read_lines(imap, &mut buffer, None)?; - assert_eq!(&read[..5], &b"45 OK"[..]); - let srv_msg = std::str::from_utf8(read)?; - - Ok(srv_msg.to_string()) -} - -pub fn append(imap: &mut TcpStream, content: Email) -> Result<String> { - let mut buffer: [u8; 6000] = [0; 6000]; - - let ref_mail = match content { - Email::Multipart => EMAIL1, - Email::Basic => EMAIL2, - }; - - let append_cmd = format!("47 append inbox (\\Seen) {{{}}}\r\n", ref_mail.len()); - println!("append cmd: {}", append_cmd); - imap.write(append_cmd.as_bytes())?; - - // wait for continuation - let read = read_lines(imap, &mut buffer, None)?; - assert_eq!(read[0], b'+'); - - // write our stuff - imap.write(ref_mail)?; - imap.write(&b"\r\n"[..])?; - let read = read_lines(imap, &mut buffer, Some(&b"47 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - - Ok(srv_msg.to_string()) -} - -pub fn search(imap: &mut TcpStream, sk: SearchKind) -> Result<String> { - let sk_str = match sk { - SearchKind::Text(x) => format!("TEXT \"{}\"", x), - SearchKind::ModSeq(x) => format!("MODSEQ {}", x), - }; - imap.write(format!("55 SEARCH {}\r\n", sk_str).as_bytes())?; - let mut buffer: [u8; 1500] = [0; 1500]; - let read = read_lines(imap, &mut buffer, Some(&b"55 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - Ok(srv_msg.to_string()) -} - -pub fn store( - imap: &mut TcpStream, - sel: Selection, - flag: Flag, - action: StoreAction, - modifier: StoreMod, -) -> Result<String> { - let mut buffer: [u8; 6000] = [0; 6000]; - - let seq = match sel { - Selection::FirstId => "1", - Selection::SecondId => "2", - Selection::All => "1:*", - }; - - let modif = match modifier { - StoreMod::None => "".into(), - StoreMod::UnchangedSince(val) => format!(" (UNCHANGEDSINCE {})", val), - }; - - let flags_str = match flag { - Flag::Deleted => "(\\Deleted)", - Flag::Important => "(\\Important)", - }; - - let action_str = match action { - StoreAction::AddFlags => "+FLAGS", - StoreAction::DelFlags => "-FLAGS", - StoreAction::SetFlags => "FLAGS", - StoreAction::AddFlagsSilent => "+FLAGS.SILENT", - StoreAction::DelFlagsSilent => "-FLAGS.SILENT", - StoreAction::SetFlagsSilent => "FLAGS.SILENT", - }; - - imap.write(format!("57 STORE {}{} {} {}\r\n", seq, modif, action_str, flags_str).as_bytes())?; - let read = read_lines(imap, &mut buffer, Some(&b"57 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - Ok(srv_msg.to_string()) -} - -pub fn expunge(imap: &mut TcpStream) -> Result<()> { - imap.write(&b"60 expunge\r\n"[..])?; - let mut buffer: [u8; 1500] = [0; 1500]; - let _read = read_lines(imap, &mut buffer, Some(&b"60 OK EXPUNGE"[..]))?; - - Ok(()) -} - -pub fn uid_expunge(imap: &mut TcpStream, sel: Selection) -> Result<String> { - use Selection::*; - let mut buffer: [u8; 6000] = [0; 6000]; - let selstr = match sel { - FirstId => "1", - SecondId => "2", - All => "1:*", - }; - imap.write(format!("61 UID EXPUNGE {}\r\n", selstr).as_bytes())?; - let read = read_lines(imap, &mut buffer, Some(&b"61 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - Ok(srv_msg.to_string()) -} - -pub fn rename_mailbox(imap: &mut TcpStream, from: Mailbox, to: Mailbox) -> Result<()> { - assert!(matches!(from, Mailbox::Archive)); - assert!(matches!(to, Mailbox::Drafts)); - - imap.write(&b"70 rename ArchiveCustom DraftsCustom\r\n"[..])?; - let mut buffer: [u8; 1500] = [0; 1500]; - let read = read_lines(imap, &mut buffer, None)?; - assert_eq!(&read[..5], &b"70 OK"[..]); - - imap.write(&b"71 list \"\" *\r\n"[..])?; - let read = read_lines(imap, &mut buffer, Some(&b"71 OK LIST"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - assert!(!srv_msg.contains(" ArchiveCustom\r\n")); - assert!(srv_msg.contains(" INBOX\r\n")); - assert!(srv_msg.contains(" DraftsCustom\r\n")); - - Ok(()) -} - -pub fn delete_mailbox(imap: &mut TcpStream, mbx: Mailbox) -> Result<()> { - let mbx_str = match mbx { - Mailbox::Inbox => "INBOX", - Mailbox::Archive => "ArchiveCustom", - Mailbox::Drafts => "DraftsCustom", - }; - let cmd = format!("80 delete {}\r\n", mbx_str); - - imap.write(cmd.as_bytes())?; - let mut buffer: [u8; 1500] = [0; 1500]; - let read = read_lines(imap, &mut buffer, None)?; - assert_eq!(&read[..5], &b"80 OK"[..]); - - imap.write(&b"81 list \"\" *\r\n"[..])?; - let read = read_lines(imap, &mut buffer, Some(&b"81 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - assert!(srv_msg.contains(" INBOX\r\n")); - assert!(!srv_msg.contains(format!(" {}\r\n", mbx_str).as_str())); - - Ok(()) -} - -pub fn close(imap: &mut TcpStream) -> Result<()> { - imap.write(&b"60 close\r\n"[..])?; - let mut buffer: [u8; 1500] = [0; 1500]; - let _read = read_lines(imap, &mut buffer, Some(&b"60 OK"[..]))?; - - Ok(()) -} - -pub fn r#move(imap: &mut TcpStream, selection: Selection, to: Mailbox) -> Result<String> { - let mut buffer: [u8; 1500] = [0; 1500]; - assert!(matches!(to, Mailbox::Archive)); - assert!(matches!(selection, Selection::FirstId)); - - imap.write(&b"35 move 1 ArchiveCustom\r\n"[..])?; - let read = read_lines(imap, &mut buffer, Some(&b"35 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - assert!(srv_msg.contains("* 1 EXPUNGE")); - - Ok(srv_msg.to_string()) -} - -pub fn enable(imap: &mut TcpStream, ask: Enable, done: Option<Enable>) -> Result<()> { - let mut buffer: [u8; 6000] = [0; 6000]; - assert!(matches!(ask, Enable::Utf8Accept)); - - imap.write(&b"36 enable UTF8=ACCEPT\r\n"[..])?; - let read = read_lines(imap, &mut buffer, Some(&b"36 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - match done { - None => assert_eq!(srv_msg.lines().count(), 1), - Some(Enable::Utf8Accept) => { - assert_eq!(srv_msg.lines().count(), 2); - assert!(srv_msg.contains("* ENABLED UTF8=ACCEPT")); - } - _ => unimplemented!(), - } - - Ok(()) -} - -pub fn start_idle(imap: &mut TcpStream) -> Result<()> { - let mut buffer: [u8; 1500] = [0; 1500]; - imap.write(&b"98 IDLE\r\n"[..])?; - let read = read_lines(imap, &mut buffer, None)?; - assert_eq!(read[0], b'+'); - Ok(()) -} - -pub fn stop_idle(imap: &mut TcpStream) -> Result<String> { - let mut buffer: [u8; 16536] = [0; 16536]; - imap.write(&b"DONE\r\n"[..])?; - let read = read_lines(imap, &mut buffer, Some(&b"98 OK"[..]))?; - let srv_msg = std::str::from_utf8(read)?; - Ok(srv_msg.to_string()) -} - -pub fn logout(imap: &mut TcpStream) -> Result<()> { - imap.write(&b"99 logout\r\n"[..])?; - let mut buffer: [u8; 1500] = [0; 1500]; - let read = read_lines(imap, &mut buffer, None)?; - assert_eq!(&read[..5], &b"* BYE"[..]); - Ok(()) -} diff --git a/tests/common/mod.rs b/tests/common/mod.rs deleted file mode 100644 index cbe0271..0000000 --- a/tests/common/mod.rs +++ /dev/null @@ -1,99 +0,0 @@ -#![allow(dead_code)] -pub mod constants; -pub mod fragments; - -use anyhow::{bail, Context, Result}; -use std::io::Read; -use std::net::{Shutdown, TcpStream}; -use std::process::Command; -use std::thread; - -use constants::SMALL_DELAY; - -pub fn aerogramme_provider_daemon_dev( - mut fx: impl FnMut(&mut TcpStream, &mut TcpStream) -> Result<()>, -) -> Result<()> { - // Check port is not used (= free) before starting the test - let mut max_retry = 20; - loop { - max_retry -= 1; - match (TcpStream::connect("[::1]:1143"), max_retry) { - (Ok(_), 0) => bail!("something is listening on [::1]:1143 and prevent the test from starting"), - (Ok(_), _) => println!("something is listening on [::1]:1143, maybe a previous daemon quitting, retrying soon..."), - (Err(_), _) => { - println!("test ready to start, [::1]:1143 is free!"); - break - } - } - thread::sleep(SMALL_DELAY); - } - - // Start daemon - let mut daemon = Command::new(env!("CARGO_BIN_EXE_aerogramme")) - .arg("--dev") - .arg("provider") - .arg("daemon") - .spawn()?; - - // Check that our daemon is correctly listening on the free port - let mut max_retry = 20; - let mut imap_socket = loop { - max_retry -= 1; - match (TcpStream::connect("[::1]:1143"), max_retry) { - (Err(e), 0) => bail!("no more retry, last error is: {}", e), - (Err(e), _) => { - println!("unable to connect: {} ; will retry soon...", e); - } - (Ok(v), _) => break v, - } - thread::sleep(SMALL_DELAY); - }; - - // Assuming now it's safe to open a LMTP socket - let mut lmtp_socket = - TcpStream::connect("[::1]:1025").context("lmtp socket must be connected")?; - - println!("-- ready to test imap features --"); - let result = fx(&mut imap_socket, &mut lmtp_socket); - println!("-- test teardown --"); - - imap_socket - .shutdown(Shutdown::Both) - .context("closing imap socket at the end of the test")?; - lmtp_socket - .shutdown(Shutdown::Both) - .context("closing lmtp socket at the end of the test")?; - daemon.kill().context("daemon should be killed")?; - - result.context("all tests passed") -} - -pub fn read_lines<'a, F: Read>( - reader: &mut F, - buffer: &'a mut [u8], - stop_marker: Option<&[u8]>, -) -> Result<&'a [u8]> { - let mut nbytes = 0; - loop { - nbytes += reader.read(&mut buffer[nbytes..])?; - //println!("partial read: {}", std::str::from_utf8(&buffer[..nbytes])?); - let pre_condition = match stop_marker { - None => true, - Some(mark) => buffer[..nbytes].windows(mark.len()).any(|w| w == mark), - }; - if pre_condition && nbytes >= 2 && &buffer[nbytes - 2..nbytes] == &b"\r\n"[..] { - break; - } - } - println!("read: {}", std::str::from_utf8(&buffer[..nbytes])?); - Ok(&buffer[..nbytes]) -} - -pub fn read_first_u32(inp: &str) -> Result<u32> { - Ok(inp - .chars() - .skip_while(|c| !c.is_digit(10)) - .take_while(|c| c.is_digit(10)) - .collect::<String>() - .parse::<u32>()?) -} |