src/EditorFeatures/TestUtilities/SpellCheck/AbstractSpellCheckSpanTests.cs - main - githubmirror/roslyn

src/EditorFeatures/TestUtilities/SpellCheck/AbstractSpellCheckSpanTests.cs

53 строки2 KB

Tomáš Matoušek

Move TestWorkspace to Workspace test layer, add EditorTestWorkspace to EditorFeatures layer (#71407)

09 янв 2024, 18:51

09 янв 2024, 18:513157d16

// 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.Collections.Generic;
using System.Collections.Immutable;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.Shared.Extensions;
using Microsoft.CodeAnalysis.SpellCheck;
using Microsoft.CodeAnalysis.Test.Utilities;
using Microsoft.CodeAnalysis.Text;
using Roslyn.Utilities;
using Xunit;

namespace Microsoft.CodeAnalysis.Editor.UnitTests.SpellCheck;

public abstract class AbstractSpellCheckSpanTests
{
    protected abstract EditorTestWorkspace CreateWorkspace(string content);

    protected async Task TestAsync(string content)
    {
        using var workspace = CreateWorkspace(content);
        var annotations = workspace.Projects.Single().Documents.Single().AnnotatedSpans;

        var document = workspace.CurrentSolution.Projects.Single().Documents.Single();
        var service = document.GetRequiredLanguageService<ISpellCheckSpanService>();

        var actual = await service.GetSpansAsync(document, CancellationToken.None);
        var expected = Flatten(annotations);

        actual = actual.Sort((s1, s2) => s1.TextSpan.Start - s2.TextSpan.Start);
        expected = expected.Sort((s1, s2) => s1.TextSpan.Start - s2.TextSpan.Start);

        Assert.Equal<SpellCheckSpan>(expected, actual);
    }

    private static ImmutableArray<SpellCheckSpan> Flatten(IDictionary<string, ImmutableArray<TextSpan>> annotations)
        => annotations.SelectManyAsArray(kvp => kvp.Value.Select(span => new SpellCheckSpan(span, ConvertKind(kvp.Key))));

    private static SpellCheckKind ConvertKind(string key)
    {
        return key switch
        {
            "Comment" => SpellCheckKind.Comment,
            "Identifier" => SpellCheckKind.Identifier,
            "String" => SpellCheckKind.String,
            _ => throw ExceptionUtilities.UnexpectedValue(key),
        };
    }
}