Merge pull request #514 from TunedMidja/fix_394

Fixes #394

packages/helpers/attachment.txt

@@ -0,0 +1 @@
+Just a little file for testing attachments.

packages/helpers/classes/attachment.js

@@ -53,13 +53,16 @@ class Attachment {
    * Set content
    */
   setContent(content) {
-    if (typeof content === 'undefined') {
-      return;
-    }
-    if (typeof content !== 'string') {
-      throw new Error('String expected for `content`');
+    //Duck type check toString on content if it's a Buffer as that's the method that will be called.
+    if (typeof content === 'string') {      
+      this.content = content;
+    } else if (content instanceof Buffer && content.toString !== undefined) {      
+      this.content = content.toString();
+    } else {
+      throw new Error('`content` expected to be either Buffer or string');
     }
-    this.content = content;
+
+    return;    
   }
 
   /**

packages/helpers/classes/attachment.spec.js

@@ -0,0 +1,43 @@
+'use strict';
+
+const fs = require('fs');
+
+/**
+ * Dependencies
+ */
+const Attachment = require('./attachment');
+
+/**
+ * Tests
+ */
+describe('Attachment', function() {
+  let attachment;
+  beforeEach(function() {
+    attachment = new Attachment();
+  });
+
+  //Set content as string
+  describe('setContent(), string', function() {
+    it('should set string as content', function() {      
+      attachment.setContent("Just a string.");
+      expect(attachment.content).to.equal('Just a string.');
+    });
+  });
+
+  //Set content as stream
+  describe('setContent(), stream', function() {
+    it('should convert stream to string and set as content', function() {      
+      const fileData = fs.readFileSync('./packages/helpers/attachment.txt');
+      attachment.setContent(fileData);
+      expect(attachment.content).to.equal('Just a little file for testing attachments.');
+    });
+  });
+
+  //Set content as wrong type
+  describe('setContent(), wrong type', function() {
+    it('should not allow setting content of wrong type', function() {      
+      expect(() => attachment.setContent(null)).to.throw('`content` expected to be either Buffer or string');
+    });
+  });
+
+});