src/EditorFeatures/TestUtilities/Formatting/AbstractNewDocumentFormattingServiceTests.cs - main - githubmirror/roslyn

src/EditorFeatures/TestUtilities/Formatting/AbstractNewDocumentFormattingServiceTests.cs

43 строки2 KB

Cyrus Najmabadi

Use file scoped namespaces. (#77854)

27 мар 2025, 09:55

27 мар 2025, 09:5598d41b8

// 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.

using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.CodeCleanup;
using Microsoft.CodeAnalysis.Editor.UnitTests.CodeActions;
using Microsoft.CodeAnalysis.Formatting;
using Microsoft.CodeAnalysis.Options;
using Microsoft.CodeAnalysis.Shared.Extensions;
using Roslyn.Test.Utilities;

namespace Microsoft.CodeAnalysis.Test.Utilities.Formatting;

[UseExportProvider]
public abstract class AbstractNewDocumentFormattingServiceTests
{
    protected abstract string Language { get; }
    protected abstract EditorTestWorkspace CreateTestWorkspace(string testCode, ParseOptions? parseOptions);

    internal async Task TestAsync(string testCode, string expected, OptionsCollection? options = null, ParseOptions? parseOptions = null)
    {
        using var workspace = CreateTestWorkspace(testCode, parseOptions);
        options?.SetGlobalOptions(workspace.GlobalOptions);

        var solution = workspace.CurrentSolution;

        var document = workspace.CurrentSolution.Projects.First().Documents.First();
        var languageServices = document.Project.Services;

        var cleanupOptions =
            options?.GetCodeCleanupOptions(languageServices, allowImportsInHiddenRegions: false) ??
            await document.GetCodeCleanupOptionsAsync(CancellationToken.None);

        var formattingService = document.GetRequiredLanguageService<INewDocumentFormattingService>();
        var formattedDocument = await formattingService.FormatNewDocumentAsync(document, hintDocument: null, cleanupOptions, CancellationToken.None);

        var actual = await formattedDocument.GetTextAsync();
        AssertEx.EqualOrDiff(expected, actual.ToString());
    }
}