// 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());
        }
    }
}