@aws/lsp-codewhisperer 0.0.9 → 0.0.10

package/out/language-server/netTransform/tests/transformHandler.test.js

@@ -0,0 +1,242 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const codewhisperer_streaming_1 = require("@amzn/codewhisperer-streaming");
+const assert = require("assert");
+const aws_sdk_1 = require("aws-sdk");
+const chai_1 = require("chai");
+const fs = require("fs");
+const got_1 = require("got");
+const ts_sinon_1 = require("ts-sinon");
+const codewhispererStreamingClient_1 = require("../../../client/streamingClient/codewhispererStreamingClient");
+const models_1 = require("../models");
+const transformHandler_1 = require("../transformHandler");
+const sinon = require("sinon");
+const mocked$Response = {
+    $response: {
+        hasNextPage: ts_sinon_1.default.mock(),
+        nextPage: ts_sinon_1.default.mock(),
+        data: undefined,
+        error: undefined,
+        requestId: '',
+        redirectCount: 0,
+        retryCount: 0,
+        httpResponse: new aws_sdk_1.HttpResponse(),
+    },
+};
+const testUploadId = 'test-upoload-id';
+const testTransformId = 'test-transform-id';
+const payloadFileName = 'C:\\test.zip';
+describe('Test Transform handler ', () => {
+    let client;
+    let workspace;
+    let transformHandler;
+    const mockedLogging = (0, ts_sinon_1.stubInterface)();
+    beforeEach(async () => {
+        // Set up the server with a mock service
+        client = (0, ts_sinon_1.stubInterface)();
+        workspace = (0, ts_sinon_1.stubInterface)();
+        transformHandler = new transformHandler_1.TransformHandler(client, workspace, mockedLogging);
+    });
+    describe('test upload artifact', () => {
+        it('call upload method correctly', async () => {
+            const putStub = sinon.stub(got_1.default, 'put').resolves({ statusCode: 'Success' });
+            const readFileSyncStub = sinon.stub(fs, 'readFileSync').returns('text file content');
+            await transformHandler.uploadArtifactToS3Async(payloadFileName, {
+                uploadId: testUploadId,
+                uploadUrl: 'dummy-upload-url',
+                kmsKeyArn: 'ResourceArn',
+                ...mocked$Response,
+            }, 'dummy-256');
+            ts_sinon_1.default.assert.callCount(putStub, 1);
+            ts_sinon_1.default.assert.callCount(readFileSyncStub, 1);
+            putStub.restore();
+            readFileSyncStub.restore();
+        });
+    });
+    describe('Test transform create presign url and upload', () => {
+        beforeEach(async () => {
+            // mock default return value for createUploadUrl
+            client.codeModernizerCreateUploadUrl.resolves({
+                uploadId: testUploadId,
+                uploadUrl: 'dummy-upload-url',
+                kmsKeyArn: 'ResourceArn',
+                ...mocked$Response,
+            });
+        });
+        it('returns upload id correctly', async () => {
+            const readFileSyncStub = sinon.stub(fs, 'readFileSync').returns('text file content');
+            const uploadStub = sinon.stub(transformHandler, 'uploadArtifactToS3Async');
+            const res = await transformHandler.uploadPayloadAsync(payloadFileName);
+            ts_sinon_1.default.assert.callCount(readFileSyncStub, 1);
+            ts_sinon_1.default.assert.callCount(uploadStub, 1);
+            assert.equal(res, testUploadId);
+            uploadStub.restore();
+            readFileSyncStub.restore();
+        });
+        it('should throw error if uploadArtifactToS3Async fails', async () => {
+            const mockError = new Error('Error in uploadArtifactToS3 call');
+            const readFileSyncStub = sinon.stub(fs, 'readFileSync').returns('text file content');
+            sinon.stub(transformHandler, 'uploadArtifactToS3Async').rejects(mockError);
+            // Call the method to be tested
+            try {
+                await transformHandler.uploadPayloadAsync(payloadFileName);
+            }
+            catch (error) {
+                // Assertions
+                (0, chai_1.expect)(error.message).to.equal(mockError.message);
+                sinon.assert.calledOnce(readFileSyncStub);
+                readFileSyncStub.restore();
+            }
+        });
+        /*
+        it('should upload payload and return uploadId', async () => {
+            workspace.fs.getTempDirPath = simon.stub().returns('C:\\tmp')
+            const zipStub = sinon.stub(transformHandler, 'zipCodeAsync').returns(Promise.resolve(payloadFileName))
+            const uploadStub = sinon.stub(transformHandler, 'uploadPayloadAsync').returns(Promise.resolve(testUploadId))
+            const result = await transformHandler.preTransformationUploadCode(payloadFileName)
+            expect(result).to.equal(testUploadId)
+            sinon.assert.calledOnce(zipStub)
+            sinon.assert.calledOnce(uploadStub)
+            zipStub.restore()
+            uploadStub.restore()
+        })
+
+        it('should throw error if upload fails', async () => {
+            // Mocking necessary dependencies and methods
+            const mockError = new Error('Upload failed')
+            const requestString = JSON.stringify(EXAMPLE_REQUEST)
+            const request = JSON.parse(requestString) as StartTransformRequest
+            workspace.fs.getTempDirPath = simon.stub().returns('C:\\tmp')
+            const zipStub = sinon.stub(transformHandler, 'zipCodeAsync').returns(Promise.resolve(payloadFileName))
+            sinon.stub(transformHandler, 'uploadPayloadAsync').rejects(mockError)
+            try {
+                await transformHandler.preTransformationUploadCode(request)
+            } catch (error) {
+                sinon.assert.calledOnce(zipStub)
+                expect((error as Error).message).to.equal(mockError.message)
+                zipStub.restore()
+            }
+        })
+            */
+    });
+    describe('Test cancel transform job', () => {
+        beforeEach(async () => {
+            // mock default return value for cancelTransform
+            client.codeModernizerStopCodeTransformation.returns(Promise.resolve({
+                transformationStatus: 'STOPPED',
+                ...mocked$Response,
+            }));
+        });
+        it('should cancel transform', async () => {
+            const requestString = JSON.stringify({ TransformationJobId: testTransformId });
+            const request = JSON.parse(requestString);
+            const res = await transformHandler.cancelTransformation(request);
+            (0, chai_1.expect)(res.TransformationJobStatus).to.equal(models_1.CancellationJobStatus.SUCCESSFULLY_CANCELLED);
+        });
+        it('should throw error if cancellation fails', async () => {
+            client.codeModernizerStopCodeTransformation.returns(Promise.resolve({
+                transformationStatus: 'COMPLETED',
+                ...mocked$Response,
+            }));
+            const requestString = JSON.stringify({ TransformationJobId: testTransformId });
+            const request = JSON.parse(requestString);
+            const res = await transformHandler.cancelTransformation(request);
+            (0, chai_1.expect)(res.TransformationJobStatus).to.equal(models_1.CancellationJobStatus.FAILED_TO_CANCEL);
+        });
+    });
+    const mockedCredentialsProvider = {
+        getCredentials: sinon.stub().returns({ token: 'mockedToken' }),
+    };
+    describe('StreamingClient', () => {
+        it('should create a new streaming client', async () => {
+            const streamingClient = new codewhispererStreamingClient_1.StreamingClient();
+            const client = await streamingClient.getStreamingClient(mockedCredentialsProvider);
+            (0, chai_1.expect)(client).to.be.instanceOf(codewhisperer_streaming_1.CodeWhispererStreaming);
+        });
+    });
+    describe('createStreamingClient', () => {
+        it('should create a new streaming client with correct configurations', async () => {
+            const client = await (0, codewhispererStreamingClient_1.createStreamingClient)(mockedCredentialsProvider);
+            (0, chai_1.expect)(client).to.be.instanceOf(codewhisperer_streaming_1.CodeWhispererStreaming);
+        });
+    });
+    describe('Test get transformJob', () => {
+        beforeEach(async () => {
+            // mock default return value for Transformation Job
+            client.codeModernizerGetCodeTransformation.returns(Promise.resolve({
+                transformationJob: {
+                    jobId: testTransformId,
+                    status: 'COMPLETED',
+                    ...mocked$Response,
+                },
+                ...mocked$Response,
+            }));
+        });
+        it('should get transform', async () => {
+            const requestString = JSON.stringify({ TransformationJobId: testTransformId });
+            const request = JSON.parse(requestString);
+            const res = await transformHandler.getTransformation(request);
+            (0, chai_1.expect)(res.TransformationJob.status).to.equal('COMPLETED');
+        });
+    });
+    describe('Test get transformJob for failed case', () => {
+        beforeEach(async () => {
+            // mock default return value for Transformation Job
+            client.codeModernizerGetCodeTransformation.returns(Promise.resolve({
+                transformationJob: {
+                    jobId: testTransformId,
+                    status: 'FAILED',
+                    ...mocked$Response,
+                },
+                ...mocked$Response,
+            }));
+        });
+        it('should get transform for failed case', async () => {
+            const requestString = JSON.stringify({ TransformationJobId: testTransformId });
+            const request = JSON.parse(requestString);
+            const res = await transformHandler.getTransformation(request);
+            (0, chai_1.expect)(res.TransformationJob.status).to.equal('FAILED');
+        });
+    });
+    describe('Test get transform plan', () => {
+        beforeEach(async () => {
+            // mock default return value for Transformation plan
+            const mockPlanString = JSON.stringify({
+                transformationPlan: {
+                    transformationSteps: [
+                        {
+                            id: '1',
+                            name: 'PlanStepName 1',
+                            description: 'PlanStepDescription 1',
+                            status: 'COMPLETED',
+                            progressUpdates: [
+                                {
+                                    name: 'ProgressUpdateName 1 for PlanStep 1',
+                                    status: 'COMPLETED',
+                                    description: 'ProgressUpdateDescription 1 for PlanStep 1',
+                                    startTime: '2024-03-11T23:27:33.935Z',
+                                    endTime: '2024-03-11T23:27:34.038Z',
+                                },
+                            ],
+                            startTime: '2024-03-12T18:49:41.431Z',
+                            endTime: '2024-03-12T18:49:42.431Z',
+                        },
+                    ],
+                },
+            });
+            const response = JSON.parse(mockPlanString);
+            client.codeModernizerGetCodeTransformationPlan.returns(Promise.resolve(response));
+        });
+        it('should get transform plan', async () => {
+            const requestString = JSON.stringify({ TransformationJobId: testTransformId });
+            const request = JSON.parse(requestString);
+            const res = await transformHandler.getTransformationPlan(request);
+            (0, chai_1.expect)(res.TransformationPlan.transformationSteps[0].status).to.equal('COMPLETED');
+            (0, chai_1.expect)(res.TransformationPlan.transformationSteps[0].name).to.equal('PlanStepName 1');
+            if (res.TransformationPlan.transformationSteps[0].progressUpdates) {
+                (0, chai_1.expect)(res.TransformationPlan.transformationSteps[0].progressUpdates[0].name).to.equal('ProgressUpdateName 1 for PlanStep 1');
+            }
+        });
+    });
+});
+//# sourceMappingURL=transformHandler.test.js.map

