fix test

Author: Artmann

src/notebooks/deepnote/deepnoteTreeDataProvider.ts

@@ -131,28 +131,29 @@ export class DeepnoteTreeDataProvider implements TreeDataProvider<DeepnoteTreeIt
     }
 
     public async getChildren(element?: DeepnoteTreeItem): Promise<DeepnoteTreeItem[]> {
-        if (!workspace.workspaceFolders || workspace.workspaceFolders.length === 0) {
+        // If element is provided, we can return children regardless of workspace
+        if (element) {
+            if (element.type === DeepnoteTreeItemType.ProjectFile) {
+                return this.getNotebooksForProject(element);
+            }
+
             return [];
         }
 
-        if (!element) {
-            if (!this.isInitialScanComplete) {
-                if (!this.initialScanPromise) {
-                    this.initialScanPromise = this.performInitialScan();
-                }
+        // For root level, we need workspace folders
+        if (!workspace.workspaceFolders || workspace.workspaceFolders.length === 0) {
+            return [];
+        }
 
-                // Show loading item
-                return [this.createLoadingTreeItem()];
+        if (!this.isInitialScanComplete) {
+            if (!this.initialScanPromise) {
+                this.initialScanPromise = this.performInitialScan();
             }
 
-            return this.getDeepnoteProjectFiles();
-        }
-
-        if (element.type === DeepnoteTreeItemType.ProjectFile) {
-            return this.getNotebooksForProject(element);
+            return [this.createLoadingTreeItem()];
         }
 
-        return [];
+        return this.getDeepnoteProjectFiles();
     }
 
     private createLoadingTreeItem(): DeepnoteTreeItem {