Test mammoth.images.imgElement directly

1 files changed, 54 insertions, 1 deletions

diff --git a/test/images.tests.js b/test/images.tests.js
index a66cdee..9a9344d 100644
--- a/test/images.tests.js
+++ b/test/images.tests.js
@@ -3,6 +3,7 @@ var assert = require("assert");
 var hamjest = require("hamjest");
 var assertThat = hamjest.assertThat;
 var contains = hamjest.contains;
+var equalTo = hamjest.equalTo;
 var hasProperties = hamjest.hasProperties;
 
 var mammoth = require("../");
@@ -25,10 +26,62 @@ test('mammoth.images.dataUri() encodes images in base64', function() {
         },
         contentType: "image/jpeg"
     });
-    
+
     return mammoth.images.dataUri(image).then(function(result) {
         assertThat(result, contains(
             hasProperties({tag: hasProperties({attributes: {"src": "data:image/jpeg;base64,YWJj"}})})
         ));
     });
 });
+
+
+test('mammoth.images.imgElement()', {
+    'when element does not have alt text then alt attribute is not set': function() {
+        var imageBuffer = new Buffer("abc");
+        var image = new documents.Image({
+            readImage: function(encoding) {
+                return promises.when(imageBuffer.toString(encoding));
+            },
+            contentType: "image/jpeg"
+        });
+
+        var result = mammoth.images.imgElement(function(image) {
+            return {src: "<src>"};
+        })(image);
+
+        return result.then(function(result) {
+            assertThat(result, contains(
+                hasProperties({
+                    tag: hasProperties({
+                        attributes: equalTo({src: "<src>"})
+                    })
+                })
+            ));
+        });
+    },
+
+    'when element has alt text then alt attribute is set': function() {
+        var imageBuffer = new Buffer("abc");
+        var image = new documents.Image({
+            readImage: function(encoding) {
+                return promises.when(imageBuffer.toString(encoding));
+            },
+            contentType: "image/jpeg",
+            altText: "<alt>"
+        });
+
+        var result = mammoth.images.imgElement(function(image) {
+            return {src: "<src>"};
+        })(image);
+
+        return result.then(function(result) {
+            assertThat(result, contains(
+                hasProperties({
+                    tag: hasProperties({
+                        attributes: equalTo({alt: "<alt>", src: "<src>"})
+                    })
+                })
+            ));
+        });
+    }
+});