// 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.DesignerAttribute;
using Microsoft.CodeAnalysis.Editor.UnitTests.Workspaces;
using Microsoft.CodeAnalysis.Shared.Extensions;
using Microsoft.CodeAnalysis.Test.Utilities;
using Roslyn.Utilities;
using Xunit;
 
namespace Microsoft.VisualStudio.LanguageServices.CSharp.UnitTests.DesignerAttributes
{
    [UseExportProvider]
    public class DesignerAttributeServiceTests
    {
        [Fact]
        public async Task NoDesignerTest1()
        {
            var code = @"class Test { }";
 
            await TestAsync(code, category: null);
        }
 
        [Fact]
        public async Task NoDesignerOnSecondClass()
        {
 
            await TestAsync(
@"class Test1 { }
 
[System.ComponentModel.DesignerCategory(""Form"")]
class Test2 { }", category: null);
        }
 
        [Fact]
        public async Task NoDesignerOnStruct()
        {
 
            await TestAsync(
@"
[System.ComponentModel.DesignerCategory(""Form"")]
struct Test1 { }", category: null);
        }
 
        [Fact]
        public async Task NoDesignerOnNestedClass()
        {
 
            await TestAsync(
@"class Test1
{
    [System.ComponentModel.DesignerCategory(""Form"")]
    class Test2 { }
}", category: null);
        }
 
        [Fact]
        public async Task SimpleDesignerTest()
        {
 
            await TestAsync(
@"[System.ComponentModel.DesignerCategory(""Form"")]
class Test { }", "Form");
        }
 
        [Fact]
        public async Task SimpleDesignerTest2()
        {
 
            await TestAsync(
@"using System.ComponentModel;
 
[DesignerCategory(""Form"")]
class Test { }", "Form");
        }
 
        private static async Task TestAsync(string codeWithMarker, string? category)
        {
            using var workspace = TestWorkspace.CreateCSharp(codeWithMarker, openDocuments: false);
 
            var hostDocument = workspace.Documents.First();
            var documentId = hostDocument.Id;
            var document = workspace.CurrentSolution.GetRequiredDocument(documentId);
 
            var compilation = await document.Project.GetRequiredCompilationAsync(CancellationToken.None);
            var actual = await DesignerAttributeDiscoveryService.ComputeDesignerAttributeCategoryAsync(
                AsyncLazy.Create(compilation.DesignerCategoryAttributeType() != null), document, CancellationToken.None);
 
            Assert.Equal(category, actual);
        }
    }
}