package/out/language-server/netTransform/tests/transformHandler.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transformHandler.test.js","sourceRoot":"","sources":["../../../../src/language-server/netTransform/tests/transformHandler.test.ts"],"names":[],"mappings":";;AAAA,2EAAsE;AAEtE,iCAAgC;AAChC,qCAAsC;AACtC,+BAA6B;AAC7B,yBAAwB;AACxB,6BAAqB;AACrB,uCAA2E;AAC3E,+GAAqH;AAErH,sCAMkB;AAClB,0DAAsD;AAEtD,+BAA+B;AAC/B,MAAM,eAAe,GAAG;IACpB,SAAS,EAAE;QACP,WAAW,EAAE,kBAAK,CAAC,IAAI,EAAE;QACzB,QAAQ,EAAE,kBAAK,CAAC,IAAI,EAAE;QACtB,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,SAAS;QAChB,SAAS,EAAE,EAAE;QACb,aAAa,EAAE,CAAC;QAChB,UAAU,EAAE,CAAC;QACb,YAAY,EAAE,IAAI,sBAAY,EAAE;KACnC;CACJ,CAAA;AACD,MAAM,YAAY,GAAG,iBAAiB,CAAA;AACtC,MAAM,eAAe,GAAG,mBAAmB,CAAA;AAC3C,MAAM,eAAe,GAAG,cAAc,CAAA;AAEtC,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACrC,IAAI,MAAkD,CAAA;IACtD,IAAI,SAAqC,CAAA;IACzC,IAAI,gBAAkC,CAAA;IACtC,MAAM,aAAa,GAAG,IAAA,wBAAa,GAAW,CAAA;IAC9C,UAAU,CAAC,KAAK,IAAI,EAAE;QAClB,wCAAwC;QACxC,MAAM,GAAG,IAAA,wBAAa,GAA6B,CAAA;QACnD,SAAS,GAAG,IAAA,wBAAa,GAAa,CAAA;QACtC,gBAAgB,GAAG,IAAI,mCAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,aAAa,CAAC,CAAA;IAC7E,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,aAAG,EAAE,KAAK,CAAC,CAAC,QAAQ,CAAC,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC,CAAA;YAC1E,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAA;YACpF,MAAM,gBAAgB,CAAC,uBAAuB,CAC1C,eAAe,EACf;gBACI,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,kBAAkB;gBAC7B,SAAS,EAAE,aAAa;gBACxB,GAAG,eAAe;aACrB,EACD,WAAW,CACd,CAAA;YACD,kBAAK,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,CAAA;YAClC,kBAAK,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAA;YAC3C,OAAO,CAAC,OAAO,EAAE,CAAA;YACjB,gBAAgB,CAAC,OAAO,EAAE,CAAA;QAC9B,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,8CAA8C,EAAE,GAAG,EAAE;QAC1D,UAAU,CAAC,KAAK,IAAI,EAAE;YAClB,gDAAgD;YAChD,MAAM,CAAC,6BAA6B,CAAC,QAAQ,CAAC;gBAC1C,QAAQ,EAAE,YAAY;gBACtB,SAAS,EAAE,kBAAkB;gBAC7B,SAAS,EAAE,aAAa;gBACxB,GAAG,eAAe;aACrB,CAAC,CAAA;QACN,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YACzC,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAA;YACpF,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,gBAAgB,EAAE,yBAAyB,CAAC,CAAA;YAC1E,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAA;YACtE,kBAAK,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAA;YAC3C,kBAAK,CAAC,MAAM,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC,CAAC,CAAA;YACrC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAA;YAC/B,UAAU,CAAC,OAAO,EAAE,CAAA;YACpB,gBAAgB,CAAC,OAAO,EAAE,CAAA;QAC9B,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAA;YAC/D,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,EAAE,cAAc,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAA;YACpF,KAAK,CAAC,IAAI,CAAC,gBAAgB,EAAE,yBAAyB,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;YAE1E,+BAA+B;YAC/B,IAAI;gBACA,MAAM,gBAAgB,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAA;aAC7D;YAAC,OAAO,KAAK,EAAE;gBACZ,aAAa;gBACb,IAAA,aAAM,EAAE,KAAe,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAA;gBAC5D,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAA;gBACzC,gBAAgB,CAAC,OAAO,EAAE,CAAA;aAC7B;QACL,CAAC,CAAC,CAAA;QACF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cA6BM;IACV,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACvC,UAAU,CAAC,KAAK,IAAI,EAAE;YAClB,gDAAgD;YAChD,MAAM,CAAC,oCAAoC,CAAC,OAAO,CAC/C,OAAO,CAAC,OAAO,CAAC;gBACZ,oBAAoB,EAAE,SAAS;gBAC/B,GAAG,eAAe;aACrB,CAAC,CACL,CAAA;QACL,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACrC,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,mBAAmB,EAAE,eAAe,EAAE,CAAC,CAAA;YAC9E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAA2B,CAAA;YACnE,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAA;YAEhE,IAAA,aAAM,EAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,8BAAqB,CAAC,sBAAsB,CAAC,CAAA;QAC9F,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,CAAC,oCAAoC,CAAC,OAAO,CAC/C,OAAO,CAAC,OAAO,CAAC;gBACZ,oBAAoB,EAAE,WAAW;gBACjC,GAAG,eAAe;aACrB,CAAC,CACL,CAAA;YAED,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,mBAAmB,EAAE,eAAe,EAAE,CAAC,CAAA;YAC9E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAA2B,CAAA;YACnE,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAA;YAEhE,IAAA,aAAM,EAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,8BAAqB,CAAC,gBAAgB,CAAC,CAAA;QACxF,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IACF,MAAM,yBAAyB,GAAG;QAC9B,cAAc,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;KACjE,CAAA;IAED,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,eAAe,GAAG,IAAI,8CAAe,EAAE,CAAA;YAC7C,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,kBAAkB,CAAC,yBAAyB,CAAC,CAAA;YAClF,IAAA,aAAM,EAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,gDAAsB,CAAC,CAAA;QAC3D,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,MAAM,GAAG,MAAM,IAAA,oDAAqB,EAAC,yBAAyB,CAAC,CAAA;YACrE,IAAA,aAAM,EAAC,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,gDAAsB,CAAC,CAAA;QAC3D,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACnC,UAAU,CAAC,KAAK,IAAI,EAAE;YAClB,mDAAmD;YACnD,MAAM,CAAC,mCAAmC,CAAC,OAAO,CAC9C,OAAO,CAAC,OAAO,CAAC;gBACZ,iBAAiB,EAAE;oBACf,KAAK,EAAE,eAAe;oBACtB,MAAM,EAAE,WAAW;oBACnB,GAAG,eAAe;iBACrB;gBACD,GAAG,eAAe;aACrB,CAAC,CACL,CAAA;QACL,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;YAClC,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,mBAAmB,EAAE,eAAe,EAAE,CAAC,CAAA;YAC9E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAwB,CAAA;YAChE,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAA;YAE7D,IAAA,aAAM,EAAC,GAAG,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,WAAW,CAAC,CAAA;QAC9D,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,uCAAuC,EAAE,GAAG,EAAE;QACnD,UAAU,CAAC,KAAK,IAAI,EAAE;YAClB,mDAAmD;YACnD,MAAM,CAAC,mCAAmC,CAAC,OAAO,CAC9C,OAAO,CAAC,OAAO,CAAC;gBACZ,iBAAiB,EAAE;oBACf,KAAK,EAAE,eAAe;oBACtB,MAAM,EAAE,QAAQ;oBAChB,GAAG,eAAe;iBACrB;gBACD,GAAG,eAAe;aACrB,CAAC,CACL,CAAA;QACL,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,mBAAmB,EAAE,eAAe,EAAE,CAAC,CAAA;YAC9E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAwB,CAAA;YAChE,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAA;YAE7D,IAAA,aAAM,EAAC,GAAG,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;QAC3D,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACrC,UAAU,CAAC,KAAK,IAAI,EAAE;YAClB,oDAAoD;YAEpD,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC;gBAClC,kBAAkB,EAAE;oBAChB,mBAAmB,EAAE;wBACjB;4BACI,EAAE,EAAE,GAAG;4BACP,IAAI,EAAE,gBAAgB;4BACtB,WAAW,EAAE,uBAAuB;4BACpC,MAAM,EAAE,WAAW;4BACnB,eAAe,EAAE;gCACb;oCACI,IAAI,EAAE,qCAAqC;oCAC3C,MAAM,EAAE,WAAW;oCACnB,WAAW,EAAE,4CAA4C;oCACzD,SAAS,EAAE,0BAA0B;oCACrC,OAAO,EAAE,0BAA0B;iCACtC;6BACJ;4BACD,SAAS,EAAE,0BAA0B;4BACrC,OAAO,EAAE,0BAA0B;yBACtC;qBACJ;iBACJ;aACJ,CAAC,CAAA;YACF,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,CAAA;YAC3C,MAAM,CAAC,uCAAuC,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAA;QACrF,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,mBAAmB,EAAE,eAAe,EAAE,CAAC,CAAA;YAC9E,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAA4B,CAAA;YACpE,MAAM,GAAG,GAAG,MAAM,gBAAgB,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAA;YAEjE,IAAA,aAAM,EAAC,GAAG,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,WAAW,CAAC,CAAA;YAClF,IAAA,aAAM,EAAC,GAAG,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAA;YACrF,IAAI,GAAG,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,eAAe,EAAE;gBAC/D,IAAA,aAAM,EAAC,GAAG,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,KAAK,CAClF,qCAAqC,CACxC,CAAA;aACJ;QACL,CAAC,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/netTransform/tests/validation.test.d.ts

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

