Add more StreamHandlerSpec unit tests

hershal · Jun 30, 2017 · 1757e3d · 1757e3d
1 parent 7fef6fa
commit 1757e3d
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/test/StreamHandlerSpec.ts b/test/StreamHandlerSpec.ts
@@ -12,6 +12,9 @@ class BasicStreamHandlerDelegate implements IStreamHandlerDelegate {
 
     constructor() {
         this.buffer = [];
+
+        /* Default */
+        this.streamSerializationString = '\n';
     }
 
     /* Delegate methods */
@@ -41,7 +44,6 @@ describe('Stream Handler', function () {
         });
 
         it('Serializes a simple hello world', function (done) {
-            handlerDelegate.streamSerializationString = '\n';
             handler.once(stdin).then(() => {
                 expect(handlerDelegate.buffer).to.deep.equal(['hello world']);
                 done();
@@ -59,5 +61,25 @@ describe('Stream Handler', function () {
             stdin.write('hello world');
             stdin.end();
         });
+
+        it('Serializes a block of input', function (done) {
+            handler.once(stdin).then(() => {
+                expect(handlerDelegate.buffer).to.deep.equal(['hello world',
+                                                              'mellow world']);
+                done();
+            });
+            stdin.write('hello world\nmellow world');
+            stdin.end();
+        });
+
+        it('Serializes a block of input with a space chunk string', function (done) {
+            handlerDelegate.streamSerializationString = ' ';
+            handler.once(stdin).then(() => {
+                expect(handlerDelegate.buffer).to.deep.equal(['hello', 'world\nmellow', 'world']);
+                done();
+            });
+            stdin.write('hello world\nmellow world');
+            stdin.end();
+        });
     });
 });