Function TestReadDirSortBeforeTruncate

pkg/aiusechat/tools_readdir_test.go:173–236

(t *testing.T)

}

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

	// Create files with names that would sort alphabetically before directories
	// but we want directories to appear first
	for i := 0; i < 5; i++ {
		testFile := filepath.Join(tmpDir, fmt.Sprintf("a_file_%d.txt", i))
		if err := os.WriteFile(testFile, []byte("test"), 0644); err != nil {
			t.Fatalf("Failed to create test file: %v", err)
		}
	}

	// Create directories with names that sort alphabetically after the files
	for i := 0; i < 3; i++ {
		testDir := filepath.Join(tmpDir, fmt.Sprintf("z_dir_%d", i))
		if err := os.Mkdir(testDir, 0755); err != nil {
			t.Fatalf("Failed to create test dir: %v", err)
		}
	}

	// Test with max_entries=5 (less than total of 8)
	// All 3 directories should still appear because they're sorted first
	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)
	entries, ok := resultMap["entries"].([]fileutil.DirEntryOut)
	if !ok {
		t.Fatalf("entries is not a slice of DirEntryOut")
	}

	// Count directories in the result
	dirCount := 0
	for _, entry := range entries {
		if entry.Dir {
			dirCount++
		}
	}

	// All 3 directories should be present because sorting happens before truncation
	if dirCount != 3 {
		t.Errorf("Expected 3 directories in truncated results, got %d", dirCount)
	}

	// First 3 entries should be directories