package/out/language-server/netTransform/tests/validation.test.js

@@ -0,0 +1,80 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const chai_1 = require("chai");
+const validation_1 = require("../validation");
+const sampleStartTransformRequest = {
+    SolutionRootPath: '',
+    SelectedProjectPath: 'test.csproj',
+    ProgramLanguage: '',
+    TargetFramework: '',
+    SolutionConfigPaths: [],
+    ProjectMetadata: [],
+    command: '',
+};
+describe('Test validation functionality', () => {
+    it('should return true when selectedProjectPath is a valid csproj', () => {
+        let mockStartTransformationRequest = sampleStartTransformRequest;
+        mockStartTransformationRequest.SelectedProjectPath = 'test.csproj';
+        (0, chai_1.expect)((0, validation_1.isProject)(mockStartTransformationRequest)).to.equal(true);
+    });
+    it('should return false when selectedProjectPath is not a valid csproj', () => {
+        let mockStartTransformationRequest = sampleStartTransformRequest;
+        mockStartTransformationRequest.SelectedProjectPath = 'test.sln';
+        (0, chai_1.expect)((0, validation_1.isProject)(mockStartTransformationRequest)).to.equal(false);
+    });
+    it('should return true when selectedProjectPath is a valid sln', () => {
+        let mockStartTransformationRequest = sampleStartTransformRequest;
+        mockStartTransformationRequest.SelectedProjectPath = 'test.sln';
+        (0, chai_1.expect)((0, validation_1.isSolution)(mockStartTransformationRequest)).to.equal(true);
+    });
+    it('should return false when selectedProjectPath is not a valid sln', () => {
+        let mockStartTransformationRequest = sampleStartTransformRequest;
+        mockStartTransformationRequest.SelectedProjectPath = 'test.csproj';
+        (0, chai_1.expect)((0, validation_1.isSolution)(mockStartTransformationRequest)).to.equal(false);
+    });
+    it('should return true when project is a supported type', () => {
+        let mockStartTransformationRequest = sampleStartTransformRequest;
+        const mockProjectMeta = {
+            Name: '',
+            ProjectTargetFramework: '',
+            ProjectPath: 'test.csproj',
+            SourceCodeFilePaths: [],
+            ProjectLanguage: '',
+            ProjectType: 'AspNetCoreMvc',
+            ExternalReferences: [],
+        };
+        mockStartTransformationRequest.ProjectMetadata.push(mockProjectMeta);
+        (0, chai_1.expect)((0, validation_1.validateProject)(mockStartTransformationRequest)).to.equal(true);
+    });
+    it('should return false when project is not a supported type', () => {
+        let mockStartTransformationRequest = sampleStartTransformRequest;
+        const mockProjectMeta = {
+            Name: '',
+            ProjectTargetFramework: '',
+            ProjectPath: 'test.csproj',
+            SourceCodeFilePaths: [],
+            ProjectLanguage: '',
+            ProjectType: 'not supported',
+            ExternalReferences: [],
+        };
+        mockStartTransformationRequest.ProjectMetadata = [];
+        mockStartTransformationRequest.ProjectMetadata.push(mockProjectMeta);
+        (0, chai_1.expect)((0, validation_1.validateProject)(mockStartTransformationRequest)).to.equal(false);
+    });
+    it('should return false when there is no project path that is the same as the selected project path', () => {
+        let mockStartTransformationRequest = sampleStartTransformRequest;
+        const mockProjectMeta = {
+            Name: '',
+            ProjectTargetFramework: '',
+            ProjectPath: 'different.csproj',
+            SourceCodeFilePaths: [],
+            ProjectLanguage: '',
+            ProjectType: 'AspNetCoreMvc',
+            ExternalReferences: [],
+        };
+        mockStartTransformationRequest.ProjectMetadata = [];
+        mockStartTransformationRequest.ProjectMetadata.push(mockProjectMeta);
+        (0, chai_1.expect)((0, validation_1.validateProject)(mockStartTransformationRequest)).to.equal(false);
+    });
+});
+//# sourceMappingURL=validation.test.js.map

