Merge pull request #47 from benbp/template-contents

Get 'contents' key from template objects
RackHD · Aug 1, 2016 · 24a2e1f · 24a2e1f
2 parents cb584c9 + ce9f87b
commit 24a2e1f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/lib/server.js b/lib/server.js
@@ -75,7 +75,7 @@ function TftpServerFactory(
                 template: req.file,
                 remoteAddress: req.stats.remoteAddress,
             });
-            var script = ejs.render(this.templates[req.file].toString(), this.renderContext);
+            var script = ejs.render(this.templates[req.file].contents.toString(), this.renderContext);
             res.setSize(script.length);
             res.end(script);
         } else {

diff --git a/spec/lib/server-spec.js b/spec/lib/server-spec.js
@@ -104,7 +104,10 @@ describe('tftp server tests', function () {
                 end: sinon.stub()
             };
             server.templates = {
-                test: '<%=switchProfileUri%> <%=switchProfileErrorUri%>'
+                test: {
+                    contents: '<%=switchProfileUri%> <%=switchProfileErrorUri%>',
+                    path: 'testpath'
+                }
             };
 
             server.requestWrapper(req, res);