expand type filtering for unit-tests and protofiles

Kong · Oct 17, 2024 · 86ccd88 · 86ccd88
1 parent 2ada790
commit 86ccd88
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/packages/insomnia/src/sync/git/ne-db-client.ts b/packages/insomnia/src/sync/git/ne-db-client.ts
@@ -156,8 +156,21 @@ export class NeDBClient {
       ];
     } else if (type !== null && id === null) {
       const workspace = await db.get(models.workspace.type, this._workspaceId);
+      let typeFilter = [type];
+
       const modelTypesWithinFolders = [models.request.type, models.grpcRequest.type, models.webSocketRequest.type];
-      const typeFilter = modelTypesWithinFolders.includes(type) ? [models.requestGroup.type, type] : [type];
+      if (modelTypesWithinFolders.includes(type)) {
+        typeFilter = [models.requestGroup.type, type];
+      };
+
+      if (type === models.unitTest.type) {
+        typeFilter = [models.unitTestSuite.type, type];
+      }
+
+      if (type === models.protoFile.type) {
+        typeFilter = [models.protoDirectory.type, type];
+      }
+
       const children = await db.withDescendants(workspace, null, typeFilter);
       docs = children.filter(d => d.type === type && !d.isPrivate);
     } else {