Function TestReadDirMaxEntries

pkg/aiusechat/tools_readdir_test.go:117–171

(t *testing.T)

}

func TestReadDirMaxEntries(t *testing.T) {
	// Create a temporary test directory with many files
	tmpDir, err := os.MkdirTemp("", "readdir_test_max")
	if err != nil {
		t.Fatalf("Failed to create temp dir: %v", err)
	}
	defer os.RemoveAll(tmpDir)

	// Create 10 test files
	for i := 0; i < 10; i++ {
		testFile := filepath.Join(tmpDir, filepath.Base(tmpDir)+string(rune('a'+i))+".txt")
		if err := os.WriteFile(testFile, []byte("test"), 0644); err != nil {
			t.Fatalf("Failed to create test file: %v", err)
		}
	}

	// Test reading with max_entries=5
	maxEntries := 5
	input := map[string]any{
		"path":        tmpDir,
		"max_entries": maxEntries,
	}

	result, err := readDirCallback(input, &uctypes.UIMessageDataToolUse{})
	if err != nil {
		t.Fatalf("readDirCallback failed: %v", err)
	}

	resultMap := result.(map[string]any)
	entryCount := resultMap["entry_count"].(int)
	totalEntries := resultMap["total_entries"].(int)

	if entryCount != maxEntries {
		t.Errorf("Expected %d entries, got %d", maxEntries, entryCount)
	}

	// Verify total_entries reports the original count, not the truncated count
	if totalEntries != 10 {
		t.Errorf("Expected total_entries to be 10, got %d", totalEntries)
	}

	if _, ok := resultMap["truncated"]; !ok {
		t.Error("Expected truncated field to be present")
	}

	// Verify the truncation message includes the correct total
	truncMsg, ok := resultMap["truncated_message"].(string)
	if !ok {
		t.Error("Expected truncated_message to be present")
	}
	expectedMsg := fmt.Sprintf("Directory listing truncated to %d entries (out of %d total)", maxEntries, 10)
	if !strings.Contains(truncMsg, expectedMsg[:len(expectedMsg)-1]) {
		t.Errorf("Expected truncated_message to contain %q, got %q", expectedMsg, truncMsg)
	}
}

func TestReadDirSortBeforeTruncate(t *testing.T) {
	// Create a temporary test directory