From 4c96ebc9f34116529cdc56a86bf061f2f49e6a2d Mon Sep 17 00:00:00 2001
From: issacto <issacto606@gmail.com>
Date: Mon, 21 Aug 2023 12:16:12 +0100
Subject: [PATCH] add test tree -3

Signed-off-by: issacto <issacto606@gmail.com>
---
 vs code extension/client/src/extension.ts      | 16 +++++++++-------
 .../client/src/services/TestTree.ts            | 18 ++++++++++++++++--
 2 files changed, 25 insertions(+), 9 deletions(-)

diff --git a/vs code extension/client/src/extension.ts b/vs code extension/client/src/extension.ts
index a8a67b30..424645ac 100644
--- a/vs code extension/client/src/extension.ts	
+++ b/vs code extension/client/src/extension.ts	
@@ -125,13 +125,14 @@ export function activate(context: ExtensionContext) {
 						run.enqueued(test);
 						queue.push({ test, data });
 					} else {
-						if (data instanceof TestFile  ) {
+						if (data instanceof TestFile ) {
 							await data.updateFromDisk(ctrl, test);
 						}
 
 						await discoverTests(gatherTestItems(test));
+						
 					}
-					if(data instanceof TestFile && test.children.size!=0 )
+					if(!(data instanceof TestFile) || !data.isDirectory(test.uri.fsPath))
 					{
 						if (test.uri && !coveredLines.has(test.uri.toString())) {
 						try {
@@ -251,9 +252,10 @@ function createDirItems( controller:vscode.TestController, uri: vscode.Uri){
 	if(!controller.items.get(rootUri)){ 
 		file = controller.createTestItem(rootUri, dirArr[0], tmpUri);
 		controller.items.add(file);
-		data = new TestFile()
-		testData.set(file, data);
+		data = new TestFile();
 		file.canResolveChildren = true;
+		testData.set(file, data);
+		
 	}
 	else{ 
 		file = controller.items.get(rootUri)
@@ -266,17 +268,17 @@ function createDirItems( controller:vscode.TestController, uri: vscode.Uri){
 	var tmpDir = rootUri
 	var tmpFile = null
 	var tmpData = null
-
 	for(var i =1;i<dirArr.length;i++){
+		tmpDir = tmpDir + "/" + dirArr[i]
+		
 		const existing = prevFile.children.get(tmpDir);
 		
 		if(!existing){
 			tmpUri = vscode.Uri.file(tmpDir);
 			tmpFile = controller.createTestItem(tmpDir, dirArr[i], tmpUri);
 			tmpData = new TestFile();
-			testData.set(tmpFile, tmpData);
 			tmpFile.canResolveChildren = true;
-
+			testData.set(tmpFile, tmpData);
 			// add to existing tree structure
 			prevFile.children.add(tmpFile)
 			
diff --git a/vs code extension/client/src/services/TestTree.ts b/vs code extension/client/src/services/TestTree.ts
index 25d4cd32..f1d6cf8b 100644
--- a/vs code extension/client/src/services/TestTree.ts	
+++ b/vs code extension/client/src/services/TestTree.ts	
@@ -33,11 +33,19 @@ export class TestFile {
 
 	public async updateFromDisk(controller: vscode.TestController, item: vscode.TestItem) {
 		try {
-			if(item.children.size==0){
+			if(!this.isDirectory(item.uri.fsPath)){
 				var content= await getContentFromFilesystem(item.uri!, (item.children.size==0));
 				item.error = undefined;
 				this.updateFromContents(controller, content, item, item.children.size);
+				
+			}else{
+				for (const nestedItem of item.children) {
+					const data =  testData.get(nestedItem[1]); 
+					if(data instanceof TestFile){
+						await this.updateFromDisk(controller,nestedItem[1] )
+					}
 			}
+		}
 		} catch (e) {
 			item.error = (e as Error).stack;
 		}
@@ -87,6 +95,12 @@ export class TestFile {
 		// else ascend(0) // finish and assign children for all remaining items
 	}
 
+
+	public isDirectory(fileName:string){
+		if(fileName.includes(".cut")) return false
+		else return true
+	}
+
 	
 
 }
@@ -120,7 +134,7 @@ export class TestCase {
 
 	async run(item: vscode.TestItem, options: vscode.TestRun): Promise<void> {
 		const start = Date.now();
-		await new Promise(resolve => setTimeout(resolve, 1000 + Math.random() * 1000));
+		await new Promise(resolve => setTimeout(resolve, 1 + Math.random() * 1));
 		const duration = Date.now() - start;
 		options.passed(item, duration);