From c045c803d60dc569baab8efd53b6722faebd3e81 Mon Sep 17 00:00:00 2001
From: Daniel Imms <daimms@microsoft.com>
Date: Wed, 26 Dec 2018 14:48:35 -0800
Subject: [PATCH] Add tests for dispose

---
 src/EscapeSequenceParser.test.ts | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/src/EscapeSequenceParser.test.ts b/src/EscapeSequenceParser.test.ts
index 3b01cb8a..9216dd4c 100644
--- a/src/EscapeSequenceParser.test.ts
+++ b/src/EscapeSequenceParser.test.ts
@@ -1216,6 +1216,15 @@ describe('EscapeSequenceParser', function (): void {
         parser2.parse('\x1b[0m');
         chai.expect(order).eql([3, 2, 1]);
       });
+      it('Dispose', () => {
+        const csiCustom: [string, number[], string][] = [];
+        parser2.setCsiHandler('m', (params, collect) => csi.push(['m', params, collect]));
+        const customHandler = parser2.addCsiHandler('m', (params, collect) => { csiCustom.push(['m', params, collect]); return true; });
+        customHandler.dispose();
+        parser2.parse(INPUT);
+        chai.expect(csi).eql([['m', [1, 31], ''], ['m', [0], '']]);
+        chai.expect(csiCustom).eql([], 'Should not use custom handler as it was disposed');
+      });
     });
     it('EXECUTE handler', function (): void {
       parser2.setExecuteHandler('\n', function (): void {
@@ -1291,6 +1300,15 @@ describe('EscapeSequenceParser', function (): void {
         parser2.parse('\x1b]1;foo=bar\x1b\\');
         chai.expect(order).eql([3, 2, 1]);
       });
+      it('Dispose', () => {
+        const oscCustom: [number, string][] = [];
+        parser2.setOscHandler(1, data => osc.push([1, data]));
+        const customHandler = parser2.addOscHandler(1, data => { oscCustom.push([1, data]); return true; });
+        customHandler.dispose();
+        parser2.parse(INPUT);
+        chai.expect(osc).eql([[1, 'foo=bar']]);
+        chai.expect(oscCustom).eql([], 'Should not use custom handler as it was disposed');
+      });
     });
     it('DCS handler', function (): void {
       parser2.setDcsHandler('+p', {