// 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.Diagnostics.CodeAnalysis;
using System.Threading.Tasks;
using Microsoft.CodeAnalysis.Editor.UnitTests;
using Microsoft.CodeAnalysis.Editor.UnitTests.Classification;
using Microsoft.CodeAnalysis.Remote.Testing;
using Microsoft.CodeAnalysis.Test.Utilities;
 
namespace Microsoft.CodeAnalysis.Editor.CSharp.UnitTests.Classification;
 
public abstract class AbstractCSharpClassifierTests : AbstractClassifierTests
{
    protected static EditorTestWorkspace CreateWorkspace(
        string code, ParseOptions? options, TestHost testHost)
    {
        var composition = EditorTestCompositions.EditorFeatures.WithTestHostParts(testHost);
        return EditorTestWorkspace.CreateCSharp(code, parseOptions: options, composition: composition, isMarkup: false);
    }
 
    protected new Task TestAsync(
        [StringSyntax(PredefinedEmbeddedLanguageNames.CSharpTest)] string code,
        TestHost testHost,
        params FormattedClassification[] expected)
    {
        return base.TestAsync(code, testHost, expected);
    }
 
    protected new Task TestAsync(
        [StringSyntax(PredefinedEmbeddedLanguageNames.CSharpTest)] string code,
        TestHost testHost,
        ParseOptions? parseOptions,
        params FormattedClassification[] expected)
    {
        return base.TestAsync(code, testHost, parseOptions, expected);
    }
 
    protected override async Task DefaultTestAsync(string code, string allCode, TestHost testHost, FormattedClassification[] expected)
    {
        await TestAsync(code, allCode, testHost, parseOptions: null, expected);
        await TestAsync(code, allCode, testHost, parseOptions: Options.Script, expected);
    }
 
    protected override string WrapInClass(string className, string code)
        => $$"""
           class {{className}} {
               {{code}}
           }
           """;
 
    protected override string WrapInExpression(string code)
        => $$"""
           class C {
               void M() {
                   var q =
                       {{code}}
               }
           }
           """;
 
    protected override string WrapInMethod(string className, string methodName, string code)
        => $$"""
           class {{className}} {
               void {{methodName}}() {
                   {{code}}
               }
           }
           """;
 
    protected override string WrapInNamespace(string code)
        => $$"""
           namespace N {
               {{code}}
           }
           """;
}