Make writer customizable

power-assert-js · Jun 9, 2014 · 4e4632c · 4e4632c
1 parent 3d19b58
commit 4e4632c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/index.js b/index.js
@@ -36,22 +36,22 @@ function create (options) {
     if (typeof config.compare !== 'function') {
         config.compare = defaultComparator(config);
     }
-    if (!config.writer) {
-        config.writer = new StringWriter(config.lineSeparator);
+    if (!config.writerClass) {
+        config.writerClass = StringWriter;
     }
     return function (context) {
-        var that = this,
-            events = [],
+        var events = [],
             pairs = [],
-            renderer = new PowerAssertContextRenderer(config);
+            writer = new config.writerClass(extend({}, config)),
+            renderer = new PowerAssertContextRenderer(extend({}, config));
         renderer.init(context);
         traverseContext(context, events, pairs);
-        renderer.render(events, config.writer);
+        renderer.render(events, writer);
         pairs.forEach(function (pair) {
-            config.compare(pair, config.writer);
+            config.compare(pair, writer);
         });
-        config.writer.write('');
-        return config.writer.flush();
+        writer.write('');
+        return writer.flush();
     };
 }
 

diff --git a/lib/string-writer.js b/lib/string-writer.js
@@ -1,6 +1,6 @@
-function StringWriter (lineSeparator) {
+function StringWriter (config) {
     this.lines = [];
-    this.lineSeparator = lineSeparator;
+    this.lineSeparator = config.lineSeparator;
 }
 
 StringWriter.prototype.write = function (str) {