package/out/language-server/netTransform/tests/validation.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"validation.test.js","sourceRoot":"","sources":["../../../../src/language-server/netTransform/tests/validation.test.ts"],"names":[],"mappings":";;AAAA,+BAA6B;AAE7B,8CAAsE;AAItE,MAAM,2BAA2B,GAA0B;IACvD,gBAAgB,EAAE,EAAE;IACpB,mBAAmB,EAAE,aAAa;IAClC,eAAe,EAAE,EAAE;IACnB,eAAe,EAAE,EAAE;IACnB,mBAAmB,EAAE,EAAE;IACvB,eAAe,EAAE,EAAE;IACnB,OAAO,EAAE,EAAE;CACd,CAAA;AAED,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;IAC3C,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACrE,IAAI,8BAA8B,GAA0B,2BAA2B,CAAA;QACvF,8BAA8B,CAAC,mBAAmB,GAAG,aAAa,CAAA;QAClE,IAAA,aAAM,EAAC,IAAA,sBAAS,EAAC,8BAA8B,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;IACpE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oEAAoE,EAAE,GAAG,EAAE;QAC1E,IAAI,8BAA8B,GAA0B,2BAA2B,CAAA;QACvF,8BAA8B,CAAC,mBAAmB,GAAG,UAAU,CAAA;QAC/D,IAAA,aAAM,EAAC,IAAA,sBAAS,EAAC,8BAA8B,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IACrE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QAClE,IAAI,8BAA8B,GAA0B,2BAA2B,CAAA;QACvF,8BAA8B,CAAC,mBAAmB,GAAG,UAAU,CAAA;QAC/D,IAAA,aAAM,EAAC,IAAA,uBAAU,EAAC,8BAA8B,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;IACrE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QACvE,IAAI,8BAA8B,GAA0B,2BAA2B,CAAA;QACvF,8BAA8B,CAAC,mBAAmB,GAAG,aAAa,CAAA;QAClE,IAAA,aAAM,EAAC,IAAA,uBAAU,EAAC,8BAA8B,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IACtE,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC3D,IAAI,8BAA8B,GAA0B,2BAA2B,CAAA;QACvF,MAAM,eAAe,GAAG;YACpB,IAAI,EAAE,EAAE;YACR,sBAAsB,EAAE,EAAE;YAC1B,WAAW,EAAE,aAAa;YAC1B,mBAAmB,EAAE,EAAE;YACvB,eAAe,EAAE,EAAE;YACnB,WAAW,EAAE,eAAe;YAC5B,kBAAkB,EAAE,EAAE;SACzB,CAAA;QACD,8BAA8B,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QAEpE,IAAA,aAAM,EAAC,IAAA,4BAAe,EAAC,8BAA8B,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;IAC1E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;QAChE,IAAI,8BAA8B,GAA0B,2BAA2B,CAAA;QACvF,MAAM,eAAe,GAAG;YACpB,IAAI,EAAE,EAAE;YACR,sBAAsB,EAAE,EAAE;YAC1B,WAAW,EAAE,aAAa;YAC1B,mBAAmB,EAAE,EAAE;YACvB,eAAe,EAAE,EAAE;YACnB,WAAW,EAAE,eAAe;YAC5B,kBAAkB,EAAE,EAAE;SACzB,CAAA;QACD,8BAA8B,CAAC,eAAe,GAAG,EAAE,CAAA;QACnD,8BAA8B,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QAEpE,IAAA,aAAM,EAAC,IAAA,4BAAe,EAAC,8BAA8B,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IAC3E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iGAAiG,EAAE,GAAG,EAAE;QACvG,IAAI,8BAA8B,GAA0B,2BAA2B,CAAA;QACvF,MAAM,eAAe,GAAG;YACpB,IAAI,EAAE,EAAE;YACR,sBAAsB,EAAE,EAAE;YAC1B,WAAW,EAAE,kBAAkB;YAC/B,mBAAmB,EAAE,EAAE;YACvB,eAAe,EAAE,EAAE;YACnB,WAAW,EAAE,eAAe;YAC5B,kBAAkB,EAAE,EAAE;SACzB,CAAA;QACD,8BAA8B,CAAC,eAAe,GAAG,EAAE,CAAA;QACnD,8BAA8B,CAAC,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,CAAA;QAEpE,IAAA,aAAM,EAAC,IAAA,4BAAe,EAAC,8BAA8B,CAAC,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IAC3E,CAAC,CAAC,CAAA;AACN,CAAC,CAAC,CAAA"}

