1 files changed, 26 insertions, 1 deletions
diff --git a/SendGrid/Tests/TestSendgridMessageSetup.cs b/SendGrid/Tests/TestSendgridMessageSetup.cs
index 2e43598..c2ad73d 100755
--- a/SendGrid/Tests/TestSendgridMessageSetup.cs
+++ b/SendGrid/Tests/TestSendgridMessageSetup.cs
@@ -3,6 +3,7 @@ using System.Collections;
 using System.Collections.Generic;
 using System.Linq;
 using System.Net.Mail;
+using System.Net.Mime;
 using System.Text;
 using Moq;
 using NUnit.Framework;
@@ -120,7 +121,6 @@ namespace Tests
         [Test]
         public void TestGetRcpts()
         {
-            //public IEnumerable<String> GetRecipients()
             var foo = new Mock<IHeader>();
             var sg = new SendGrid(foo.Object);
 
@@ -135,5 +135,30 @@ namespace Tests
             Assert.AreEqual("cj@sendgrid.com", rcpts.Skip(2).First(), "getRecipients check Bcc");
             Assert.AreEqual("foo@sendgrid.com", rcpts.Skip(3).First(), "getRecipients check Bcc x2");
         }
+
+        [Test]
+        public void TestAddAttachment()
+        {
+            var foo = new Mock<IHeader>();
+            var sg = new SendGrid(foo.Object);
+
+            var data = new Attachment("pnunit.framework.dll", MediaTypeNames.Application.Octet);
+            sg.AddAttachment("pnunit.framework.dll");
+            sg.AddAttachment("pnunit.framework.dll");
+            Assert.AreEqual(data.ContentStream, sg.Attachments.First().ContentStream, "Attach via path");
+            Assert.AreEqual(data.ContentStream, sg.Attachments.Skip(1).First().ContentStream, "Attach via path x2");
+
+            sg = new SendGrid(foo.Object);
+            sg.AddAttachment(data);
+            sg.AddAttachment(data);
+            Assert.AreEqual(data.ContentStream, sg.Attachments.First().ContentStream, "Attach via attachment");
+            Assert.AreEqual(data.ContentStream, sg.Attachments.Skip(1).First().ContentStream, "Attach via attachment x2");
+
+            sg = new SendGrid(foo.Object);
+            sg.AddAttachment(data.ContentStream, data.ContentType);
+            sg.AddAttachment(data.ContentStream, data.ContentType);
+            Assert.AreEqual(data.ContentStream, sg.Attachments.First().ContentStream, "Attach via stream");
+            Assert.AreEqual(data.ContentStream, sg.Attachments.Skip(1).First().ContentStream, "Attach via stream x2");
+        }
     }
 }