src/EditorFeatures/CSharpTest/EncapsulateField/EncapsulateFieldTestState.cs - main - githubmirror/roslyn

src/EditorFeatures/CSharpTest/EncapsulateField/EncapsulateFieldTestState.cs
79 строк3 KB
Cyrus Najmabadi
Make sealed
26 мар 2025, 22:21
26 мар 2025, 22:21d99b771
// 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;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.CSharp.CodeStyle;
using Microsoft.CodeAnalysis.Editor.CSharp.EncapsulateField;
using Microsoft.CodeAnalysis.Editor.UnitTests;
using Microsoft.CodeAnalysis.Editor.UnitTests.CodeActions;
using Microsoft.CodeAnalysis.Editor.UnitTests.Extensions;
using Microsoft.CodeAnalysis.Editor.UnitTests.Utilities;
using Microsoft.CodeAnalysis.Notification;
using Microsoft.CodeAnalysis.Shared.TestHooks;
using Microsoft.CodeAnalysis.Test.Utilities;
using Microsoft.VisualStudio.Text.Editor.Commanding.Commands;
using Xunit;

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

internal sealed class EncapsulateFieldTestState : IDisposable
{
    private readonly EditorTestHostDocument _testDocument;
    public EditorTestWorkspace Workspace { get; }
    public Document TargetDocument { get; }
    public string NotificationMessage { get; private set; }

    public EncapsulateFieldTestState(EditorTestWorkspace workspace)
    {
        Workspace = workspace;
        _testDocument = Workspace.Documents.Single(d => d.CursorPosition.HasValue || d.SelectedSpans.Any());
        TargetDocument = Workspace.CurrentSolution.GetDocument(_testDocument.Id);

        var notificationService = Workspace.Services.GetService<INotificationService>() as INotificationServiceCallback;
        var callback = new Action<string, string, NotificationSeverity>((message, title, severity) => NotificationMessage = message);
        notificationService.NotificationCallback = callback;
    }

    public static EncapsulateFieldTestState Create(string markup)
    {
        var workspace = EditorTestWorkspace.CreateCSharp(markup, composition: EditorTestCompositions.EditorFeatures);

        workspace.SetAnalyzerFallbackOptions(new OptionsCollection(LanguageNames.CSharp)
        {
            { CSharpCodeStyleOptions.PreferExpressionBodiedAccessors, CSharpCodeStyleOptions.NeverWithSilentEnforcement },
            { CSharpCodeStyleOptions.PreferExpressionBodiedProperties, CSharpCodeStyleOptions.NeverWithSilentEnforcement }
        });

        return new EncapsulateFieldTestState(workspace);
    }

    public async Task EncapsulateAsync()
    {
        var args = new EncapsulateFieldCommandArgs(_testDocument.GetTextView(), _testDocument.GetTextBuffer());
        var commandHandler = Workspace.ExportProvider.GetCommandHandler<EncapsulateFieldCommandHandler>(PredefinedCommandHandlerNames.EncapsulateField, ContentTypeNames.CSharpContentType);
        var provider = Workspace.ExportProvider.GetExportedValue<IAsynchronousOperationListenerProvider>();
        var waiter = (IAsynchronousOperationWaiter)provider.GetListener(FeatureAttribute.EncapsulateField);
        commandHandler.ExecuteCommand(args, TestCommandExecutionContext.Create());
        await waiter.ExpeditedWaitAsync();
    }

    public void Dispose()
        => Workspace?.Dispose();

    public async Task AssertEncapsulateAsAsync(string expected)
    {
        await EncapsulateAsync();
        Assert.Equal(expected, _testDocument.GetTextBuffer().CurrentSnapshot.GetText().ToString());
    }

    public async Task AssertErrorAsync()
    {
        await EncapsulateAsync();
        Assert.NotNull(NotificationMessage);
    }
}