ProxyGenerator/tests/ProxyInterfaceSourceGeneratorTests/PnPTests.cs

using System.IO;
using System.Linq;
using CSharp.SourceGenerators.Extensions;
using CSharp.SourceGenerators.Extensions.Models;
using FluentAssertions;
using Speckle.ProxyGenerator;
using Xunit;

namespace ProxyInterfaceSourceGeneratorTests;

public class PnPTests
{
    private bool Write = true;

    private readonly ProxyInterfaceCodeGenerator _sut;

    public PnPTests()
    {
        _sut = new ProxyInterfaceCodeGenerator();
    }

    [Fact]
    public void GenerateFiles_Should_GenerateCorrectFiles()
    {
        // Arrange
        var fileNames = new[]
        {
            "ProxyInterfaceSourceGeneratorTests.Source.PnP.IClientObject.g.cs",
            "ProxyInterfaceSourceGeneratorTests.Source.PnP.ISecurableObject.g.cs",
            "ProxyInterfaceSourceGeneratorTests.Source.PnP.IWeb.g.cs",
            "ProxyInterfaceSourceGeneratorTests.Source.PnP.IClientRuntimeContext.g.cs",
            "ProxyInterfaceSourceGeneratorTests.Source.PnP.IClientContext.g.cs",
            "Microsoft.SharePoint.Client.ClientObjectProxy.g.cs",
            "Microsoft.SharePoint.Client.SecurableObjectProxy.g.cs",
            "Microsoft.SharePoint.Client.WebProxy.g.cs",
            "Microsoft.SharePoint.Client.ClientRuntimeContextProxy.g.cs",
            "Microsoft.SharePoint.Client.ClientContextProxy.g.cs"
        };

        var pathClientObject = "./Source/PnP/IClientObject.cs";
        var sourceFileClientObject = new SourceFile
        {
            Path = pathClientObject,
            Text = File.ReadAllText(pathClientObject),
            AttributeToAddToInterface = new ExtraAttribute
            {
                Name = "Speckle.ProxyGenerator.Proxy",
                ArgumentList = new[]
                {
                    "typeof(Microsoft.SharePoint.Client.ClientObject)",
                    "ImplementationOptions.UseExtendedInterfaces"
                }
            }
        };

        var pathSec = "./Source/PnP/ISecurableObject.cs";
        var sourceFileSec = new SourceFile
        {
            Path = pathSec,
            Text = File.ReadAllText(pathSec),
            AttributeToAddToInterface = new ExtraAttribute
            {
                Name = "Speckle.ProxyGenerator.Proxy",
                ArgumentList = new[]
                {
                    "typeof(SecurableObject)",
                    "ImplementationOptions.UseExtendedInterfaces"
                } // Only name, no namespace
            }
        };

        var pathWeb = "./Source/PnP/IWeb.cs";
        var sourceFileWeb = new SourceFile
        {
            Path = pathWeb,
            Text = File.ReadAllText(pathWeb),
            AttributeToAddToInterface = new ExtraAttribute
            {
                Name = "Speckle.ProxyGenerator.Proxy",
                ArgumentList = new[]
                {
                    "typeof(Web)",
                    "ImplementationOptions.UseExtendedInterfaces"
                } // Only name, no namespace
            }
        };

        var pathClientRuntimeContext = "./Source/Pnp/IClientRuntimeContext.cs";
        var sourceFileClientRuntimeContext = new SourceFile
        {
            Path = pathClientRuntimeContext,
            Text = File.ReadAllText(pathClientRuntimeContext),
            AttributeToAddToInterface = new ExtraAttribute
            {
                Name = "Speckle.ProxyGenerator.Proxy",
                ArgumentList = new[]
                {
                    "typeof(Microsoft.SharePoint.Client.ClientRuntimeContext)",
                    "ImplementationOptions.UseExtendedInterfaces"
                }
            }
        };

        var pathClientContext = "./Source/PnP/IClientContext.cs";
        var sourceFileClientContext = new SourceFile
        {
            Path = pathClientContext,
            Text = File.ReadAllText(pathClientContext),
            AttributeToAddToInterface = new ExtraAttribute
            {
                Name = "Speckle.ProxyGenerator.Proxy",
                ArgumentList = new[]
                {
                    "typeof(Microsoft.SharePoint.Client.ClientContext)",
                    "ImplementationOptions.UseExtendedInterfaces"
                } // Only name, no namespace
            }
        };

        // Act
        var result = _sut.Execute(
            new[]
            {
                sourceFileClientObject,
                sourceFileSec,
                sourceFileWeb,
                sourceFileClientRuntimeContext,
                sourceFileClientContext
            }
        );

        // Assert
        result.Valid.Should().BeTrue();
        result.Files.Should().HaveCount(fileNames.Length + 1);

        foreach (var fileName in fileNames.Select((fileName, index) => new { fileName, index }))
        {
            var builder = result.Files[fileName.index]; // attribute is last
            builder.Path.Should().EndWith(fileName.fileName);

            if (Write)
                File.WriteAllText($"../../../Destination/{fileName.fileName}", builder.Text);
            builder.Text.Should().Be(File.ReadAllText($"../../../Destination/{fileName.fileName}"));
        }
    }
}