@devrev/ts-adaas 1.8.0 → 1.8.1-beta.1

package/dist/common/control-protocol.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.emit = void 0;
-const axios_client_1 = require("../http/axios-client");
+const axios_client_internal_1 = require("../http/axios-client-internal");
 const constants_1 = require("./constants");
 const emit = async ({ event, eventType, data, }) => {
     const newEvent = {
@@ -13,7 +13,7 @@ const emit = async ({ event, eventType, data, }) => {
         },
     };
     console.info('Emitting event', newEvent);
-    return axios_client_1.axiosClient.post(event.payload.event_context.callback_url, Object.assign({}, newEvent), {
+    return axios_client_internal_1.axiosClient.post(event.payload.event_context.callback_url, Object.assign({}, newEvent), {
         headers: {
             Accept: 'application/json, text/plain, */*',
             Authorization: event.context.secrets.service_account_token,

package/dist/common/install-initial-domain-mapping.js

@@ -1,7 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.installInitialDomainMapping = installInitialDomainMapping;
-const axios_client_1 = require("../http/axios-client");
+const axios_client_internal_1 = require("../http/axios-client-internal");
 const logger_1 = require("../logger/logger");
 async function installInitialDomainMapping(event, initialDomainMappingJson) {
     var _a, _b, _c, _d, _e, _f, _g;
@@ -13,7 +13,7 @@ async function installInitialDomainMapping(event, initialDomainMappingJson) {
         return;
     }
     // Get snap-in details
-    const snapInResponse = await axios_client_1.axiosClient.get(devrevEndpoint + '/internal/snap-ins.get', {
+    const snapInResponse = await axios_client_internal_1.axiosClient.get(devrevEndpoint + '/internal/snap-ins.get', {
         headers: {
             Authorization: devrevToken,
         },
@@ -33,7 +33,7 @@ async function installInitialDomainMapping(event, initialDomainMappingJson) {
     if (startingRecipeBlueprint &&
         Object.keys(startingRecipeBlueprint).length !== 0) {
         try {
-            const recipeBlueprintResponse = await axios_client_1.axiosClient.post(`${devrevEndpoint}/internal/airdrop.recipe.blueprints.create`, Object.assign({}, startingRecipeBlueprint), {
+            const recipeBlueprintResponse = await axios_client_internal_1.axiosClient.post(`${devrevEndpoint}/internal/airdrop.recipe.blueprints.create`, Object.assign({}, startingRecipeBlueprint), {
                 headers: {
                     Authorization: devrevToken,
                 },
@@ -47,7 +47,7 @@ async function installInitialDomainMapping(event, initialDomainMappingJson) {
     }
     // Install the initial domain mappings
     const additionalMappings = initialDomainMappingJson.additional_mappings || {};
-    const initialDomainMappingInstallResponse = await axios_client_1.axiosClient.post(`${devrevEndpoint}/internal/airdrop.recipe.initial-domain-mappings.install`, Object.assign(Object.assign({ external_system_type: 'ADaaS', import_slug: importSlug, snap_in_slug: snapInSlug }, (recipeBlueprintId && {
+    const initialDomainMappingInstallResponse = await axios_client_internal_1.axiosClient.post(`${devrevEndpoint}/internal/airdrop.recipe.initial-domain-mappings.install`, Object.assign(Object.assign({ external_system_type: 'ADaaS', import_slug: importSlug, snap_in_slug: snapInSlug }, (recipeBlueprintId && {
         starting_recipe_blueprint: recipeBlueprintId,
     })), additionalMappings), {
         headers: {

package/dist/common/install-initial-domain-mapping.test.js

@@ -5,14 +5,20 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const axios_1 = __importDefault(require("axios"));
 const install_initial_domain_mapping_1 = require("./install-initial-domain-mapping");
-const axios_client_1 = require("../http/axios-client");
+const axios_client_internal_1 = require("../http/axios-client-internal");
 const test_helpers_1 = require("../tests/test-helpers");
 const extraction_1 = require("../types/extraction");
 // Mock dependencies
 jest.mock('axios', () => (Object.assign(Object.assign({}, jest.requireActual('axios')), { isAxiosError: jest.fn() })));
-jest.mock('../http/axios-client');
+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(),
+        } });
+});
 jest.mock('../logger/logger');
-const mockAxiosClient = axios_client_1.axiosClient;
+const mockAxiosClient = axios_client_internal_1.axiosClient;
 const mockIsAxiosError = axios_1.default.isAxiosError;
 describe('installInitialDomainMapping', () => {
     // Create mock objects

package/dist/deprecated/uploader/index.js

@@ -34,7 +34,7 @@ var __importStar = (this && this.__importStar) || (function () {
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Uploader = void 0;
-const axios_client_1 = require("../../http/axios-client");
+const axios_client_internal_1 = require("../../http/axios-client-internal");
 const typescript_sdk_1 = require("@devrev/typescript-sdk");
 const fs_1 = __importStar(require("fs"));
 const helpers_1 = require("../common/helpers");
@@ -118,7 +118,7 @@ class Uploader {
     ) {
         const formData = (0, helpers_1.createFormData)(preparedArtifact, fetchedObjects);
         try {
-            const response = await axios_client_1.axiosClient.post(preparedArtifact.url, formData, {
+            const response = await axios_client_internal_1.axiosClient.post(preparedArtifact.url, formData, {
                 headers: {
                     'Content-Type': 'multipart/form-data',
                 },

package/dist/http/axios-client-internal.d.ts

@@ -0,0 +1,2 @@
+declare const axiosClient: import("axios").AxiosInstance;
+export { axiosClient };

package/dist/http/axios-client-internal.js

@@ -0,0 +1,60 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.axiosClient = void 0;
+const axios_1 = __importDefault(require("axios"));
+const axios_retry_1 = __importDefault(require("axios-retry"));
+const axiosClient = axios_1.default.create();
+exports.axiosClient = axiosClient;
+(0, axios_retry_1.default)(axiosClient, {
+    retries: 5,
+    retryDelay: (retryCount, error) => {
+        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k;
+        // If the response status is 429, retry after the time specified in the Retry-After header
+        if (((_a = error.response) === null || _a === void 0 ? void 0 : _a.status) === 429) {
+            const retryAfter = ((_c = (_b = error.response) === null || _b === void 0 ? void 0 : _b.headers) === null || _c === void 0 ? void 0 : _c['retry-after']) ||
+                ((_e = (_d = error.response) === null || _d === void 0 ? void 0 : _d.headers) === null || _e === void 0 ? void 0 : _e['Retry-After']);
+            // Since DevRev API returns the retry-after header in seconds and axios-retry expects milliseconds, we need to convert it to milliseconds
+            const delay = parseInt(retryAfter, 10) * 1000;
+            const delayInSeconds = Math.round(delay / 1000);
+            console.warn(`Retrying ${(_f = error.config) === null || _f === void 0 ? void 0 : _f.method} request to ${(_g = error.config) === null || _g === void 0 ? void 0 : _g.url} in ${delayInSeconds}s due to 429 Too Many Requests.`);
+            return delay;
+        }
+        // Default exponential backoff algorithm: 1 * 2 ^ retryCount * 1000ms
+        // This will retry requests after 2, 4, 8, 16 and 32 seconds
+        const delay = axios_retry_1.default.exponentialDelay(retryCount, error, 1000);
+        const delayInSeconds = Math.round(delay / 1000);
+        console.warn(`Retrying ${(_h = error.config) === null || _h === void 0 ? void 0 : _h.method} request to ${(_j = error.config) === null || _j === void 0 ? void 0 : _j.url} in ${delayInSeconds}s due to ${(_k = error.response) === null || _k === void 0 ? void 0 : _k.status} error.`);
+        return delay;
+    },
+    retryCondition: (error) => {
+        var _a, _b, _c, _d, _e, _f;
+        const retryAfter = ((_b = (_a = error.response) === null || _a === void 0 ? void 0 : _a.headers) === null || _b === void 0 ? void 0 : _b['retry-after']) ||
+            ((_d = (_c = error.response) === null || _c === void 0 ? void 0 : _c.headers) === null || _d === void 0 ? void 0 : _d['Retry-After']);
+        // 5xx errors
+        if (((_e = error.response) === null || _e === void 0 ? void 0 : _e.status) && error.response.status >= 500) {
+            return true;
+        }
+        // 429 errors when retry-after header is present
+        else if (((_f = error.response) === null || _f === void 0 ? void 0 : _f.status) &&
+            error.response.status === 429 &&
+            retryAfter &&
+            !isNaN(Number(retryAfter)) &&
+            Number(retryAfter) >= 0) {
+            return true;
+        }
+        // all other errors
+        else {
+            return false;
+        }
+    },
+    onMaxRetryTimesExceeded(error) {
+        var _a, _b, _c, _d, _e, _f;
+        (_b = (_a = error.config) === null || _a === void 0 ? void 0 : _a.headers) === null || _b === void 0 ? true : delete _b.authorization;
+        (_d = (_c = error.config) === null || _c === void 0 ? void 0 : _c.headers) === null || _d === void 0 ? true : delete _d.Authorization;
+        (_e = error.request) === null || _e === void 0 ? true : delete _e._header;
+        console.error(`Request to ${(_f = error.config) === null || _f === void 0 ? void 0 : _f.url} failed after max retries. Error`, error);
+    },
+});

package/dist/http/axios-client-internal.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/http/axios-client-internal.test.js

@@ -0,0 +1,189 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const axios_mock_adapter_1 = __importDefault(require("axios-mock-adapter"));
+const axios_client_internal_1 = require("./axios-client-internal");
+describe('Internal Axios Client', () => {
+    let mockAdapter;
+    beforeEach(() => {
+        jest.clearAllMocks();
+        jest.spyOn(console, 'warn').mockImplementation(() => { });
+        mockAdapter = new axios_mock_adapter_1.default(axios_client_internal_1.axiosClient);
+    });
+    afterEach(() => {
+        jest.restoreAllMocks();
+        mockAdapter.restore();
+    });
+    it('should not retry on 200 response', async () => {
+        // Arrange
+        const testUrl = '/test-endpoint';
+        const responseData = { id: 1, title: 'test title' };
+        mockAdapter.onGet(testUrl).reply(200, responseData);
+        // Act
+        await axios_client_internal_1.axiosClient.get(testUrl);
+        // Assert
+        expect(mockAdapter.history.get).toHaveLength(1);
+    });
+    it('should not retry when response is 400', async () => {
+        // Arrange
+        const testUrl = '/test-endpoint';
+        const errorData = { error: 'Bad Request' };
+        mockAdapter.onGet(testUrl).reply(400, errorData);
+        // Act
+        await expect(axios_client_internal_1.axiosClient.get(testUrl)).rejects.toMatchObject({
+            response: { status: 400 },
+        });
+        // Assert
+        expect(mockAdapter.history.get).toHaveLength(1);
+    });
+    // TODO: This test is working as expected, but it takes too long to run. Not
+    // sure if it is good idea to have it.
+    // it('should retry on 500 response', async () => {
+    //   // Arrange
+    //   const testUrl = '/test-endpoint';
+    //   mockAdapter.onGet(testUrl).reply(500);
+    //   // Act & Assert
+    //   await expect(axiosClient.get(testUrl)).rejects.toMatchObject({
+    //     response: { status: 500 },
+    //   });
+    //   // Assert
+    //   expect(mockAdapter.history.get).toHaveLength(6);
+    // }, 100000);
+    it('should retry 2 times when response is 500 and then succeed third time when response is 200', async () => {
+        // Arrange
+        const testUrl = '/test-endpoint';
+        const successData = { message: 'success after 2 retries' };
+        mockAdapter
+            .onGet(testUrl)
+            .replyOnce(500)
+            .onGet(testUrl)
+            .replyOnce(500)
+            .onGet(testUrl)
+            .reply(200, successData);
+        // Act
+        const response = await axios_client_internal_1.axiosClient.get(testUrl);
+        // Assert
+        expect(response.status).toBe(200);
+        expect(response.data).toEqual(successData);
+        expect(mockAdapter.history.get).toHaveLength(3); // 1 initial + 2 retries
+    }, 10000); // Allow time for 2 retries
+    it('should retry once after 2 seconds when response is 429 and Retry-After header is valid value', async () => {
+        // Arrange
+        const testUrl = '/test-endpoint';
+        const errorData = { error: 'Too Many Requests' };
+        const successData = { message: 'success after rate limit retry' };
+        mockAdapter
+            .onGet(testUrl)
+            .replyOnce(429, errorData, {
+            'Retry-After': '2',
+        })
+            .onGet(testUrl)
+            .reply(200, successData);
+        // Act
+        const response = await axios_client_internal_1.axiosClient.get(testUrl);
+        // Assert
+        expect(response.status).toBe(200);
+        expect(response.data).toEqual(successData);
+        expect(mockAdapter.history.get).toHaveLength(2); // 1 initial + 1 retry
+    });
+    it('should retry once after 2 seconds and measure time between retries when response is 429 and Retry-After header is valid value', async () => {
+        // Arrange
+        const testUrl = '/test-endpoint';
+        const errorData = { error: 'Too Many Requests' };
+        const successData = { message: 'success after rate limit retry' };
+        const retryAfterSeconds = 2;
+        mockAdapter
+            .onGet(testUrl)
+            .replyOnce(429, errorData, {
+            'Retry-After': retryAfterSeconds.toString(),
+        })
+            .onGet(testUrl)
+            .reply(200, successData);
+        // Act
+        const startTime = Date.now();
+        const response = await axios_client_internal_1.axiosClient.get(testUrl);
+        const endTime = Date.now();
+        const actualWaitTime = endTime - startTime;
+        // Assert
+        expect(response.status).toBe(200);
+        expect(response.data).toEqual(successData);
+        expect(mockAdapter.history.get).toHaveLength(2); // 1 initial + 1 retry
+        const expectedWaitTime = retryAfterSeconds * 1000;
+        expect(actualWaitTime).toBeGreaterThanOrEqual(expectedWaitTime - 100);
+        expect(actualWaitTime).toBeLessThan(expectedWaitTime + 1000); // Allow up to 1s extra for processing
+    });
+    it('should retry when response is 429 and Retry-After header is lowercase', async () => {
+        // Arrange
+        const testUrl = '/test-endpoint';
+        const errorData = { error: 'Too Many Requests' };
+        const successData = { message: 'success after rate limit retry' };
+        mockAdapter
+            .onGet(testUrl)
+            .replyOnce(429, errorData, { 'retry-after': '2' })
+            .onGet(testUrl)
+            .reply(200, successData);
+        // Act
+        const response = await axios_client_internal_1.axiosClient.get(testUrl);
+        // Assert
+        expect(response.status).toBe(200);
+        expect(response.data).toEqual(successData);
+        expect(mockAdapter.history.get).toHaveLength(2); // 1 initial + 1 retry
+    });
+    it('[edge] should not retry when response is 429 and there is no Retry-After header', async () => {
+        // Arrange
+        const testUrl = '/test-endpoint';
+        const errorData = { error: 'Too Many Requests' };
+        mockAdapter.onGet(testUrl).reply(429, errorData);
+        // Act
+        await expect(axios_client_internal_1.axiosClient.get(testUrl)).rejects.toMatchObject({
+            response: { status: 429 },
+        });
+        // Assert
+        expect(mockAdapter.history.get).toHaveLength(1);
+    });
+    it('[edge] should retry when response is 429 and Retry-After header is 0', async () => {
+        // Arrange
+        const testUrl = '/test-endpoint';
+        const errorData = { error: 'Too Many Requests' };
+        const successData = { message: 'success after rate limit retry' };
+        mockAdapter
+            .onGet(testUrl)
+            .replyOnce(429, errorData, { 'Retry-After': '0' })
+            .onGet(testUrl)
+            .reply(200, successData);
+        // Act
+        const response = await axios_client_internal_1.axiosClient.get(testUrl);
+        // Assert
+        expect(response.status).toBe(200);
+        expect(response.data).toEqual(successData);
+        expect(mockAdapter.history.get).toHaveLength(2); // 1 initial + 1 retry
+    });
+    it('[edge] should not retry when response is 429 and Retry-After header is negative', async () => {
+        // Arrange
+        const testUrl = '/test-endpoint';
+        const errorData = { error: 'Too Many Requests' };
+        mockAdapter.onGet(testUrl).reply(429, errorData, { 'Retry-After': '-1' });
+        // Act
+        await expect(axios_client_internal_1.axiosClient.get(testUrl)).rejects.toMatchObject({
+            response: { status: 429 },
+        });
+        // Assert
+        expect(mockAdapter.history.get).toHaveLength(1);
+    });
+    it('[edge] should not retry when response is 429 and Retry-After header is invalid value', async () => {
+        // Arrange
+        const testUrl = '/test-endpoint';
+        const errorData = { error: 'Too Many Requests' };
+        mockAdapter
+            .onGet(testUrl)
+            .reply(429, errorData, { 'Retry-After': 'invalid' });
+        // Act
+        await expect(axios_client_internal_1.axiosClient.get(testUrl)).rejects.toMatchObject({
+            response: { status: 429 },
+        });
+        // Assert
+        expect(mockAdapter.history.get).toHaveLength(1);
+    });
+});

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/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/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(),
@@ -41,9 +41,9 @@ describe('Uploader Class Tests', () => {
     });
     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);
@@ -58,7 +58,7 @@ describe('Uploader Class Tests', () => {
     it('should handle failure in getArtifactUploadUrl', 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);
@@ -68,9 +68,9 @@ describe('Uploader Class Tests', () => {
     it('should handle failure in uploadArtifact', 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);
@@ -80,11 +80,11 @@ describe('Uploader Class Tests', () => {
     it('should handle failure in confirmArtifactUpload', 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);

package/dist/workers/default-workers/attachments-extraction.js

@@ -1,11 +1,15 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
+const axios_1 = __importDefault(require("axios"));
 const index_1 = require("../../index");
-const axios_client_1 = require("../../http/axios-client");
+const axios_client_internal_1 = require("../../http/axios-client-internal");
 const getAttachmentStream = async ({ item, }) => {
     const { id, url } = item;
     try {
-        const fileStreamResponse = await axios_client_1.axiosClient.get(url, {
+        const fileStreamResponse = await axios_client_internal_1.axiosClient.get(url, {
             responseType: 'stream',
             headers: {
                 'Accept-Encoding': 'identity',
@@ -14,7 +18,7 @@ const getAttachmentStream = async ({ item, }) => {
         return { httpStream: fileStreamResponse };
     }
     catch (error) {
-        if (axios_client_1.axios.isAxiosError(error)) {
+        if (axios_1.default.isAxiosError(error)) {
             console.warn(`Error while fetching attachment ${id} from URL.`, (0, index_1.serializeAxiosError)(error));
             console.warn('Failed attachment metadata', item);
         }

package/dist/workers/worker-adapter.js

@@ -1,8 +1,11 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.WorkerAdapter = void 0;
 exports.createWorkerAdapter = createWorkerAdapter;
-const axios_client_1 = require("../http/axios-client");
+const axios_1 = __importDefault(require("axios"));
 const extraction_1 = require("../types/extraction");
 const loading_1 = require("../types/loading");
 const constants_1 = require("../common/constants");
@@ -419,7 +422,7 @@ class WorkerAdapter {
             // TODO: Update mapper (optional)
         }
         catch (error) {
-            if (axios_client_1.axios.isAxiosError(error)) {
+            if (axios_1.default.isAxiosError(error)) {
                 if (((_a = error.response) === null || _a === void 0 ? void 0 : _a.status) === 404) {
                     // Create item in external system if mapper record not found
                     const { id, delay, error } = await itemTypeToLoad.create({

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

@@ -74,16 +74,32 @@ describe('WorkerAdapter', () => {
                 },
             };
             // Mock getting attachments from each artifact
-            adapter['uploader'].getAttachmentsFromArtifactId = jest.fn()
+            adapter['uploader'].getAttachmentsFromArtifactId = jest
+                .fn()
                 .mockResolvedValueOnce({
                 attachments: [
-                    { url: 'http://example.com/file1.pdf', id: 'attachment1', file_name: 'file1.pdf', parent_id: 'parent1' },
-                    { url: 'http://example.com/file2.pdf', id: 'attachment2', file_name: 'file2.pdf', parent_id: 'parent2' },
+                    {
+                        url: 'http://example.com/file1.pdf',
+                        id: 'attachment1',
+                        file_name: 'file1.pdf',
+                        parent_id: 'parent1',
+                    },
+                    {
+                        url: 'http://example.com/file2.pdf',
+                        id: 'attachment2',
+                        file_name: 'file2.pdf',
+                        parent_id: 'parent2',
+                    },
                 ],
             })
                 .mockResolvedValueOnce({
                 attachments: [
-                    { url: 'http://example.com/file3.pdf', id: 'attachment3', file_name: 'file3.pdf', parent_id: 'parent3' },
+                    {
+                        url: 'http://example.com/file3.pdf',
+                        id: 'attachment3',
+                        file_name: 'file3.pdf',
+                        parent_id: 'parent3',
+                    },
                 ],
             });
             // Mock the initializeRepos method
@@ -115,9 +131,16 @@ describe('WorkerAdapter', () => {
                 },
             };
             // Mock getting attachments
-            adapter['uploader'].getAttachmentsFromArtifactId = jest.fn().mockResolvedValue({
+            adapter['uploader'].getAttachmentsFromArtifactId = jest
+                .fn()
+                .mockResolvedValue({
                 attachments: [
-                    { url: 'http://example.com/file1.pdf', id: 'attachment1', file_name: 'file1.pdf', parent_id: 'parent1' },
+                    {
+                        url: 'http://example.com/file1.pdf',
+                        id: 'attachment1',
+                        file_name: 'file1.pdf',
+                        parent_id: 'parent1',
+                    },
                 ],
             });
             adapter.initializeRepos = jest.fn();
@@ -145,9 +168,16 @@ describe('WorkerAdapter', () => {
                 },
             };
             // Mock getting attachments
-            adapter['uploader'].getAttachmentsFromArtifactId = jest.fn().mockResolvedValue({
+            adapter['uploader'].getAttachmentsFromArtifactId = jest
+                .fn()
+                .mockResolvedValue({
                 attachments: [
-                    { url: 'http://example.com/file1.pdf', id: 'attachment1', file_name: 'file1.pdf', parent_id: 'parent1' },
+                    {
+                        url: 'http://example.com/file1.pdf',
+                        id: 'attachment1',
+                        file_name: 'file1.pdf',
+                        parent_id: 'parent1',
+                    },
                 ],
             });
             // Mock the required methods
@@ -197,7 +227,9 @@ describe('WorkerAdapter', () => {
             };
             // Mock error when getting attachments
             const mockError = new Error('Failed to get attachments');
-            adapter['uploader'].getAttachmentsFromArtifactId = jest.fn().mockResolvedValue({
+            adapter['uploader'].getAttachmentsFromArtifactId = jest
+                .fn()
+                .mockResolvedValue({
                 error: mockError,
             });
             // Mock methods
@@ -227,7 +259,9 @@ describe('WorkerAdapter', () => {
                 },
             };
             // Mock getting empty attachments
-            adapter['uploader'].getAttachmentsFromArtifactId = jest.fn().mockResolvedValue({
+            adapter['uploader'].getAttachmentsFromArtifactId = jest
+                .fn()
+                .mockResolvedValue({
                 attachments: [],
             });
             // Mock methods
@@ -258,7 +292,9 @@ describe('WorkerAdapter', () => {
                 },
             };
             // Mock getting attachments
-            adapter['uploader'].getAttachmentsFromArtifactId = jest.fn().mockResolvedValue({
+            adapter['uploader'].getAttachmentsFromArtifactId = jest
+                .fn()
+                .mockResolvedValue({
                 attachments: [{ id: 'attachment1' }],
             });
             // Mock methods
@@ -305,7 +341,9 @@ describe('WorkerAdapter', () => {
                 },
             };
             // Mock getting attachments
-            adapter['uploader'].getAttachmentsFromArtifactId = jest.fn().mockResolvedValue({
+            adapter['uploader'].getAttachmentsFromArtifactId = jest
+                .fn()
+                .mockResolvedValue({
                 attachments: [{ id: 'attachment1' }],
             });
             // Mock methods
@@ -319,7 +357,9 @@ describe('WorkerAdapter', () => {
                 delay: 30,
             });
             // The artifactIds array should remain unchanged
-            expect(adapter.state.toDevRev.attachmentsMetadata.artifactIds).toEqual(['artifact1']);
+            expect(adapter.state.toDevRev.attachmentsMetadata.artifactIds).toEqual([
+                'artifact1',
+            ]);
         });
         it('should handle error from iterator', async () => {
             // Arrange
@@ -328,7 +368,7 @@ describe('WorkerAdapter', () => {
                 return {
                     // Return an object with a `streamAll` method that resolves to your desired value.
                     streamAll: jest.fn().mockResolvedValue({
-                        error: "Mock error",
+                        error: 'Mock error',
                     }),
                 };
             });
@@ -341,7 +381,9 @@ describe('WorkerAdapter', () => {
                 },
             };
             // Mock getting attachments
-            adapter['uploader'].getAttachmentsFromArtifactId = jest.fn().mockResolvedValue({
+            adapter['uploader'].getAttachmentsFromArtifactId = jest
+                .fn()
+                .mockResolvedValue({
                 attachments: [{ id: 'attachment1' }],
             });
             // Mock methods
@@ -352,10 +394,12 @@ describe('WorkerAdapter', () => {
             });
             // Assert
             expect(result).toEqual({
-                error: "Mock error",
+                error: 'Mock error',
             });
             // The artifactIds array should remain unchanged
-            expect(adapter.state.toDevRev.attachmentsMetadata.artifactIds).toEqual(['artifact1']);
+            expect(adapter.state.toDevRev.attachmentsMetadata.artifactIds).toEqual([
+                'artifact1',
+            ]);
         });
         it('should reset lastProcessed and attachment IDs list after processing all artifacts', async () => {
             const mockStream = jest.fn();
@@ -366,12 +410,28 @@ describe('WorkerAdapter', () => {
                     lastProcessedAttachmentsIdsList: [],
                 },
             };
-            adapter['uploader'].getAttachmentsFromArtifactId = jest.fn()
+            adapter['uploader'].getAttachmentsFromArtifactId = jest
+                .fn()
                 .mockResolvedValueOnce({
                 attachments: [
-                    { url: 'http://example.com/file1.pdf', id: 'attachment1', file_name: 'file1.pdf', parent_id: 'parent1' },
-                    { url: 'http://example.com/file2.pdf', id: 'attachment2', file_name: 'file2.pdf', parent_id: 'parent2' },
-                    { url: 'http://example.com/file3.pdf', id: 'attachment3', file_name: 'file3.pdf', parent_id: 'parent3' },
+                    {
+                        url: 'http://example.com/file1.pdf',
+                        id: 'attachment1',
+                        file_name: 'file1.pdf',
+                        parent_id: 'parent1',
+                    },
+                    {
+                        url: 'http://example.com/file2.pdf',
+                        id: 'attachment2',
+                        file_name: 'file2.pdf',
+                        parent_id: 'parent2',
+                    },
+                    {
+                        url: 'http://example.com/file3.pdf',
+                        id: 'attachment3',
+                        file_name: 'file3.pdf',
+                        parent_id: 'parent3',
+                    },
                 ],
             });
             adapter.processAttachment = jest.fn().mockResolvedValue(null);
@@ -395,7 +455,9 @@ describe('WorkerAdapter', () => {
             });
             // Spy on the parentPort.postMessage method
             if (workerThreads.parentPort) {
-                jest.spyOn(workerThreads.parentPort, 'postMessage').mockImplementation(mockPostMessage);
+                jest
+                    .spyOn(workerThreads.parentPort, 'postMessage')
+                    .mockImplementation(mockPostMessage);
             }
             else {
                 // If parentPort is null (not in worker context), create a mock
@@ -409,7 +471,9 @@ describe('WorkerAdapter', () => {
             jest.restoreAllMocks();
         });
         test('should correctly emit event', async () => {
-            adapter['adapterState'].postState = jest.fn().mockResolvedValue(undefined);
+            adapter['adapterState'].postState = jest
+                .fn()
+                .mockResolvedValue(undefined);
             adapter.uploadAllRepos = jest.fn().mockResolvedValue(undefined);
             await adapter.emit(types_1.ExtractorEventType.ExtractionMetadataError, {
                 reports: [],
@@ -426,7 +490,9 @@ describe('WorkerAdapter', () => {
             expect(counter.counter).toBe(1);
         });
         test('should correctly emit one event even if postState errors', async () => {
-            adapter['adapterState'].postState = jest.fn().mockRejectedValue(new Error('postState error'));
+            adapter['adapterState'].postState = jest
+                .fn()
+                .mockRejectedValue(new Error('postState error'));
             adapter.uploadAllRepos = jest.fn().mockResolvedValue(undefined);
             await adapter.emit(types_1.ExtractorEventType.ExtractionMetadataError, {
                 reports: [],
@@ -435,8 +501,12 @@ describe('WorkerAdapter', () => {
             expect(counter.counter).toBe(1);
         });
         test('should correctly emit one event even if uploadAllRepos errors', async () => {
-            adapter['adapterState'].postState = jest.fn().mockResolvedValue(undefined);
-            adapter.uploadAllRepos = jest.fn().mockRejectedValue(new Error('uploadAllRepos error'));
+            adapter['adapterState'].postState = jest
+                .fn()
+                .mockResolvedValue(undefined);
+            adapter.uploadAllRepos = jest
+                .fn()
+                .mockRejectedValue(new Error('uploadAllRepos error'));
             await adapter.emit(types_1.ExtractorEventType.ExtractionMetadataError, {
                 reports: [],
                 processed_files: [],

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@devrev/ts-adaas",
-  "version": "1.8.0",
+  "version": "1.8.1-beta.1",
   "description": "Typescript library containing the ADaaS(AirDrop as a Service) control protocol.",
   "type": "commonjs",
   "main": "./dist/index.js",
@@ -30,6 +30,7 @@
     "@types/yargs": "^17.0.33",
     "@typescript-eslint/eslint-plugin": "^8.32.1",
     "@typescript-eslint/parser": "^8.32.1",
+    "axios-mock-adapter": "^2.1.0",
     "eslint": "^9.26.0",
     "eslint-config-prettier": "^10.1.5",
     "eslint-plugin-prettier": "^5.1.3",