From 2aa66be25dda8c87d6e921eb0322a459d8281aba Mon Sep 17 00:00:00 2001
From: Matt McCoy <mattnenterprise@yahoo.com>
Date: Tue, 21 Jun 2016 23:16:03 -0400
Subject: [PATCH] Adding capability tests

---
 src/client.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/client.rs b/src/client.rs
index 8c37878..f03d7ed 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -485,6 +485,18 @@ mod tests {
 		assert!(mailbox == expected_mailbox, "Unexpected mailbox returned");
 	}
 
+	#[test]
+	fn capability() {
+		let response = b"* CAPABILITY IMAP4rev1 STARTTLS AUTH=GSSAPI LOGINDISABLED\r\n\
+			a1 OK CAPABILITY completed\r\n".to_vec();
+		let expected_capabilities = vec!["IMAP4rev1", "STARTTLS", "AUTH=GSSAPI", "LOGINDISABLED"];
+		let mock_stream = MockStream::new(response);
+		let mut client = create_client_with_mock_stream(mock_stream);
+		let capabilities = client.capability().unwrap();
+		assert!(client.stream.written_buf == b"a1 CAPABILITY\r\n".to_vec(), "Invalid capability command");
+		assert!(capabilities == expected_capabilities, "Unexpected capabilities response");
+	}
+
 	#[test]
 	fn create() {
 		// TODO Make sure the response was read correctly