From b4fd21150c8a52fc47a33698ce8096b45bdf760d Mon Sep 17 00:00:00 2001
From: Evgeny Poberezkin <e.poberezkin@me.com>
Date: Sat, 7 Dec 2019 12:35:44 +0000
Subject: [PATCH] test: doT.process

---
 test/process.test.js | 74 ++++++++++++++++++++++++++++++--------------
 1 file changed, 51 insertions(+), 23 deletions(-)

diff --git a/test/process.test.js b/test/process.test.js
index c2ace5c..e8acd1d 100644
--- a/test/process.test.js
+++ b/test/process.test.js
@@ -2,30 +2,58 @@
 
 var assert = require('assert');
 var doT = require('..');
+var fs = require('fs');
 
 
 describe('doT.process', function() {
-	describe('polluting object prototype should not affect template compilation', function() {
-		it('should ignore varname on object prototype', function() {
-      var currentLog = console.log;
-      console.log = log;
-      var logged;
-      
-			Object.prototype.templateSettings = {varname: 'it=(console.log("executed"),{})'};
-
-      try {
-        const templates = doT.process({path: './test'});
-        assert.notEqual(logged, 'executed');
-        // injected code can only be executed if undefined is passed to template function
-        templates.test();
-        assert.notEqual(logged, 'executed');
-      } finally {
-        console.log = currentLog;
-      }
-      
-      function log(str) {
-        logged = str;
-      }
-		})
-	});
+  beforeEach(function() {
+    removeCompiledTemplateFiles();
+  });
+
+  afterEach(function() {
+    removeCompiledTemplateFiles();
+  });
+
+  function removeCompiledTemplateFiles() {
+    try { fs.unlinkSync('./test/templates/test.js'); } catch(e) {}
+  }
+
+  it('should compile all templates in folder', function() {
+    const templates = doT.process({path: './test/templates'});
+    var str = templates.test({data: 2});
+    assert.equal(str, '21');
+
+    var js = fs.statSync('./test/templates/test.js');
+    assert.ok(js.isFile());
+
+    // code below passes if the test is run without coverage using `npm run test-spec`
+    // because source code of doT.encodeHTMLSource is used inside compiled templates
+
+    // var fn = require('./templates/test.js');
+    // var str = fn({data: 2});
+    // assert.equal(str, '21');
+  });
+
+  
+	it('should ignore varname with polluted object prototype', function() {
+    var currentLog = console.log;
+    console.log = log;
+    var logged;
+    
+    Object.prototype.templateSettings = {varname: 'it=(console.log("executed"),{})'};
+
+    try {
+      const templates = doT.process({path: './test/templates'});
+      assert.notEqual(logged, 'executed');
+      // injected code can only be executed if undefined is passed to template function
+      templates.test();
+      assert.notEqual(logged, 'executed');
+    } finally {
+      console.log = currentLog;
+    }
+    
+    function log(str) {
+      logged = str;
+    }
+  });
 });