test for adding sections pass

1 files changed, 13 insertions, 0 deletions

diff --git a/lib/smtpapi_headers.js b/lib/smtpapi_headers.js
index f15cd57..bc8a4bf 100644
--- a/lib/smtpapi_headers.js
+++ b/lib/smtpapi_headers.js
@@ -6,6 +6,7 @@ function SmtpapiHeaders() {
   this.unique_args = {};
   this.category = [];
   this.filters = {};
+  this.section = {};
 }
 
 SmtpapiHeaders.prototype.addTo = function(to) {
@@ -48,6 +49,18 @@ SmtpapiHeaders.prototype.addCategory = function(val) {
   }
 }
 
+SmtpapiHeaders.prototype.setSection = function(val){
+  if (_.isObject(val)) {
+    this.section = val;
+  }
+}
+
+SmtpapiHeaders.prototype.addSection = function(val){
+  if (_.isObject(val)) {
+    _.extend(this.section, val);
+  }
+}
+
 SmtpapiHeaders.prototype.setFilterSetting = function(filters) {
   this.filters = filters;
 }