1 files changed, 75 insertions, 6 deletions
diff --git a/SendGrid/Tests/TestHeader.cs b/SendGrid/Tests/TestHeader.cs
index 323e394..8a130e9 100755
--- a/SendGrid/Tests/TestHeader.cs
+++ b/SendGrid/Tests/TestHeader.cs
@@ -1,10 +1,11 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using System.Net.Mail;
 using System.Text;
 using Moq;
 using NUnit.Framework;
-using SendGrid;
+using SendGridMail;
 
 namespace Tests
 {
@@ -12,14 +13,82 @@ namespace Tests
     public class TestHeader
     {
         [Test]
-        public void TestAddTo()
+        public void TestAddSubVal()
         {
-            var foo = new Mock<IHeader>();
-            foo.Setup(m => m.Enable("foo"));
+            var test = new Header();
+            test.AddSubVal("foo", new List<string>{"bar", "raz"});
+            var result = test.AsJson();
+            Assert.AreEqual("{\"sub\" : {\"foo\" : [\"bar\", \"raz\"]}}", result);            
+        }
+
+        [Test]
+        public void TestAddUniqueIdentifier()
+        {
+            var test = new Header();
+            test.AddUniqueIdentifier(new Dictionary<string, string>(){{"foo", "bar"}});
+            var result = test.AsJson();
+            Assert.AreEqual("{\"unique_args\" : {\"foo\" : \"bar\"}}", result);
+        }
+
+        [Test]
+        public void TestSetCategory()
+        {
+            var test = new Header();
+            test.SetCategory("foo");
+            var result = test.AsJson();
+            Assert.AreEqual("{\"category\" : \"foo\"}", result);
+        }
 
-            var bar = new SendGrid.SendGrid(foo.Object);
-            Assert.AreEqual(1, 2, "I suck");
+        [Test]
+        public void TestEnable()
+        {
+            var test = new Header();
+            test.Enable("foo");
+            var result = test.AsJson();
+            Assert.AreEqual("{\"filters\" : {\"foo\" : {\"settings\" : {\"enable\" : \"1\"}}}}", result); 
+        }
+
+        [Test]
+        public void TestDisable()
+        {
+            var test = new Header();
+            test.Disable("foo");
+            var result = test.AsJson();
+            Assert.AreEqual("{\"filters\" : {\"foo\" : {\"settings\" : {\"enable\" : \"0\"}}}}", result);
+        }
+
+        [Test]
+        public void TestAddFilterSetting()
+        {
+            var test = new Header();
+            test.AddFilterSetting("foo", new List<string> { "a", "b" }, "bar");
+            var result = test.AsJson();
+            Assert.AreEqual("{\"filters\" : {\"foo\" : {\"settings\" : {\"a\" : {\"b\" : \"bar\"}}}}}", result);
+            
+        }
+
+        [Test]
+        public void TestAddHeader()
+        {
+            var test = new Header();
+            test.AddSubVal("foo", new List<string> { "a", "b" });
+            var mime = new MailMessage();
+            test.AddHeader(mime);
+            var result = mime.Headers.Get("x-smtpapi");
+            Assert.AreEqual("{\"sub\" : {\"foo\" : [\"a\", \"b\"]}}", result);
+        }
+
+        [Test]
+        public void TestAsJson()
+        {
+            var test = new Header();
+            var result = test.AsJson();
+            Assert.AreEqual("", result);
 
+            test = new Header();
+            test.AddSubVal("foo", new List<string>{"a", "b"});
+            result = test.AsJson();
+            Assert.AreEqual("{\"sub\" : {\"foo\" : [\"a\", \"b\"]}}", result);
         }
     }
 }