package/out/language-server/netTransform/transformHandler.d.ts

@@ -0,0 +1,27 @@
+import { CodeWhispererStreaming } from '@amzn/codewhisperer-streaming';
+import { Logging, Workspace } from '@aws/language-server-runtimes/server-interface';
+import { CreateUploadUrlResponse } from '../../client/token/codewhispererbearertokenclient';
+import { CodeWhispererServiceToken } from '../codeWhispererService';
+import { ArtifactManager } from './artifactManager';
+import { CancelTransformRequest, CancelTransformResponse, DownloadArtifactsResponse, GetTransformPlanRequest, GetTransformPlanResponse, GetTransformRequest, GetTransformResponse, StartTransformRequest, StartTransformResponse } from './models';
+export declare class TransformHandler {
+    private client;
+    private workspace;
+    private logging;
+    constructor(client: CodeWhispererServiceToken, workspace: Workspace, logging: Logging);
+    startTransformation(userInputrequest: StartTransformRequest): Promise<StartTransformResponse>;
+    preTransformationUploadCode(payloadFilePath: string): Promise<string>;
+    uploadPayloadAsync(payloadFileName: string): Promise<string>;
+    zipCodeAsync(request: StartTransformRequest, artifactManager: ArtifactManager): Promise<string>;
+    uploadArtifactToS3Async(fileName: string, resp: CreateUploadUrlResponse, sha256: string): Promise<void>;
+    getHeadersObj(sha256: string, kmsKeyArn: string | undefined): {};
+    getTransformation(request: GetTransformRequest): Promise<GetTransformResponse>;
+    getTransformationPlan(request: GetTransformPlanRequest): Promise<GetTransformPlanResponse>;
+    cancelTransformation(request: CancelTransformRequest): Promise<CancelTransformResponse>;
+    sleep(duration?: number): Promise<void>;
+    pollTransformation(request: GetTransformRequest, validExitStatus: string[], failureStates: string[]): Promise<GetTransformResponse>;
+    downloadExportResultArchive(cwStreamingClient: CodeWhispererStreaming, exportId: string, saveToDir: string): Promise<DownloadArtifactsResponse>;
+    archivePathGenerator(exportId: string, buffer: Uint8Array[], saveToDir: string): Promise<string>;
+    directoryExists(directoryPath: any): Promise<void>;
+    getWorkspacePath(solutionRootPath: string): string;
+}