From 6cab1706f0cec27a9fd5a729d3ecbdb23d66510c Mon Sep 17 00:00:00 2001
From: eleanorjboyd <eleanorboyd@microsoft.com>
Date: Fri, 7 Apr 2023 11:59:59 -0700
Subject: [PATCH] add tests for unittest execution adapter

---
 .../testExecutionAdapter.unit.test.ts         | 114 ++++++++++++++++++
 1 file changed, 114 insertions(+)
 create mode 100644 src/test/testing/testController/unittest/testExecutionAdapter.unit.test.ts

diff --git a/src/test/testing/testController/unittest/testExecutionAdapter.unit.test.ts b/src/test/testing/testController/unittest/testExecutionAdapter.unit.test.ts
new file mode 100644
index 000000000000..f8648c8963fa
--- /dev/null
+++ b/src/test/testing/testController/unittest/testExecutionAdapter.unit.test.ts
@@ -0,0 +1,114 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+import * as assert from 'assert';
+import * as path from 'path';
+import { Uri } from 'vscode';
+import { IConfigurationService } from '../../../../client/common/types';
+import { EXTENSION_ROOT_DIR } from '../../../../client/constants';
+import { ITestServer, TestCommandOptions } from '../../../../client/testing/testController/common/types';
+import { UnittestTestExecutionAdapter } from '../../../../client/testing/testController/unittest/testExecutionAdapter';
+
+suite('Unittest test execution adapter', () => {
+    let stubConfigSettings: IConfigurationService;
+
+    setup(() => {
+        stubConfigSettings = ({
+            getSettings: () => ({
+                testing: { unittestArgs: ['-v', '-s', '.', '-p', 'test*'] },
+            }),
+        } as unknown) as IConfigurationService;
+    });
+
+    test('runTests should send the run command to the test server', async () => {
+        let options: TestCommandOptions | undefined;
+
+        const stubTestServer = ({
+            sendCommand(opt: TestCommandOptions): Promise<void> {
+                options = opt;
+                return Promise.resolve();
+            },
+            onDataReceived: () => {
+                // no body
+            },
+            createUUID: () => '123456789',
+        } as unknown) as ITestServer;
+
+        const uri = Uri.file('/foo/bar');
+        const script = path.join(EXTENSION_ROOT_DIR, 'pythonFiles', 'unittestadapter', 'execution.py');
+
+        const adapter = new UnittestTestExecutionAdapter(stubTestServer, stubConfigSettings);
+        adapter.runTests(uri, [], false);
+
+        const expectedOptions: TestCommandOptions = {
+            workspaceFolder: uri,
+            command: { script, args: ['--udiscovery', '-v', '-s', '.', '-p', 'test*'] },
+            cwd: uri.fsPath,
+            uuid: '123456789',
+            debugBool: false,
+            testIds: [],
+        };
+
+        assert.deepStrictEqual(options, expectedOptions);
+    });
+    test("onDataReceivedHandler should parse the data if the cwd from the payload matches the test adapter's cwd", async () => {
+        const stubTestServer = ({
+            sendCommand(): Promise<void> {
+                return Promise.resolve();
+            },
+            onDataReceived: () => {
+                // no body
+            },
+            createUUID: () => '123456789',
+        } as unknown) as ITestServer;
+
+        const uri = Uri.file('/foo/bar');
+        const data = { status: 'success' };
+        const uuid = '123456789';
+
+        const adapter = new UnittestTestExecutionAdapter(stubTestServer, stubConfigSettings);
+
+        // triggers runTests flow which will run onDataReceivedHandler and the
+        // promise resolves into the parsed data.
+        const promise = adapter.runTests(uri, [], false);
+
+        adapter.onDataReceivedHandler({ uuid, data: JSON.stringify(data) });
+
+        const result = await promise;
+
+        assert.deepStrictEqual(result, data);
+    });
+    test("onDataReceivedHandler should ignore the data if the cwd from the payload does not match the test adapter's cwd", async () => {
+        const correctUuid = '123456789';
+        const incorrectUuid = '987654321';
+        const stubTestServer = ({
+            sendCommand(): Promise<void> {
+                return Promise.resolve();
+            },
+            onDataReceived: () => {
+                // no body
+            },
+            createUUID: () => correctUuid,
+        } as unknown) as ITestServer;
+
+        const uri = Uri.file('/foo/bar');
+
+        const adapter = new UnittestTestExecutionAdapter(stubTestServer, stubConfigSettings);
+
+        // triggers runTests flow which will run onDataReceivedHandler and the
+        // promise resolves into the parsed data.
+        const promise = adapter.runTests(uri, [], false);
+
+        const data = { status: 'success' };
+        // will not resolve due to incorrect UUID
+        adapter.onDataReceivedHandler({ uuid: incorrectUuid, data: JSON.stringify(data) });
+
+        const nextData = { status: 'error' };
+        // will resolve and nextData will be returned as result
+        adapter.onDataReceivedHandler({ uuid: correctUuid, data: JSON.stringify(nextData) });
+
+        const result = await promise;
+
+        assert.deepStrictEqual(result, nextData);
+    });
+});