feat: add ability to specify message stream

fielda · Jun 4, 2021 · 7461aa0 · 7461aa0
1 parent 3c7de2c
commit 7461aa0
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/lib/message.js b/lib/message.js
@@ -113,6 +113,12 @@ Message.prototype.setMetadata = function (meta) {
   this.Metadata = meta;
 };
 
+Message.prototype.setMessageStream = function (messageStream) {
+  if (!messageStream) { return; }
+
+  this.MessageStream = messageStream;
+};
+
 Message.prototype.setTrackOpens = function (trackOpens) {
   if (typeof trackOpens === 'undefined') { return; }
 

diff --git a/lib/postmark-transport.js b/lib/postmark-transport.js
@@ -57,6 +57,7 @@ function parser(mail, next) {
     message.setAttachments(attachments);
     message.setTag(data.tag);
     message.setMetadata(data.metadata);
+    message.setMessageStream(data.messageStream);
     message.setTrackOpens(data.trackOpens);
 
     try {

diff --git a/test/postmark-transport.test.js b/test/postmark-transport.test.js
@@ -364,6 +364,18 @@ describe('PostmarkTransport', () => {
       });
     });
 
+    describe('messageStream', () => {
+      it('should be parsed', (done) => {
+        mail.data.messageStream = 'outbound';
+
+        transport._parse(mails, (err, messages) => {
+          expect(err).to.not.exist();
+          expect(messages[0].MessageStream).equal('outbound');
+          done();
+        });
+      });
+    });
+
     describe('trackOpens', () => {
       it('should be ignored', (done) => {
         transport._parse(mails, (err, messages) => {