Improve tests for plugins with filters

1 files changed, 7 insertions, 5 deletions

diff --git a/lib/template.js b/lib/template.js
index 69a1bd5..4b3fdbe 100644
--- a/lib/template.js
+++ b/lib/template.js
@@ -69,14 +69,16 @@ var TemplateEngine = function(book) {
 
 
 // Add filter
-TemplateEngine.prototype.addFilter = function(key, filter) {
-    if (this.env.getFilter(filterName)) {
+TemplateEngine.prototype.addFilter = function(filterName, func) {
+    try {
+        this.env.getFilter(filterName);
         this.log.warn.ln("conflict in filters, '"+filterName+"' is already set");
         return false;
+    } catch(e) {
+        this.log.debug.ln("add filter '"+filterName+"'");
+        this.env.addFilter(filterName, func, true);
+        return true;
     }
-    this.log.debug.ln("add filter '"+key+"'");
-    this.env.addFilter(key, func, true);
-    return true;
 };
 
 // Render a string from the book