chore: unit tests

Author: aryamohanan

packages/core/test/util/excludedFromInstrumentation_test.js

@@ -66,6 +66,165 @@ describe('util.excludedFromInstrumentation', () => {
     expect(isExcluded('/usr/local/npmx.js')).to.be.false;
   });
 
+  describe('pino thread-stream worker detection', () => {
+    let originalRequire;
+
+    beforeEach(() => {
+      originalRequire = require.cache[require.resolve('worker_threads')];
+    });
+
+    afterEach(() => {
+      if (originalRequire) {
+        require.cache[require.resolve('worker_threads')] = originalRequire;
+      } else {
+        delete require.cache[require.resolve('worker_threads')];
+      }
+      delete require.cache[require.resolve('../../src/util/excludedFromInstrumentation')];
+    });
+
+    it('should exclude Pino thread-stream worker with valid structure', () => {
+      mockWorkerThreads({
+        isMainThread: false,
+        workerData: {
+          filename: '/path/to/worker.js',
+          workerData: {
+            $context: {
+              threadStreamVersion: '2.0.0'
+            },
+            target: 'pino-pretty'
+          }
+        }
+      });
+
+      const checkExclusion = require('../../src/util/excludedFromInstrumentation');
+      expect(checkExclusion()).to.be.true;
+    });
+
+    it('should exclude Pino thread-stream worker with targets property', () => {
+      mockWorkerThreads({
+        isMainThread: false,
+        workerData: {
+          filename: '/path/to/worker.js',
+          workerData: {
+            $context: {
+              threadStreamVersion: '2.1.0'
+            },
+            targets: [{ target: 'pino-pretty' }]
+          }
+        }
+      });
+
+      const checkExclusion = require('../../src/util/excludedFromInstrumentation');
+      expect(checkExclusion()).to.be.true;
+    });
+
+    it('should not exclude main thread', () => {
+      mockWorkerThreads({
+        isMainThread: true,
+        workerData: {
+          filename: '/path/to/worker.js',
+          workerData: {
+            $context: {
+              threadStreamVersion: '2.0.0'
+            }
+          }
+        }
+      });
+
+      const checkExclusion = require('../../src/util/excludedFromInstrumentation');
+      expect(checkExclusion()).to.be.false;
+    });
+
+    it('should not exclude worker without workerData', () => {
+      mockWorkerThreads({
+        isMainThread: false,
+        workerData: null
+      });
+
+      const checkExclusion = require('../../src/util/excludedFromInstrumentation');
+      expect(checkExclusion()).to.be.false;
+    });
+
+    it('should not exclude worker without filename', () => {
+      mockWorkerThreads({
+        isMainThread: false,
+        workerData: {
+          workerData: {
+            $context: {
+              threadStreamVersion: '2.0.0'
+            }
+          }
+        }
+      });
+
+      const checkExclusion = require('../../src/util/excludedFromInstrumentation');
+      expect(checkExclusion()).to.be.false;
+    });
+
+    it('should not exclude worker without nested workerData', () => {
+      mockWorkerThreads({
+        isMainThread: false,
+        workerData: {
+          filename: '/path/to/worker.js'
+        }
+      });
+
+      const checkExclusion = require('../../src/util/excludedFromInstrumentation');
+      expect(checkExclusion()).to.be.false;
+    });
+
+    it('should not exclude worker without $context', () => {
+      mockWorkerThreads({
+        isMainThread: false,
+        workerData: {
+          filename: '/path/to/worker.js',
+          workerData: {
+            someOtherProperty: 'value'
+          }
+        }
+      });
+
+      const checkExclusion = require('../../src/util/excludedFromInstrumentation');
+      expect(checkExclusion()).to.be.false;
+    });
+
+    it('should not exclude worker without threadStreamVersion', () => {
+      mockWorkerThreads({
+        isMainThread: false,
+        workerData: {
+          filename: '/path/to/worker.js',
+          workerData: {
+            $context: {
+              someOtherProperty: 'value'
+            }
+          }
+        }
+      });
+
+      const checkExclusion = require('../../src/util/excludedFromInstrumentation');
+      expect(checkExclusion()).to.be.false;
+    });
+
+    it('should not exclude non-Pino worker thread', () => {
+      mockWorkerThreads({
+        isMainThread: false,
+        workerData: {
+          customData: 'some other worker'
+        }
+      });
+
+      const checkExclusion = require('../../src/util/excludedFromInstrumentation');
+      expect(checkExclusion()).to.be.false;
+    });
+
+    function mockWorkerThreads(mockData) {
+      const mockModule = {
+        exports: mockData
+      };
+      require.cache[require.resolve('worker_threads')] = mockModule;
+    }
+  });
+
   function isExcluded(argv1) {
     process.argv[1] = argv1;
     return isExcludedFromInstrumentation();