Add tests for ServerInfo

src/extension.rs

@@ -39,19 +39,17 @@ pub enum Extension {
 
 impl Display for Extension {
     fn fmt(&self, f: &mut Formatter) -> fmt::Result {
-        write!(f, "{}",
-            match *self {
-                Extension::EightBitMime => "8BITMIME",
-                Extension::SmtpUtfEight => "SMTPUTF8",
-                Extension::StartTls => "STARTTLS",
-                Extension::Authentication(_) => "AUTH",
-            }
-        )
+        match *self {
+            Extension::EightBitMime => write!(f, "{}", "8BITMIME"),
+            Extension::SmtpUtfEight => write!(f, "{}", "SMTPUTF8"),
+            Extension::StartTls => write!(f, "{}", "STARTTLS"),
+            Extension::Authentication(ref mecanism) => write!(f, "{} {}", "AUTH", mecanism),
+        }
     }
 }
 
 /// Contains information about an SMTP server
-#[derive(Clone,Debug)]
+#[derive(Clone,Debug,Eq,PartialEq)]
 pub struct ServerInfo {
     /// Server name
     ///
@@ -127,21 +125,82 @@ mod test {
 	use std::collections::HashSet;
 	
     use super::{ServerInfo, Extension};
+    use authentication::Mecanism;
+    use response::{Code, Response, Severity, Category};
+    
+    #[test]
+    fn test_extension_fmt() {
+		assert_eq!(format!("{}", Extension::EightBitMime), "8BITMIME".to_string());
+		assert_eq!(format!("{}", Extension::Authentication(Mecanism::Plain)), "AUTH PLAIN".to_string());
+    }
     
     #[test]
     fn test_serverinfo_fmt() {
     	let mut eightbitmime = HashSet::new();
     	assert!(eightbitmime.insert(Extension::EightBitMime));
     	
-    	let empty = HashSet::new();
-    	
         assert_eq!(format!("{}", ServerInfo{
             name: "name".to_string(),
             esmtp_features: eightbitmime.clone()
         }), "name with {EightBitMime}".to_string());
+        
+        let empty = HashSet::new();
+        
         assert_eq!(format!("{}", ServerInfo{
             name: "name".to_string(),
             esmtp_features: empty,
         }), "name with no supported features".to_string());
+        
+        let mut plain = HashSet::new();
+    	assert!(plain.insert(Extension::Authentication(Mecanism::Plain)));
+    	
+    	assert_eq!(format!("{}", ServerInfo{
+            name: "name".to_string(),
+            esmtp_features: plain.clone()
+        }), "name with {Authentication(Plain)}".to_string());
+    }
+    
+    #[test]
+    fn test_serverinfo() {
+    	let response = Response::new(
+            Code::new(Severity::PositiveCompletion, Category::Unspecified4, 1),
+            vec!["me".to_string(), "8BITMIME".to_string(), "SIZE 42".to_string()]
+        );
+    	
+    	let mut features = HashSet::new();
+    	assert!(features.insert(Extension::EightBitMime));
+    	
+    	let server_info = ServerInfo {
+    		name: "me".to_string(),
+    		esmtp_features: features,
+    	};
+    	
+    	assert_eq!(ServerInfo::from_response(&response).unwrap(), server_info);
+    	
+    	assert!(server_info.supports_feature(&Extension::EightBitMime));
+    	assert!(!server_info.supports_feature(&Extension::StartTls));
+    	assert!(!server_info.supports_auth_mecanism(Mecanism::CramMd5));
+    	
+    	let response2 = Response::new(
+            Code::new(Severity::PositiveCompletion, Category::Unspecified4, 1),
+            vec!["me".to_string(), "AUTH PLAIN CRAM-MD5 OTHER".to_string(), "8BITMIME".to_string(), "SIZE 42".to_string()]
+        );
+    	
+    	let mut features2 = HashSet::new();
+    	assert!(features2.insert(Extension::EightBitMime));
+    	assert!(features2.insert(Extension::Authentication(Mecanism::Plain)));
+    	assert!(features2.insert(Extension::Authentication(Mecanism::CramMd5)));
+    	
+    	let server_info2 = ServerInfo {
+    		name: "me".to_string(),
+    		esmtp_features: features2,
+    	};
+    	
+    	assert_eq!(ServerInfo::from_response(&response2).unwrap(), server_info2);
+    	
+    	assert!(server_info2.supports_feature(&Extension::EightBitMime));
+    	assert!(server_info2.supports_auth_mecanism(Mecanism::Plain));
+    	assert!(server_info2.supports_auth_mecanism(Mecanism::CramMd5));
+    	assert!(!server_info2.supports_feature(&Extension::StartTls));
     }
 }