Add failing tests for #2971

open-telemetry · Jul 28, 2022 · ff2d24f · ff2d24f
1 parent 3db1056
commit ff2d24f
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/packages/opentelemetry-context-async-hooks/test/AsyncHooksContextManager.test.ts b/packages/opentelemetry-context-async-hooks/test/AsyncHooksContextManager.test.ts
@@ -415,6 +415,28 @@ for (const contextManagerClass of [
         patchedEE.emit('test');
       });
 
+      it('should remove event handler enabled by .once using removeListener (when enabled)', () => {
+        const ee = new EventEmitter();
+        const context = ROOT_CONTEXT.setValue(key1, 1);
+        const patchedEE = contextManager.bind(context, ee);
+        function handler() {}
+        patchedEE.once('test', handler);
+        assert.strictEqual(patchedEE.listeners('test').length, 1);
+        patchedEE.removeListener('test', handler);
+        assert.strictEqual(patchedEE.listeners('test').length, 0);
+      });
+
+      it('should remove event handler enabled by .once using off (when enabled)', () => {
+        const ee = new EventEmitter();
+        const context = ROOT_CONTEXT.setValue(key1, 1);
+        const patchedEE = contextManager.bind(context, ee);
+        const handler = () => { };
+        patchedEE.once('test', handler);
+        assert.strictEqual(patchedEE.listeners('test').length, 1);
+        patchedEE.off('test', handler);
+        assert.strictEqual(patchedEE.listeners('test').length, 0);
+      });
+
       it('should return current context and removeAllListeners (when enabled)', done => {
         const ee = new EventEmitter();
         const context = ROOT_CONTEXT.setValue(key1, 1);