rust/lettre
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test(transport): add sendmail transport

Browse Source
This commit is contained in:
Alexis Mousset
2016-10-23 20:42:13 +02:00
parent cc6ca7633d
commit 73e7aa3639
8 changed files with 92 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/email/mod.rs
View File

@@ -82,8 +82,6 @@ impl IntoEmail for SimpleEmail {
builder.add_cc(cc_address.into_mailbox());
}
// No bcc for now
if self.reply_to.is_some() {
builder.add_reply_to(self.reply_to.unwrap().into_mailbox());
}

22
src/lib.rs
View File

@@ -180,6 +180,28 @@
//! let _ = email_client.quit();
//! ```
//!
//! ### Sendmail transport
//!
//! The sendmail transport sends the email using the local sendmail command.
//!
//! ```rust
//! use lettre::transport::sendmail::SendmailTransport;
//! use lettre::transport::EmailTransport;
//! use lettre::email::EmailBuilder;
//!
//! let email = EmailBuilder::new()
//! .to("root@localhost")
//! .from("user@localhost")
//! .body("Hello World!")
//! .subject("Hello")
//! .build()
//! .unwrap();
//!
//! let mut sender = SendmailTransport::new();
//! let result = sender.send(email);
//! assert!(result.is_ok());
//! ```
//!
//! ### Stub transport
//!
//! The stub transport only logs message envelope and drops the content. It can be useful for

2
src/transport/file/mod.rs
View File

@@ -38,7 +38,7 @@ impl EmailTransport<FileResult> for FileEmailTransport {
email.to_addresses().join("> to=<"));
try!(f.write_all(log_line.as_bytes()));
try!(f.write_all(email.message().clone().as_bytes()));
try!(f.write_all(email.message().as_bytes()));
info!("{} status=<written>", log_line);

50
src/transport/sendmail/mod.rs
View File

@@ -1,7 +1,6 @@
//! This transport uilizes the sendmail executable for each email.
use email::SendableEmail;
use std::error::Error;
use std::io::prelude::*;
use std::process::{Command, Stdio};
@@ -10,32 +9,49 @@ use transport::sendmail::error::SendmailResult;
pub mod error;
/// Writes the content and the envelope information to a file
pub struct SendmailTransport;
/// Sends an email using the `sendmail` command
#[derive(Debug,Default)]
pub struct SendmailTransport {
command: String,
}
impl SendmailTransport {
/// Creates a new transport with the default `/usr/sbin/sendmail` command
pub fn new() -> SendmailTransport {
SendmailTransport { command: "/usr/sbin/sendmail".to_string() }
}
/// Creates a new transport to the given sendmail command
pub fn new_with_command<S: Into<String>>(command: S) -> SendmailTransport {
SendmailTransport { command: command.into() }
}
}
impl EmailTransport<SendmailResult> for SendmailTransport {
fn send<T: SendableEmail>(&mut self, email: T) -> SendmailResult {
// Spawn the `wc` command
let process = try!(Command::new("/usr/sbin/sendmail")
.args(&email.to_addresses())
// Spawn the sendmail command
let mut process = try!(Command::new(&self.command)
.args(&["-i", "-f", &email.from_address(), &email.to_addresses().join(" ")])
.stdin(Stdio::piped())
.stdout(Stdio::piped())
.spawn());
match process.stdin.unwrap().write_all(email.message().clone().as_bytes()) {
Err(why) => error!("couldn't write to sendmail stdin: {}",
why.description()),
Ok(_) => info!("sent pangram to sendmail"),
match process.stdin.as_mut().unwrap().write_all(email.message().as_bytes()) {
Ok(_) => (),
Err(error) => return Err(From::from(error)),
}
let mut s = String::new();
match process.stdout.unwrap().read_to_string(&mut s) {
Err(why) => error!("couldn't read sendmail stdout: {}",
why.description()),
Ok(_) => info!("sendmail responded with:\n{}", s),
info!("Wrote message to stdin");
if let Ok(output) = process.wait_with_output() {
if output.status.success() {
Ok(())
} else {
Err(From::from("The message could not be sent"))
}
} else {
Err(From::from("The sendmail process stopped"))
}
Ok(())
}
fn close(&mut self) {

20
tests/transport_file.rs
View File

@@ -1,24 +1,24 @@
extern crate lettre;
use lettre::email::{EmailBuilder, SendableEmail};
use lettre::transport::EmailTransport;
use lettre::transport::file::FileEmailTransport;
use std::env::temp_dir;
use std::fs::File;
use std::fs::remove_file;
use std::io::Read;
use lettre::transport::file::FileEmailTransport;
use lettre::transport::EmailTransport;
use lettre::email::{EmailBuilder, SendableEmail};
#[test]
fn file_transport() {
let mut sender = FileEmailTransport::new(temp_dir());
let email = EmailBuilder::new()
.to("root@localhost")
.from("user@localhost")
.body("Hello World!")
.subject("Hello")
.build()
.unwrap();
.to("root@localhost")
.from("user@localhost")
.body("Hello World!")
.subject("Hello file")
.build()
.unwrap();
let result = sender.send(email.clone());
assert!(result.is_ok());

19
tests/transport_sendmail.rs
View File

@@ -1,19 +1,20 @@
extern crate lettre;
use lettre::transport::sendmail::SendmailTransport;
use lettre::transport::EmailTransport;
use lettre::email::EmailBuilder;
use lettre::transport::EmailTransport;
use lettre::transport::sendmail::SendmailTransport;
#[test]
fn sendmail_transport_simple() {
let mut sender = SendmailTransport;
let mut sender = SendmailTransport::new();
let email = EmailBuilder::new()
.to("root@localhost")
.from("user@localhost")
.body("Hello World!")
.subject("Hello")
.build()
.unwrap();
.to("root@localhost")
.from("user@localhost")
.body("Hello World!")
.subject("Hello sendmail")
.build()
.unwrap();
let result = sender.send(email);
println!("{:?}", result);
assert!(result.is_ok());
}

16
tests/transport_smtp.rs
View File

@@ -1,19 +1,19 @@
extern crate lettre;
use lettre::transport::smtp::SmtpTransportBuilder;
use lettre::transport::EmailTransport;
use lettre::email::EmailBuilder;
use lettre::transport::EmailTransport;
use lettre::transport::smtp::SmtpTransportBuilder;
#[test]
fn smtp_transport_simple() {
let mut sender = SmtpTransportBuilder::localhost().unwrap().build();
let email = EmailBuilder::new()
.to("root@localhost")
.from("user@localhost")
.body("Hello World!")
.subject("Hello")
.build()
.unwrap();
.to("root@localhost")
.from("user@localhost")
.body("Hello World!")
.subject("Hello smtp")
.build()
.unwrap();
let result = sender.send(email);
assert!(result.is_ok());
}

16
tests/transport_stub.rs
View File

@@ -1,19 +1,19 @@
extern crate lettre;
use lettre::transport::stub::StubEmailTransport;
use lettre::transport::EmailTransport;
use lettre::email::EmailBuilder;
use lettre::transport::EmailTransport;
use lettre::transport::stub::StubEmailTransport;
#[test]
fn stub_transport() {
let mut sender = StubEmailTransport;
let email = EmailBuilder::new()
.to("root@localhost")
.from("user@localhost")
.body("Hello World!")
.subject("Hello")
.build()
.unwrap();
.to("root@localhost")
.from("user@localhost")
.body("Hello World!")
.subject("Hello stub")
.build()
.unwrap();
let result = sender.send(email);
assert!(result.is_ok());
}