Add tests for conrefs loader

2 files changed, 36 insertions, 2 deletions

diff --git a/lib/templating/__tests__/conrefsLoader.js b/lib/templating/__tests__/conrefsLoader.js
new file mode 100644
index 0000000..3480a48
--- /dev/null
+++ b/lib/templating/__tests__/conrefsLoader.js
@@ -0,0 +1,34 @@
+var TemplateEngine = require('../../models/templateEngine');
+var renderTemplate = require('../render');
+
+describe('ConrefsLoader', function() {
+    var ConrefsLoader = require('../conrefsLoader');
+
+    var engine = TemplateEngine({
+        loader: new ConrefsLoader(__dirname)
+    });
+
+    describe('Git', function() {
+        pit('should include content from git', function() {
+            return renderTemplate(engine, 'test.md', '{% include "git+https://gist.github.com/69ea4542e4c8967d2fa7.git/test.md" %}')
+            .then(function(str) {
+                expect(str).toBe('Hello from git');
+            });
+        });
+
+        pit('should handle deep inclusion (1)', function() {
+            return renderTemplate(engine, 'test.md', '{% include "git+https://gist.github.com/69ea4542e4c8967d2fa7.git/test2.md" %}')
+            .then(function(str) {
+                expect(str).toBe('First Hello. Hello from git');
+            });
+        });
+
+        pit('should handle deep inclusion (2)', function() {
+            return renderTemplate(engine, 'test.md', '{% include "git+https://gist.github.com/69ea4542e4c8967d2fa7.git/test3.md" %}')
+            .then(function(str) {
+                expect(str).toBe('First Hello. Hello from git');
+            });
+        });
+    });
+});
+
diff --git a/lib/templating/conrefsLoader.js b/lib/templating/conrefsLoader.js
index e46b4d0..c3e5048 100644
--- a/lib/templating/conrefsLoader.js
+++ b/lib/templating/conrefsLoader.js
@@ -48,9 +48,9 @@ var ConrefsLoader = nunjucks.Loader.extend({
 
     resolve: function(from, to) {
         // If origin is in the book, we enforce result file to be in the book
-        if (PathUtils.isInRoot(this.root, from)) {
+        if (PathUtils.isInRoot(this.rootFolder, from)) {
             var href = LocationUtils.toAbsolute(to, path.dirname(from), '');
-            return PathUtils.resolveInRoot(this.root, href);
+            return PathUtils.resolveInRoot(this.rootFolder, href);
         }
 
         // If origin is in a git repository, we resolve file in the git repository