fix(doc-gen): correctly transform index files

Closes #3021

1 files changed, 32 insertions, 2 deletions

diff --git a/docs/spec/writerSpec.js b/docs/spec/writerSpec.js
index c0f80de3..1c25078c 100644
--- a/docs/spec/writerSpec.js
+++ b/docs/spec/writerSpec.js
@@ -1,7 +1,26 @@
-var writer = require('../src/writer.js');
+var writer,
+    rewire = require('rewire');
+
+function mockResolvedPromise(resolvedValue) {
+  return {
+    then: function(success, failure) {
+      success(resolvedValue);
+    }
+  };
+}
+
 describe('writer', function() {
+
+  beforeEach(function() {
+     writer = rewire('../src/writer.js');
+  });
+
   describe('toString', function() {
-    var toString = writer.toString;
+    var toString;
+
+    beforeEach(function() {
+      toString = writer.toString;
+    });
 
     it('should merge string', function() {
       expect(toString('abc')).toEqual('abc');
@@ -31,4 +50,15 @@ describe('writer', function() {
       expect(content).toBe('ng super jqlite manifest');
     });
   });
+
+  describe('copy', function() {
+    it('should call the transformation function', function() {
+      var readMock = jasmine.createSpy('readMock').andReturn(mockResolvedPromise('DUMMY CONTENT'));
+      writer.__set__("qfs.read", readMock);
+      var transformationFn = jasmine.createSpy('transformationFn');
+      writer.copy('from', 'to', transformationFn, 'arg1', 'arg2');
+      expect(readMock).toHaveBeenCalled();
+      expect(transformationFn).toHaveBeenCalledWith('DUMMY CONTENT', 'arg1', 'arg2');
+    });
+  });
 });