Merge pull request #29 from partkyle/category-array

category should be an array
sendgrid · Mar 1, 2012 · 2ec8484 · 2ec8484
2 parents 3825387 + 6a00a6e
commit 2ec8484
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lib/smtpapi_headers.js b/lib/smtpapi_headers.js
@@ -38,7 +38,11 @@ SmtpapiHeaders.prototype.addUniqueArgs = function(val) {
 }
 
 SmtpapiHeaders.prototype.setCategory = function(val) {
-  this.category = val;
+  if (_.isArray(val)) {
+    this.category = val;
+  } else {
+    this.category = [val];
+  }
 }
 
 SmtpapiHeaders.prototype.addCategory = function(val) {

diff --git a/test/lib/smtpapi_header.test.js b/test/lib/smtpapi_header.test.js
@@ -46,7 +46,7 @@ describe('SmtpapiHeader', function() {
 
     it('should allow setting a single category', function() {
       header.setCategory(categories[0]);
-      expect(header.category).to.equal(categories[0]);
+      expect(header.category).to.eql([categories[0]]);
     });
 
     it('should allow adding categories one at a time', function() {