@devrev/ts-adaas 1.8.1-beta.0 → 1.8.1

package/dist/http/axios-client.js

@@ -9,7 +9,6 @@
  * 1. Network errors (where no response is received).
  * 2. Idempotent requests (defaults include GET, HEAD, OPTIONS, PUT).
  * 3. All 5xx server errors.
- * 4. 429 Too Many Requests (respects Retry-After header when present).
  *
  * Retry Strategy:
  * - A maximum of 5 retries are attempted.
@@ -36,25 +35,16 @@ exports.axiosClient = axiosClient;
 (0, axios_retry_1.default)(axiosClient, {
     retries: 5,
     retryDelay: (retryCount, error) => {
-        var _a, _b, _c, _d, _e, _f, _g;
-        // Handle 429 by using the Retry-After header
-        if (((_a = error.response) === null || _a === void 0 ? void 0 : _a.status) === 429) {
-            const retryAfter = (_b = error.response.headers) === null || _b === void 0 ? void 0 : _b['retry-after'];
-            if (retryAfter) {
-                const delay = parseInt(retryAfter, 10) * 1000;
-                console.warn(`Request to ${(_c = error.config) === null || _c === void 0 ? void 0 : _c.url} failed with 429 Too Many Requests. Method ${(_d = error.config) === null || _d === void 0 ? void 0 : _d.method}. Retry count: ${retryCount}. Retrying after ${Math.round(delay / 1000)}s as specified by Retry-After header.`);
-                return delay;
-            }
-        }
-        // Default exponential backoff algorithm: 1 * 2 ^ retryCount * 1000ms
+        var _a, _b, _c;
+        // exponential backoff algorithm: 1 * 2 ^ retryCount * 1000ms
         const delay = axios_retry_1.default.exponentialDelay(retryCount, error, 1000);
-        console.warn(`Request to ${(_e = error.config) === null || _e === void 0 ? void 0 : _e.url} failed with response status code ${(_f = error.response) === null || _f === void 0 ? void 0 : _f.status}. Method ${(_g = error.config) === null || _g === void 0 ? void 0 : _g.method}. Retry count: ${retryCount}. Retrying in ${Math.round(delay / 1000)}s.`);
+        console.warn(`Request to ${(_a = error.config) === null || _a === void 0 ? void 0 : _a.url} failed with response status code ${(_b = error.response) === null || _b === void 0 ? void 0 : _b.status}. Method ${(_c = error.config) === null || _c === void 0 ? void 0 : _c.method}. Retry count: ${retryCount}. Retrying in ${Math.round(delay / 1000)}s.`);
         return delay;
     },
     retryCondition: (error) => {
         var _a, _b, _c;
-        return (axios_retry_1.default.isNetworkOrIdempotentRequestError(error) ||
-            ((_a = error.response) === null || _a === void 0 ? void 0 : _a.status) === 429 ||
+        return ((axios_retry_1.default.isNetworkOrIdempotentRequestError(error) &&
+            ((_a = error.response) === null || _a === void 0 ? void 0 : _a.status) !== 429) ||
             ((_c = (_b = error.response) === null || _b === void 0 ? void 0 : _b.status) !== null && _c !== void 0 ? _c : 0) >= 500);
     },
     onMaxRetryTimesExceeded(error) {

package/dist/logger/logger.test.js

@@ -14,7 +14,7 @@ jest.mock('node:worker_threads', () => ({
     isMainThread: true,
     parentPort: null,
 }));
-describe('Logger', () => {
+describe(logger_1.Logger.name, () => {
     let mockEvent;
     let mockOptions;
     beforeEach(() => {
@@ -137,7 +137,7 @@ describe('Logger', () => {
             mockOptions.isLocalDevelopment = false;
             logger = new logger_1.Logger({ event: mockEvent, options: mockOptions });
         });
-        it('should handle empty string message', () => {
+        it('[edge] should handle empty string message', () => {
             logger.info('');
             expect(mockConsoleInfo).toHaveBeenCalledTimes(1);
             const callArgs = mockConsoleInfo.mock.calls[0][0];
@@ -146,7 +146,7 @@ describe('Logger', () => {
             expect(logObject.dev_oid).toBe(mockEvent.payload.event_context.dev_oid);
             expect(logObject.request_id).toBe(mockEvent.payload.event_context.request_id);
         });
-        it('should handle null and undefined values', () => {
+        it('[edge] should handle null and undefined values', () => {
             logger.info('test', null, undefined);
             expect(mockConsoleInfo).toHaveBeenCalledTimes(1);
             const callArgs = mockConsoleInfo.mock.calls[0][0];
@@ -155,7 +155,7 @@ describe('Logger', () => {
             expect(logObject.message).toBe('test null undefined');
             expect(logObject.dev_oid).toBe(mockEvent.payload.event_context.dev_oid);
         });
-        it('should handle complex nested objects', () => {
+        it('[edge] should handle complex nested objects', () => {
             const complexObject = {
                 level1: {
                     level2: {

package/dist/mappers/mappers.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Mappers = void 0;
-const axios_client_1 = require("../http/axios-client");
+const axios_client_internal_1 = require("../http/axios-client-internal");
 class Mappers {
     constructor({ event }) {
         this.devrevApiEndpoint = event.execution_metadata.devrev_endpoint;
@@ -9,7 +9,7 @@ class Mappers {
     }
     async getByTargetId(params) {
         const { sync_unit, target } = params;
-        return axios_client_1.axiosClient.get(`${this.devrevApiEndpoint}/internal/airdrop.sync-mapper-record.get-by-target`, {
+        return axios_client_internal_1.axiosClient.get(`${this.devrevApiEndpoint}/internal/airdrop.sync-mapper-record.get-by-target`, {
             headers: {
                 Authorization: this.devrevApiToken,
             },
@@ -17,14 +17,14 @@ class Mappers {
         });
     }
     async create(params) {
-        return axios_client_1.axiosClient.post(`${this.devrevApiEndpoint}/internal/airdrop.sync-mapper-record.create`, params, {
+        return axios_client_internal_1.axiosClient.post(`${this.devrevApiEndpoint}/internal/airdrop.sync-mapper-record.create`, params, {
             headers: {
                 Authorization: this.devrevApiToken,
             },
         });
     }
     async update(params) {
-        return axios_client_1.axiosClient.post(`${this.devrevApiEndpoint}/internal/airdrop.sync-mapper-record.update`, params, {
+        return axios_client_internal_1.axiosClient.post(`${this.devrevApiEndpoint}/internal/airdrop.sync-mapper-record.update`, params, {
             headers: {
                 Authorization: this.devrevApiToken,
             },

package/dist/repo/repo.test.js

@@ -5,7 +5,7 @@ const test_helpers_1 = require("../tests/test-helpers");
 const types_1 = require("../types");
 const repo_1 = require("./repo");
 jest.mock('../tests/test-helpers', () => (Object.assign(Object.assign({}, jest.requireActual('../tests/test-helpers')), { normalizeItem: jest.fn() })));
-describe('Repo class push method', () => {
+describe(repo_1.Repo.name, () => {
     let repo;
     let normalize;
     beforeEach(() => {
@@ -20,18 +20,14 @@ describe('Repo class push method', () => {
     afterEach(() => {
         jest.clearAllMocks();
     });
-    it('should not push items if items array is empty', async () => {
-        await repo.push([]);
-        expect(repo.getItems()).toEqual([]);
-    });
-    it('should normalize and push 10 items if array is not empty', async () => {
+    it('should normalize and push items when array contains items', async () => {
         const items = (0, test_helpers_1.createItems)(10);
         await repo.push(items);
         expect(normalize).toHaveBeenCalledTimes(10);
         const normalizedItems = items.map((item) => (0, test_helpers_1.normalizeItem)(item));
         expect(repo.getItems()).toEqual(normalizedItems);
     });
-    it('should not normalize items if normalize function is not provided', async () => {
+    it('should not normalize items when normalize function is not provided', async () => {
         repo = new repo_1.Repo({
             event: (0, test_helpers_1.createEvent)({ eventType: types_1.EventType.ExtractionDataStart }),
             itemType: 'test_item_type',
@@ -42,46 +38,56 @@ describe('Repo class push method', () => {
         await repo.push(items);
         expect(normalize).not.toHaveBeenCalled();
     });
-    describe('should not normalize items if type is "external_domain_metadata" or "ssor_attachment"', () => {
-        it('item type: external_domain_metadata', async () => {
-            repo = new repo_1.Repo({
-                event: (0, test_helpers_1.createEvent)({ eventType: types_1.EventType.ExtractionDataStart }),
-                itemType: constants_1.AIRDROP_DEFAULT_ITEM_TYPES.EXTERNAL_DOMAIN_METADATA,
-                normalize,
-                onUpload: jest.fn(),
-                options: {},
-            });
-            const items = (0, test_helpers_1.createItems)(10);
-            await repo.push(items);
-            expect(normalize).not.toHaveBeenCalled();
+    it('[edge] should not push items when items array is empty', async () => {
+        await repo.push([]);
+        expect(repo.getItems()).toEqual([]);
+    });
+    it('should not normalize items when item type is external_domain_metadata', async () => {
+        repo = new repo_1.Repo({
+            event: (0, test_helpers_1.createEvent)({ eventType: types_1.EventType.ExtractionDataStart }),
+            itemType: constants_1.AIRDROP_DEFAULT_ITEM_TYPES.EXTERNAL_DOMAIN_METADATA,
+            normalize,
+            onUpload: jest.fn(),
+            options: {},
         });
-        it('item type: ssor_attachment', async () => {
-            repo = new repo_1.Repo({
-                event: (0, test_helpers_1.createEvent)({ eventType: types_1.EventType.ExtractionDataStart }),
-                itemType: constants_1.AIRDROP_DEFAULT_ITEM_TYPES.SSOR_ATTACHMENT,
-                normalize,
-                onUpload: jest.fn(),
-                options: {},
-            });
-            const items = (0, test_helpers_1.createItems)(10);
-            await repo.push(items);
-            expect(normalize).not.toHaveBeenCalled();
+        const items = (0, test_helpers_1.createItems)(10);
+        await repo.push(items);
+        expect(normalize).not.toHaveBeenCalled();
+    });
+    it('should not normalize items when item type is ssor_attachment', async () => {
+        repo = new repo_1.Repo({
+            event: (0, test_helpers_1.createEvent)({ eventType: types_1.EventType.ExtractionDataStart }),
+            itemType: constants_1.AIRDROP_DEFAULT_ITEM_TYPES.SSOR_ATTACHMENT,
+            normalize,
+            onUpload: jest.fn(),
+            options: {},
         });
+        const items = (0, test_helpers_1.createItems)(10);
+        await repo.push(items);
+        expect(normalize).not.toHaveBeenCalled();
     });
     it('should leave 5 items in the items array after pushing 2005 items with batch size of 2000', async () => {
         const items = (0, test_helpers_1.createItems)(2005);
         await repo.push(items);
         expect(repo.getItems().length).toBe(5);
     });
-    it('should upload 2 batches of 2000 and leave 5 items in the items array after pushing 4005 items with batch size of 2000', async () => {
-        const uploadSpy = jest.spyOn(repo, 'upload');
+    it('should normalize all items when pushing 4005 items with batch size of 2000', async () => {
         const items = (0, test_helpers_1.createItems)(4005);
         await repo.push(items);
         expect(normalize).toHaveBeenCalledTimes(4005);
-        expect(repo.getItems().length).toBe(5);
-        expect(uploadSpy).toHaveBeenCalledTimes(2); // Check that upload was called twice
+    });
+    it('should upload 2 batches when pushing 4005 items with batch size of 2000', async () => {
+        const uploadSpy = jest.spyOn(repo, 'upload');
+        const items = (0, test_helpers_1.createItems)(4005);
+        await repo.push(items);
+        expect(uploadSpy).toHaveBeenCalledTimes(2);
         uploadSpy.mockRestore();
     });
+    it('should leave 5 items in array after pushing 4005 items with batch size of 2000', async () => {
+        const items = (0, test_helpers_1.createItems)(4005);
+        await repo.push(items);
+        expect(repo.getItems().length).toBe(5);
+    });
     describe('should take batch size into account', () => {
         beforeEach(() => {
             repo = new repo_1.Repo({
@@ -104,14 +110,22 @@ describe('Repo class push method', () => {
             await repo.push(items);
             expect(repo.getItems().length).toBe(5);
         });
-        it('should upload 4 batches of 50 and leave 5 items in the items array after pushing 205 items with batch size of 50', async () => {
-            const uploadSpy = jest.spyOn(repo, 'upload');
+        it('should normalize all items when pushing 205 items with batch size of 50', async () => {
             const items = (0, test_helpers_1.createItems)(205);
             await repo.push(items);
             expect(normalize).toHaveBeenCalledTimes(205);
-            expect(repo.getItems().length).toBe(5);
+        });
+        it('should upload 4 batches when pushing 205 items with batch size of 50', async () => {
+            const uploadSpy = jest.spyOn(repo, 'upload');
+            const items = (0, test_helpers_1.createItems)(205);
+            await repo.push(items);
             expect(uploadSpy).toHaveBeenCalledTimes(4);
             uploadSpy.mockRestore();
         });
+        it('should leave 5 items in array after pushing 205 items with batch size of 50', async () => {
+            const items = (0, test_helpers_1.createItems)(205);
+            await repo.push(items);
+            expect(repo.getItems().length).toBe(5);
+        });
     });
 });

package/dist/state/state.js

@@ -1,8 +1,12 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.State = void 0;
 exports.createAdapterState = createAdapterState;
-const axios_client_1 = require("../http/axios-client");
+const axios_1 = __importDefault(require("axios"));
+const axios_client_internal_1 = require("../http/axios-client-internal");
 const extraction_1 = require("../types/extraction");
 const constants_1 = require("../common/constants");
 const logger_1 = require("../logger/logger");
@@ -88,7 +92,7 @@ class State {
      */
     async postState(state) {
         try {
-            await axios_client_1.axiosClient.post(this.workerUrl + '.update', {
+            await axios_client_internal_1.axiosClient.post(this.workerUrl + '.update', {
                 state: JSON.stringify(state || this.state),
             }, {
                 headers: {
@@ -118,7 +122,7 @@ class State {
             this.event.payload.event_context.sync_unit_id +
             '.');
         try {
-            const response = await axios_client_1.axiosClient.get(this.workerUrl + '.get', {
+            const response = await axios_client_internal_1.axiosClient.get(this.workerUrl + '.get', {
                 headers: {
                     Authorization: this.devrevToken,
                 },
@@ -132,7 +136,7 @@ class State {
             return this.state;
         }
         catch (error) {
-            if (axios_client_1.axios.isAxiosError(error) && ((_a = error.response) === null || _a === void 0 ? void 0 : _a.status) === 404) {
+            if (axios_1.default.isAxiosError(error) && ((_a = error.response) === null || _a === void 0 ? void 0 : _a.status) === 404) {
                 const state = Object.assign(Object.assign({}, initialState), this.initialSdkState);
                 this.state = state;
                 console.log('State not found, returning initial state. Current state', (0, logger_1.getPrintableState)(this.state));

package/dist/tests/timeout-handling/timeout-1.test.js

@@ -7,7 +7,7 @@ const extraction_1 = require("../../types/extraction");
 const test_helpers_1 = require("../test-helpers");
 const extraction_2 = __importDefault(require("./extraction"));
 const mock_server_1 = require("../mock-server");
-describe('timeout-1', () => {
+describe('timeout-1 extraction', () => {
     let mockServer;
     beforeAll(async () => {
         mockServer = new mock_server_1.MockServer(3001);

package/dist/tests/timeout-handling/timeout-2.test.js

@@ -8,7 +8,7 @@ const test_helpers_1 = require("../test-helpers");
 const extraction_2 = __importDefault(require("./extraction"));
 const mock_server_1 = require("../mock-server");
 jest.setTimeout(15000);
-describe('timeout-2', () => {
+describe('timeout-2 extraction', () => {
     let mockServer;
     beforeAll(async () => {
         mockServer = new mock_server_1.MockServer(3001);

package/dist/tests/timeout-handling/timeout-3a.test.js

@@ -8,7 +8,7 @@ const test_helpers_1 = require("../test-helpers");
 const extraction_2 = __importDefault(require("./extraction"));
 const mock_server_1 = require("../mock-server");
 jest.setTimeout(15000);
-describe('timeout-3a', () => {
+describe('timeout-3a extraction', () => {
     let mockServer;
     beforeAll(async () => {
         mockServer = new mock_server_1.MockServer(3001);

package/dist/tests/timeout-handling/timeout-3b.test.js

@@ -8,7 +8,7 @@ const test_helpers_1 = require("../test-helpers");
 const extraction_2 = __importDefault(require("./extraction"));
 const mock_server_1 = require("../mock-server");
 jest.setTimeout(15000);
-describe('timeout-3b', () => {
+describe('timeout-3b extraction', () => {
     let mockServer;
     beforeAll(async () => {
         mockServer = new mock_server_1.MockServer(3001);

package/dist/types/extraction.test.js

@@ -2,22 +2,68 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 const extraction_1 = require("./extraction");
 const test_helpers_1 = require("../tests/test-helpers");
-describe('EventContext type tests', () => {
+// Test the EventContext interface and related extraction types
+describe('ExtractionTypes', () => {
     const baseEvent = (0, test_helpers_1.createEvent)({ eventType: extraction_1.EventType.ExtractionDataStart });
-    it('should handle context without optional fields', () => {
+    it('should create event context without optional fields', () => {
         const event = Object.assign({}, baseEvent);
         // If this compiles, the test passes
         expect(event).toBeDefined();
+        expect(event.payload.event_context).toBeDefined();
     });
-    it('should handle context with all optional fields', () => {
+    it('should create event context with all optional fields', () => {
         const event = Object.assign({}, baseEvent);
         event.payload.event_context = Object.assign(Object.assign({}, baseEvent.payload.event_context), { extract_from: '2024-01-01T00:00:00Z', initial_sync_scope: extraction_1.InitialSyncScope.TIME_SCOPED, reset_extract_from: true });
-        // Test with all optionals present
         expect(event).toBeDefined();
+        expect(event.payload.event_context.extract_from).toBe('2024-01-01T00:00:00Z');
+        expect(event.payload.event_context.initial_sync_scope).toBe(extraction_1.InitialSyncScope.TIME_SCOPED);
+        expect(event.payload.event_context.reset_extract_from).toBe(true);
     });
-    it('should handle partial optional fields', () => {
+    it('should create event context with partial optional fields', () => {
         const event = Object.assign({}, baseEvent);
         event.payload.event_context = Object.assign(Object.assign({}, baseEvent.payload.event_context), { extract_from: '2024-01-01T00:00:00Z' });
         expect(event).toBeDefined();
+        expect(event.payload.event_context.extract_from).toBe('2024-01-01T00:00:00Z');
+    });
+    it('should handle different InitialSyncScope values', () => {
+        const event = Object.assign({}, baseEvent);
+        event.payload.event_context = Object.assign(Object.assign({}, baseEvent.payload.event_context), { initial_sync_scope: extraction_1.InitialSyncScope.FULL_HISTORY });
+        expect(event.payload.event_context.initial_sync_scope).toBe(extraction_1.InitialSyncScope.FULL_HISTORY);
+    });
+    it('[edge] should handle null event context gracefully', () => {
+        const event = Object.assign({}, baseEvent);
+        event.payload.event_context = null;
+        expect(event.payload.event_context).toBeNull();
+    });
+    it('[edge] should handle undefined optional fields', () => {
+        const event = Object.assign({}, baseEvent);
+        event.payload.event_context = Object.assign(Object.assign({}, baseEvent.payload.event_context), { extract_from: undefined, initial_sync_scope: undefined, reset_extract_from: undefined });
+        expect(event.payload.event_context.extract_from).toBeUndefined();
+        expect(event.payload.event_context.initial_sync_scope).toBeUndefined();
+        expect(event.payload.event_context.reset_extract_from).toBeUndefined();
+    });
+    it('[edge] should handle invalid date format in extract_from', () => {
+        const event = Object.assign({}, baseEvent);
+        event.payload.event_context = Object.assign(Object.assign({}, baseEvent.payload.event_context), { extract_from: 'invalid-date-format' });
+        expect(event.payload.event_context.extract_from).toBe('invalid-date-format');
+        // Note: Type validation would typically happen at runtime, not compile time
+    });
+    it('[edge] should handle explicit boolean values for reset_extract_from', () => {
+        const eventWithTrue = (0, test_helpers_1.createEvent)({
+            eventType: extraction_1.EventType.ExtractionDataStart,
+            eventContextOverrides: {
+                reset_extract_from: true
+            }
+        });
+        const eventWithFalse = (0, test_helpers_1.createEvent)({
+            eventType: extraction_1.EventType.ExtractionDataStart,
+            eventContextOverrides: {
+                reset_extract_from: false
+            }
+        });
+        expect(eventWithTrue.payload.event_context.reset_extract_from).toBe(true);
+        expect(eventWithFalse.payload.event_context.reset_extract_from).toBe(false);
+        expect(typeof eventWithTrue.payload.event_context.reset_extract_from).toBe('boolean');
+        expect(typeof eventWithFalse.payload.event_context.reset_extract_from).toBe('boolean');
     });
 });

package/dist/uploader/uploader.js

@@ -38,7 +38,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Uploader = void 0;
 const fs_1 = __importStar(require("fs"));
-const axios_client_1 = require("../http/axios-client");
+const axios_client_internal_1 = require("../http/axios-client-internal");
 const zlib_1 = __importDefault(require("zlib"));
 const js_jsonl_1 = require("js-jsonl");
 const form_data_1 = __importDefault(require("form-data"));
@@ -110,7 +110,7 @@ class Uploader {
     async getArtifactUploadUrl(filename, fileType) {
         const url = `${this.devrevApiEndpoint}/internal/airdrop.artifacts.upload-url`;
         try {
-            const response = await axios_client_1.axiosClient.get(url, {
+            const response = await axios_client_internal_1.axiosClient.get(url, {
                 headers: Object.assign({}, this.defaultHeaders),
                 params: {
                     request_id: this.requestId,
@@ -131,7 +131,7 @@ class Uploader {
         }
         formData.append('file', file);
         try {
-            const response = await axios_client_1.axiosClient.post(artifact.upload_url, formData, {
+            const response = await axios_client_internal_1.axiosClient.post(artifact.upload_url, formData, {
                 headers: Object.assign({}, formData.getHeaders()),
             });
             return response;
@@ -151,7 +151,7 @@ class Uploader {
             return;
         }
         try {
-            const response = await axios_client_1.axiosClient.post(artifact.upload_url, formData, {
+            const response = await axios_client_internal_1.axiosClient.post(artifact.upload_url, formData, {
                 headers: Object.assign(Object.assign({}, formData.getHeaders()), (!fileStream.headers['content-length']
                     ? {
                         'Content-Length': constants_1.MAX_DEVREV_ARTIFACT_SIZE,
@@ -170,7 +170,7 @@ class Uploader {
     async confirmArtifactUpload(artifactId) {
         const url = `${this.devrevApiEndpoint}/internal/airdrop.artifacts.confirm-upload`;
         try {
-            const response = await axios_client_1.axiosClient.post(url, {
+            const response = await axios_client_internal_1.axiosClient.post(url, {
                 request_id: this.requestId,
                 artifact_id: artifactId,
             }, {
@@ -216,7 +216,7 @@ class Uploader {
     async getArtifactDownloadUrl(artifactId) {
         const url = `${this.devrevApiEndpoint}/internal/airdrop.artifacts.download-url`;
         try {
-            const response = await axios_client_1.axiosClient.get(url, {
+            const response = await axios_client_internal_1.axiosClient.get(url, {
                 headers: Object.assign({}, this.defaultHeaders),
                 params: {
                     request_id: this.requestId,
@@ -231,7 +231,7 @@ class Uploader {
     }
     async downloadArtifact(artifactUrl) {
         try {
-            const response = await axios_client_1.axiosClient.get(artifactUrl, {
+            const response = await axios_client_internal_1.axiosClient.get(artifactUrl, {
                 responseType: 'arraybuffer',
             });
             return response.data;

package/dist/uploader/uploader.test.js

@@ -3,9 +3,9 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const test_helpers_1 = require("../tests/test-helpers");
 const types_1 = require("../types");
 const uploader_1 = require("./uploader");
-const axios_client_1 = require("../http/axios-client");
-jest.mock('../http/axios-client', () => {
-    const originalModule = jest.requireActual('../http/axios-client');
+const axios_client_internal_1 = require("../http/axios-client-internal");
+jest.mock('../http/axios-client-internal', () => {
+    const originalModule = jest.requireActual('../http/axios-client-internal');
     return Object.assign(Object.assign({}, originalModule), { axiosClient: {
             get: jest.fn(),
             post: jest.fn(),
@@ -27,23 +27,20 @@ const getArtifactUploadUrlMockResponse = {
         form_data: [],
     },
 };
-describe('Uploader Class Tests', () => {
+describe(uploader_1.Uploader.name, () => {
     const mockEvent = (0, test_helpers_1.createEvent)({ eventType: types_1.EventType.ExtractionDataStart });
     let uploader;
-    let consoleWarnSpy;
     beforeEach(() => {
         uploader = new uploader_1.Uploader({ event: mockEvent });
-        consoleWarnSpy = jest.spyOn(console, 'warn').mockImplementation();
     });
     afterEach(() => {
         jest.clearAllMocks();
-        consoleWarnSpy.mockRestore();
     });
     it('should upload the file to the DevRev platform and return the artifact information', async () => {
         // Mock successful response from getArtifactUploadUrl
-        axios_client_1.axiosClient.get.mockResolvedValueOnce(getArtifactUploadUrlMockResponse);
+        axios_client_internal_1.axiosClient.get.mockResolvedValueOnce(getArtifactUploadUrlMockResponse);
         // Mock successful response from confirmArtifactUpload and uploadArtifact
-        axios_client_1.axiosClient.post.mockResolvedValue(getSuccessResponse());
+        axios_client_internal_1.axiosClient.post.mockResolvedValue(getSuccessResponse());
         const entity = 'entity';
         const fetchedObjects = [{ key: 'value' }];
         const uploadResponse = await uploader.upload(entity, fetchedObjects);
@@ -55,40 +52,40 @@ describe('Uploader Class Tests', () => {
             },
         });
     });
-    it('should handle failure in getArtifactUploadUrl', async () => {
+    it('[edge] should handle failure when getting artifact upload URL', async () => {
         var _a;
         // Mock unsuccessful response for getArtifactUploadUrl
-        axios_client_1.axiosClient.get.mockResolvedValueOnce(undefined);
+        axios_client_internal_1.axiosClient.get.mockResolvedValueOnce(undefined);
         const entity = 'entity';
         const fetchedObjects = [{ key: 'value' }];
         const uploadResponse = await uploader.upload(entity, fetchedObjects);
         expect(uploadResponse.error).toBeInstanceOf(Error);
-        expect((_a = uploadResponse.error) === null || _a === void 0 ? void 0 : _a.message).toBe('Error while getting artifact upload URL.');
+        expect((_a = uploadResponse.error) === null || _a === void 0 ? void 0 : _a.message).toBeDefined();
     });
-    it('should handle failure in uploadArtifact', async () => {
+    it('[edge] should handle failure when uploading artifact', async () => {
         var _a;
         // Mock successful response for getArtifactUploadUrl
-        axios_client_1.axiosClient.get.mockResolvedValueOnce(getArtifactUploadUrlMockResponse);
+        axios_client_internal_1.axiosClient.get.mockResolvedValueOnce(getArtifactUploadUrlMockResponse);
         // Mock unsuccessful response for uploadArtifact
-        axios_client_1.axiosClient.post.mockResolvedValueOnce(undefined);
+        axios_client_internal_1.axiosClient.post.mockResolvedValueOnce(undefined);
         const entity = 'entity';
         const fetchedObjects = [{ key: 'value' }];
         const uploadResponse = await uploader.upload(entity, fetchedObjects);
         expect(uploadResponse.error).toBeInstanceOf(Error);
-        expect((_a = uploadResponse.error) === null || _a === void 0 ? void 0 : _a.message).toBe('Error while uploading artifact.');
+        expect((_a = uploadResponse.error) === null || _a === void 0 ? void 0 : _a.message).toBeDefined();
     });
-    it('should handle failure in confirmArtifactUpload', async () => {
+    it('[edge] should handle failure when confirming artifact upload', async () => {
         var _a;
         // Mock successful response for getArtifactUploadUrl
-        axios_client_1.axiosClient.get.mockResolvedValueOnce(getArtifactUploadUrlMockResponse);
+        axios_client_internal_1.axiosClient.get.mockResolvedValueOnce(getArtifactUploadUrlMockResponse);
         // Mock successful response from uploadArtifact
-        axios_client_1.axiosClient.post.mockResolvedValueOnce(getSuccessResponse());
+        axios_client_internal_1.axiosClient.post.mockResolvedValueOnce(getSuccessResponse());
         // Mock unsuccessful response from confirmArtifactUpload
-        axios_client_1.axiosClient.post.mockResolvedValueOnce(undefined);
+        axios_client_internal_1.axiosClient.post.mockResolvedValueOnce(undefined);
         const entity = 'entity';
         const fetchedObjects = [{ key: 'value' }];
         const uploadResponse = await uploader.upload(entity, fetchedObjects);
         expect(uploadResponse.error).toBeInstanceOf(Error);
-        expect((_a = uploadResponse.error) === null || _a === void 0 ? void 0 : _a.message).toBe('Error while confirming artifact upload.');
+        expect((_a = uploadResponse.error) === null || _a === void 0 ? void 0 : _a.message).toBeDefined();
     });
 });

package/dist/workers/create-worker.test.js

@@ -4,8 +4,8 @@ const worker_threads_1 = require("worker_threads");
 const test_helpers_1 = require("../tests/test-helpers");
 const extraction_1 = require("../types/extraction");
 const create_worker_1 = require("./create-worker");
-describe('createWorker function', () => {
-    it('should return a Worker instance when a valid worker script is found', async () => {
+describe(create_worker_1.createWorker.name, () => {
+    it('should create a Worker instance when valid parameters are provided', async () => {
         const workerPath = __dirname + '../tests/dummy-worker.ts';
         const worker = worker_threads_1.isMainThread
             ? await (0, create_worker_1.createWorker)({
@@ -22,4 +22,66 @@ describe('createWorker function', () => {
             await worker.terminate();
         }
     });
+    it('should throw error when not in main thread', async () => {
+        const originalIsMainThread = worker_threads_1.isMainThread;
+        worker_threads_1.isMainThread = false;
+        const workerPath = __dirname + '../tests/dummy-worker.ts';
+        await expect((0, create_worker_1.createWorker)({
+            event: (0, test_helpers_1.createEvent)({
+                eventType: extraction_1.EventType.ExtractionExternalSyncUnitsStart,
+            }),
+            initialState: {},
+            workerPath,
+        })).rejects.toThrow('Worker threads can not start more worker threads.');
+        // Restore original value
+        worker_threads_1.isMainThread = originalIsMainThread;
+    });
+    it('[edge] should handle worker creation with minimal valid data', async () => {
+        const workerPath = __dirname + '../tests/dummy-worker.ts';
+        if (worker_threads_1.isMainThread) {
+            const worker = await (0, create_worker_1.createWorker)({
+                event: (0, test_helpers_1.createEvent)({
+                    eventType: extraction_1.EventType.ExtractionExternalSyncUnitsStart,
+                }),
+                initialState: {},
+                workerPath,
+            });
+            expect(worker).toBeInstanceOf(worker_threads_1.Worker);
+            await worker.terminate();
+        }
+    });
+    it('[edge] should handle worker creation with complex initial state', async () => {
+        const workerPath = __dirname + '../tests/dummy-worker.ts';
+        const complexState = {
+            nested: {
+                data: [1, 2, 3],
+                config: { enabled: true }
+            }
+        };
+        if (worker_threads_1.isMainThread) {
+            const worker = await (0, create_worker_1.createWorker)({
+                event: (0, test_helpers_1.createEvent)({
+                    eventType: extraction_1.EventType.ExtractionDataStart,
+                }),
+                initialState: complexState,
+                workerPath,
+            });
+            expect(worker).toBeInstanceOf(worker_threads_1.Worker);
+            await worker.terminate();
+        }
+    });
+    it('[edge] should handle different event types', async () => {
+        const workerPath = __dirname + '../tests/dummy-worker.ts';
+        if (worker_threads_1.isMainThread) {
+            const worker = await (0, create_worker_1.createWorker)({
+                event: (0, test_helpers_1.createEvent)({
+                    eventType: extraction_1.EventType.ExtractionMetadataStart,
+                }),
+                initialState: {},
+                workerPath,
+            });
+            expect(worker).toBeInstanceOf(worker_threads_1.Worker);
+            await worker.terminate();
+        }
+    });
 });