src/EditorFeatures/CSharpTest/CodeActions/PreviewExceptionTests.cs - main - githubmirror/roslyn

src/EditorFeatures/CSharpTest/CodeActions/PreviewExceptionTests.cs
115 строк5 KB
Cyrus Najmabadi
Rename types
12 сен 2025, 12:39
12 сен 2025, 12:39a23d0d6
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

#nullable disable

using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.CodeActions;
using Microsoft.CodeAnalysis.CodeRefactorings;
using Microsoft.CodeAnalysis.Editor.Implementation.Suggestions;
using Microsoft.CodeAnalysis.Editor.Shared.Utilities;
using Microsoft.CodeAnalysis.ErrorReporting;
using Microsoft.CodeAnalysis.Extensions;
using Microsoft.CodeAnalysis.Test.Utilities;
using Roslyn.Test.Utilities;
using Xunit;

namespace Microsoft.CodeAnalysis.Editor.CSharp.UnitTests.CodeRefactorings;

public sealed partial class PreviewTests
{
    [WpfFact]
    public async Task TestExceptionInComputePreview()
    {
        using var workspace = CreateWorkspaceFromOptions("class D {}", TestParameters.Default);

        var errorReportingService = (TestErrorReportingService)workspace.Services.GetRequiredService<IErrorReportingService>();
        var errorReported = false;
        errorReportingService.OnError = message => errorReported = true;

        await GetPreview(workspace, new ErrorCases.ExceptionInCodeAction());
        Assert.True(errorReported);
    }

    [WpfFact]
    public void TestExceptionInDisplayText()
    {
        using var workspace = CreateWorkspaceFromOptions("class D {}", TestParameters.Default);

        var errorReportingService = (TestErrorReportingService)workspace.Services.GetRequiredService<IErrorReportingService>();
        var errorReported = false;
        errorReportingService.OnError = message => errorReported = true;

        DisplayText(workspace, new ErrorCases.ExceptionInCodeAction());
        Assert.True(errorReported);
    }

    [WpfFact]
    public async Task TestExceptionInActionSets()
    {
        using var workspace = CreateWorkspaceFromOptions("class D {}", TestParameters.Default);

        var errorReportingService = (TestErrorReportingService)workspace.Services.GetRequiredService<IErrorReportingService>();
        var errorReported = false;
        errorReportingService.OnError = message => errorReported = true;

        await ActionSets(workspace, new ErrorCases.ExceptionInCodeAction());

        // No exception is thrown in the call to GetActionSetsAsync because the preview is only lazily evaluated.
        Assert.False(errorReported);
    }

    private static async Task GetPreview(EditorTestWorkspace workspace, CodeRefactoringProvider provider)
    {
        var suggestedAction = CreateRefactoringSuggestedAction(workspace, provider, out var extensionManager);
        await suggestedAction.GetPreviewAsync(CancellationToken.None);
        Assert.True(extensionManager.IsDisabled(provider));
        Assert.False(extensionManager.IsIgnored(provider));
    }

    private static void DisplayText(EditorTestWorkspace workspace, CodeRefactoringProvider provider)
    {
        var suggestedAction = CreateRefactoringSuggestedAction(workspace, provider, out var extensionManager);
        _ = suggestedAction.DisplayText;
        Assert.True(extensionManager.IsDisabled(provider));
        Assert.False(extensionManager.IsIgnored(provider));
    }

    private static async Task ActionSets(EditorTestWorkspace workspace, CodeRefactoringProvider provider)
    {
        var suggestedAction = CreateRefactoringSuggestedAction(workspace, provider, out var extensionManager);
        _ = await suggestedAction.GetActionSetsAsync(CancellationToken.None);
        Assert.False(extensionManager.IsDisabled(provider));
        Assert.False(extensionManager.IsIgnored(provider));
    }

    private static EditorSuggestedActionWithNestedFlavors CreateRefactoringSuggestedAction(EditorTestWorkspace workspace, CodeRefactoringProvider provider, out EditorLayerExtensionManager.ExtensionManager extensionManager)
    {
        var codeActions = new List<CodeAction>();
        RefactoringSetup(workspace, provider, codeActions, out extensionManager, out var textBuffer, out var document);
        var suggestedAction = new EditorSuggestedActionWithNestedFlavors(
            workspace.ExportProvider.GetExportedValue<IThreadingContext>(),
            workspace.ExportProvider.GetExportedValue<SuggestedActionsSourceProvider>(),
            document, textBuffer, provider, codeActions.First(), fixAllFlavors: null, diagnostics: []);
        return suggestedAction;
    }

    private static void RefactoringSetup(
        EditorTestWorkspace workspace, CodeRefactoringProvider provider, List<CodeAction> codeActions,
        out EditorLayerExtensionManager.ExtensionManager extensionManager,
        out VisualStudio.Text.ITextBuffer textBuffer,
        out Document document)
    {
        document = GetDocument(workspace);
        textBuffer = workspace.GetTestDocument(document.Id).GetTextBuffer();
        var span = document.GetSyntaxRootAsync().Result.Span;
        var context = new CodeRefactoringContext(document, span, (a) => codeActions.Add(a), CancellationToken.None);
        provider.ComputeRefactoringsAsync(context).Wait();
        var action = codeActions.Single();
        extensionManager = document.Project.Solution.Services.GetService<IExtensionManager>() as EditorLayerExtensionManager.ExtensionManager;
    }
}