@aigne/doc-smith 0.8.11-beta.5 → 0.8.11-beta.7

package/tests/agents/generate/document-structure-tools/move-document.test.mjs

@@ -158,7 +158,7 @@ describe("move-document", () => {
     expect(result.documentStructure).toEqual(baseDocumentStructure);
     expect(result.originalDocument).toBeUndefined();
     expect(result.updatedDocument).toBeUndefined();
-    expect(consoleSpy).toHaveBeenCalledWith("Error: Cannot move document - path: Required");
+    expect(consoleSpy).toHaveBeenCalledWith("⚠️  Cannot move document: path: Required");
   });
 
   test("should return error when path is empty string", async () => {
@@ -169,7 +169,7 @@ describe("move-document", () => {
     });
 
     expect(result.documentStructure).toEqual(baseDocumentStructure);
-    expect(consoleSpy).toHaveBeenCalledWith("Error: Cannot move document - path: Path is required");
+    expect(consoleSpy).toHaveBeenCalledWith("⚠️  Cannot move document: path: Path is required");
   });
 
   test("should return error when document does not exist", async () => {
@@ -181,7 +181,7 @@ describe("move-document", () => {
 
     expect(result.documentStructure).toEqual(baseDocumentStructure);
     expect(consoleSpy).toHaveBeenCalledWith(
-      "Error: Cannot move document - Document '/nonexistent-document' does not exist. Please select an existing document to move.",
+      "⚠️  Cannot move document: Document '/nonexistent-document' does not exist. Please select an existing document to move.",
     );
   });
 
@@ -194,7 +194,7 @@ describe("move-document", () => {
 
     expect(result.documentStructure).toEqual(baseDocumentStructure);
     expect(consoleSpy).toHaveBeenCalledWith(
-      "Error: Cannot move document - Target parent document '/nonexistent-parent' does not exist. Please select an existing parent document.",
+      "⚠️  Cannot move document: Target parent document '/nonexistent-parent' does not exist. Please select an existing parent document.",
     );
   });
 
@@ -208,7 +208,7 @@ describe("move-document", () => {
 
     expect(result.documentStructure).toEqual(baseDocumentStructure);
     expect(consoleSpy).toHaveBeenCalledWith(
-      "Error: Cannot move document - Moving '/api' under '/api/auth' would create an invalid hierarchy. Please select a parent that is not nested under the document being moved.",
+      "⚠️  Cannot move document: Moving '/api' under '/api/auth' would create an invalid hierarchy. Please select a parent that is not nested under the document being moved.",
     );
   });
 
@@ -221,7 +221,7 @@ describe("move-document", () => {
 
     expect(result.documentStructure).toEqual(baseDocumentStructure);
     expect(consoleSpy).toHaveBeenCalledWith(
-      "Error: Cannot move document - Moving '/api' under '/api/auth/oauth' would create an invalid hierarchy. Please select a parent that is not nested under the document being moved.",
+      "⚠️  Cannot move document: Moving '/api' under '/api/auth/oauth' would create an invalid hierarchy. Please select a parent that is not nested under the document being moved.",
     );
   });
 
@@ -278,7 +278,7 @@ describe("move-document", () => {
 
     expect(result.documentStructure).toEqual(complexStructure);
     expect(consoleSpy).toHaveBeenCalledWith(
-      "Error: Cannot move document - Moving '/api' under '/api/auth/oauth/scopes' would create an invalid hierarchy. Please select a parent that is not nested under the document being moved.",
+      "⚠️  Cannot move document: Moving '/api' under '/api/auth/oauth/scopes' would create an invalid hierarchy. Please select a parent that is not nested under the document being moved.",
     );
   });
 
@@ -326,7 +326,7 @@ describe("move-document", () => {
     expect(updatedDoc.parentId).not.toBe(originalDoc.parentId);
   });
 
-  test("should handle empty document structure", async () => {
+  test("should handle empty documentation structure", async () => {
     const result = await moveDocument({
       documentStructure: [],
       path: "/any-path",
@@ -335,11 +335,11 @@ describe("move-document", () => {
 
     expect(result.documentStructure).toEqual([]);
     expect(consoleSpy).toHaveBeenCalledWith(
-      "Error: Cannot move document - Document '/any-path' does not exist. Please select an existing document to move.",
+      "⚠️  Cannot move document: Document '/any-path' does not exist. Please select an existing document to move.",
     );
   });
 
-  test("should handle single document structure", async () => {
+  test("should handle single documentation structure", async () => {
     const singleDocStructure = [
       {
         title: "Only Document",
@@ -358,12 +358,12 @@ describe("move-document", () => {
 
     expect(result.documentStructure).toEqual(singleDocStructure);
     expect(consoleSpy).toHaveBeenCalledWith(
-      "Error: Cannot move document - Target parent document '/nonexistent' does not exist. Please select an existing parent document.",
+      "⚠️  Cannot move document: Target parent document '/nonexistent' does not exist. Please select an existing parent document.",
     );
   });
 
   // DATA INTEGRITY TESTS
-  test("should not modify original document structure", async () => {
+  test("should not modify original documentation structure", async () => {
     const originalStructure = [...baseDocumentStructure];
 
     await moveDocument({

package/tests/agents/generate/document-structure-tools/update-document.test.mjs

@@ -319,7 +319,7 @@ describe("update-document", () => {
     expect(consoleSpy).not.toHaveBeenCalled();
   });
 
-  test("should handle empty document structure", async () => {
+  test("should handle empty documentation structure", async () => {
     const result = await updateDocument({
       documentStructure: [],
       path: "/any-path",
@@ -332,7 +332,7 @@ describe("update-document", () => {
     );
   });
 
-  test("should handle single document structure", async () => {
+  test("should handle single documentation structure", async () => {
     const singleDocStructure = [
       {
         title: "Only Document",
@@ -377,7 +377,7 @@ describe("update-document", () => {
   });
 
   // DATA INTEGRITY TESTS
-  test("should not modify original document structure", async () => {
+  test("should not modify original documentation structure", async () => {
     const originalStructure = [...baseDocumentStructure];
 
     await updateDocument({

package/tests/agents/generate/user-review-document-structure.test.mjs

@@ -49,6 +49,9 @@ describe("user-review-document-structure", () => {
         invoke: mock(async () => ({
           documentStructure: documentStructure,
         })),
+        userContext: {
+          currentStructure: [],
+        },
       },
     };
 
@@ -71,13 +74,13 @@ describe("user-review-document-structure", () => {
     consoleSpy?.mockRestore();
   });
 
-  test("should return original structure when no document structure provided", async () => {
+  test("should return original structure when no documentation structure provided", async () => {
     const result = await userReviewDocumentStructure({}, mockOptions);
 
     expect(result).toBeDefined();
     expect(result.documentStructure).toBeUndefined();
     expect(mockOptions.prompts.select).not.toHaveBeenCalled();
-    expect(consoleSpy).toHaveBeenCalledWith("No document structure was generated to review.");
+    expect(consoleSpy).toHaveBeenCalledWith("No documentation structure was generated to review.");
   });
 
   test("should return original structure when empty array provided", async () => {
@@ -86,7 +89,7 @@ describe("user-review-document-structure", () => {
     expect(result).toBeDefined();
     expect(result.documentStructure).toEqual([]);
     expect(mockOptions.prompts.select).not.toHaveBeenCalled();
-    expect(consoleSpy).toHaveBeenCalledWith("No document structure was generated to review.");
+    expect(consoleSpy).toHaveBeenCalledWith("No documentation structure was generated to review.");
   });
 
   test("should return original structure when user chooses not to review", async () => {
@@ -110,7 +113,9 @@ describe("user-review-document-structure", () => {
     expect(result.documentStructure).toEqual(documentStructure);
     expect(mockOptions.prompts.select).toHaveBeenCalled();
     expect(mockOptions.prompts.input).toHaveBeenCalled();
-    expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining("Current Document Structure"));
+    expect(consoleSpy).toHaveBeenCalledWith(
+      expect.stringContaining("Current Documentation Structure"),
+    );
   });
 
   test("should process user feedback and call refineDocumentStructure agent", async () => {
@@ -130,9 +135,13 @@ describe("user-review-document-structure", () => {
       .mockImplementationOnce(async () => feedback)
       .mockImplementationOnce(async () => ""); // Exit loop on second call
 
-    mockOptions.context.invoke.mockImplementation(async () => ({
-      documentStructure: refinedStructure,
-    }));
+    mockOptions.context.invoke.mockImplementation(async () => {
+      // Simulate the agent updating the shared context
+      mockOptions.context.userContext.currentStructure = refinedStructure;
+      return {
+        documentStructure: refinedStructure,
+      };
+    });
 
     const result = await userReviewDocumentStructure({ documentStructure }, mockOptions);
 
@@ -188,7 +197,7 @@ describe("user-review-document-structure", () => {
 
     expect(result.documentStructure).toEqual(documentStructure);
     expect(consoleSpy).toHaveBeenCalledWith(
-      "Unable to process your feedback - the document structure update feature is unavailable.",
+      "Unable to process your feedback - the documentation structure update feature is unavailable.",
     );
   });
 
@@ -240,11 +249,25 @@ describe("user-review-document-structure", () => {
       .mockImplementationOnce(async () => secondFeedback)
       .mockImplementationOnce(async () => ""); // Exit loop
 
-    mockOptions.context.invoke
-      .mockImplementationOnce(async () => ({ documentStructure: firstRefinedStructure })) // refineDocumentStructure 1st call
-      .mockImplementationOnce(async () => ({})) // checkFeedbackRefiner 1st call
-      .mockImplementationOnce(async () => ({ documentStructure: finalRefinedStructure })) // refineDocumentStructure 2nd call
-      .mockImplementationOnce(async () => ({})); // checkFeedbackRefiner 2nd call
+    let invokeCount = 0;
+    mockOptions.context.invoke.mockImplementation(async () => {
+      invokeCount++;
+      if (invokeCount === 1) {
+        // refineDocumentStructure 1st call
+        mockOptions.context.userContext.currentStructure = firstRefinedStructure;
+        return { documentStructure: firstRefinedStructure };
+      } else if (invokeCount === 2) {
+        // checkFeedbackRefiner 1st call
+        return {};
+      } else if (invokeCount === 3) {
+        // refineDocumentStructure 2nd call
+        mockOptions.context.userContext.currentStructure = finalRefinedStructure;
+        return { documentStructure: finalRefinedStructure };
+      } else {
+        // checkFeedbackRefiner 2nd call
+        return {};
+      }
+    });
 
     const result = await userReviewDocumentStructure({ documentStructure }, mockOptions);
 

package/tests/agents/update/check-document.test.mjs

@@ -177,7 +177,7 @@ describe("check-document", () => {
     expect(mockOptions.context.invoke).not.toHaveBeenCalled();
   });
 
-  test("should handle missing original node in document structure", async () => {
+  test("should handle missing original node in documentation structure", async () => {
     accessSpy.mockResolvedValue();
     checkDetailResultSpy.mockResolvedValue({ isApproved: true });