InterfaceGenerator/InterfaceGenerator.Tests/MethodGenerationTests.cs

using System;
using System.Linq;
using System.Reflection;
using System.Text;
using FluentAssertions;
using Xunit;

namespace InterfaceGenerator.Tests;

public class MethodGenerationTests
{
    private readonly IMethodsTestService _sut;

    public MethodGenerationTests()
    {
        _sut = new MethodsTestService();
    }

    [Fact]
    public void VoidMethod_IsImplemented()
    {
        var method = typeof(IMethodsTestService).GetMethod(
            nameof(MethodsTestService.VoidMethod))!;

        method.Should().NotBeNull();
        method.ReturnType.Should().Be(typeof(void));

        var parameters = method.GetParameters();
        parameters.Should().BeEmpty();

        _sut.VoidMethod();
    }

    [Fact]
    public void VoidMethodWithKeywordParam_IsImplemented()
    {
        var method = typeof(IMethodsTestService).GetMethod(
            nameof(MethodsTestService.VoidMethodWithKeywordParam))!;

        method.Should().NotBeNull();
        method.ReturnType.Should().Be(typeof(void));

        var parameters = method.GetParameters();
        parameters.Select(x => x.ParameterType).Should().AllBeEquivalentTo(typeof(string));
        parameters.Should().HaveCount(1);

        parameters[0].Name.Should().Be("void");

        _sut.VoidMethodWithKeywordParam("");
    }

    [Fact]
    public void VoidMethodWithParams_IsImplemented()
    {
        var method = typeof(IMethodsTestService).GetMethod(
            nameof(MethodsTestService.VoidMethodWithParams),
            new[] { typeof(string), typeof(string) })!;

        method.Should().NotBeNull();
        method.ReturnType.Should().Be(typeof(void));

        var parameters = method.GetParameters();
        parameters.Select(x => x.ParameterType).Should().AllBeEquivalentTo(typeof(string));
        parameters.Should().HaveCount(2);

        _sut.VoidMethodWithParams(string.Empty, string.Empty);
    }

    [Fact]
    public void VoidMethodWithOutParam_IsImplemented()
    {
        var method = typeof(IMethodsTestService).GetMethod(
            nameof(MethodsTestService.VoidMethodWithOutParam),
            new[] { typeof(string).MakeByRefType() })!;

        method.Should().NotBeNull();
        method.ReturnType.Should().Be(typeof(void));

        var parameters = method.GetParameters();
        parameters.Select(x => x.ParameterType).Should().AllBeEquivalentTo(typeof(string).MakeByRefType());
        parameters.Should().HaveCount(1);
        parameters[0].IsOut.Should().BeTrue();

        _sut.VoidMethodWithOutParam(out var _);
    }

    [Fact]
    public void VoidMethodWithInParam_IsImplemented()
    {
        var method = typeof(IMethodsTestService).GetMethod(
            nameof(MethodsTestService.VoidMethodWithInParam),
            new[] { typeof(string).MakeByRefType() })!;

        method.Should().NotBeNull();
        method.ReturnType.Should().Be(typeof(void));

        var parameters = method.GetParameters();
        parameters.Select(x => x.ParameterType).Should().AllBeEquivalentTo(typeof(string).MakeByRefType());
        parameters.Should().HaveCount(1);
        parameters[0].IsIn.Should().BeTrue();

        var stub = string.Empty;
        _sut.VoidMethodWithInParam(in stub);
    }

    [Fact]
    public void VoidMethodWithRefParam_IsImplemented()
    {
        var method = typeof(IMethodsTestService).GetMethod(
            nameof(MethodsTestService.VoidMethodWithRefParam),
            new[] { typeof(string).MakeByRefType() })!;

        method.Should().NotBeNull();
        method.ReturnType.Should().Be(typeof(void));

        var parameters = method.GetParameters();
        parameters.Select(x => x.ParameterType).Should().AllBeEquivalentTo(typeof(string).MakeByRefType());
        parameters.Should().HaveCount(1);
        parameters[0].IsIn.Should().BeFalse();
        parameters[0].IsOut.Should().BeFalse();

        var stub = string.Empty;
        _sut.VoidMethodWithRefParam(ref stub);
    }

    [Fact]
    public void StringMethod_IsImplemented()
    {
        var method = typeof(IMethodsTestService).GetMethod(
            nameof(MethodsTestService.StringMethod))!;

        method.Should().NotBeNull();
        method.ReturnType.Should().Be(typeof(string));

        var parameters = method.GetParameters();
        parameters.Should().BeEmpty();

        var _ = _sut.StringMethod();
    }

    [Fact]
    public void GenericVoidMethod_IsImplemented()
    {
        var method = typeof(IMethodsTestService)
                     .GetMethods()
                     .First(x => x.Name == nameof(MethodsTestService.GenericVoidMethod));

        method.Should().NotBeNull();
        method.ReturnType.Should().Be(typeof(void));

        method.GetParameters().Should().BeEmpty();

        var genericArgs = method.GetGenericArguments();
        genericArgs.Should().HaveCount(2);

        _sut.GenericVoidMethod<string, int>();
    }

    [Fact]
    public void GenericVoidMethodWithGenericParam_IsImplemented()
    {
        var method = typeof(IMethodsTestService)
                     .GetMethods()
                     .First(x => x.Name == nameof(MethodsTestService.GenericVoidMethodWithGenericParam));

        method.Should().NotBeNull();
        method.ReturnType.Should().Be(typeof(void));

        var genericArgs = method.GetGenericArguments();
        genericArgs.Should().HaveCount(2);

        var parameters = method.GetParameters();
        parameters.Should().HaveCount(1);
        parameters[0].ParameterType.Should().Be(genericArgs[0]);

        _sut.GenericVoidMethodWithGenericParam<string, int>(string.Empty);
    }

    [Fact]
    public void GenericVoidMethodWithConstraints_IsImplemented()
    {
        var method = typeof(IMethodsTestService)
                     .GetMethods()
                     .First(x => x.Name == nameof(MethodsTestService.GenericVoidMethodWithConstraints));

        method.Should().NotBeNull();
        method.ReturnType.Should().Be(typeof(void));

        var genericArgs = method.GetGenericArguments();
        genericArgs.Should().HaveCount(2);

        genericArgs[0].IsClass.Should().BeTrue();
        genericArgs[0].GetGenericParameterConstraints().Should().HaveCount(0);

        genericArgs[1].IsClass.Should().BeTrue();
        genericArgs[1].GetGenericParameterConstraints().Should().HaveCount(1);
        genericArgs[1].GetGenericParameterConstraints()[0].Should().Be(genericArgs[0]);
        genericArgs[1]
            .GenericParameterAttributes.Should()
            .HaveFlag(GenericParameterAttributes.DefaultConstructorConstraint);

        _sut.GenericVoidMethodWithConstraints<object, StringBuilder>();
    }

    [Fact]
    public void VoidMethodWithOptionalParams_IsImplemented()
    {
        var method = typeof(IMethodsTestService)
                     .GetMethods()
                     .First(x => x.Name == nameof(MethodsTestService.VoidMethodWithOptionalParams));

        method.Should().NotBeNull();
        method.ReturnType.Should().Be(typeof(void));

        var parameters = method.GetParameters();
        parameters.Should().HaveCount(7);
        parameters.Select(x => x.IsOptional).Should().AllBeEquivalentTo(true);

        parameters[0].DefaultValue.Should().Be("cGFyYW0=");
        parameters[1].DefaultValue.Should().Be(MethodsTestService.StringConstant);
        parameters[2].DefaultValue.Should().Be(0.1f);
        parameters[3].DefaultValue.Should().Be(0.2d);
        parameters[4].DefaultValue.Should().Be(0.3d);
        parameters[5].DefaultValue.Should().Be(true);
        parameters[6].DefaultValue.Should().Be(false);

        _sut.VoidMethodWithOptionalParams();
    }

    [Fact]
    public void VoidMethodWithExpandingParam_IsImplemented()
    {
        var method = typeof(IMethodsTestService)
                     .GetMethods()
                     .First(x => x.Name == nameof(MethodsTestService.VoidMethodWithExpandingParam));

        method.ReturnType.Should().Be(typeof(void));

        var parameters = method.GetParameters();
        parameters.Should().HaveCount(1);
        parameters[0].ParameterType.Should().Be(typeof(string[]));
        parameters[0].GetCustomAttribute<ParamArrayAttribute>().Should().NotBeNull();
    }

    [Fact]
    public void IgnoreMethod_IsOmitted()
    {
        var method = typeof(IMethodsTestService)
                     .GetMethods()
                     .FirstOrDefault(x => x.Name == nameof(MethodsTestService.IgnoredMethod));

        method.Should().BeNull();
    }

    [Fact]
    public void StaticMethod_IsOmitted()
    {
        var method = typeof(IMethodsTestService)
                     .GetMethods()
                     .FirstOrDefault(x => x.Name == nameof(MethodsTestService.StaticMethod));

        method.Should().BeNull();
    }
}

[GenerateAutoInterface]
internal class MethodsTestService : IMethodsTestService
{
    public const string StringConstant = "Const";

    public void VoidMethod()
    {
    }

    public void VoidMethodWithParams(string a, string b)
    {
    }

    public void VoidMethodWithKeywordParam(string @void)
    {
    }

    public void VoidMethodWithOutParam(out string a)
    {
        a = default;
    }

    public void VoidMethodWithRefParam(ref string a)
    {
    }

    public void VoidMethodWithInParam(in string a)
    {
    }

    public string StringMethod()
    {
        return string.Empty;
    }

    public void GenericVoidMethod<TX, TY>()
    {
    }

    public void GenericVoidMethodWithGenericParam<TX, TY>(TX a)
    {
    }

    public void GenericVoidMethodWithConstraints<TX, TY>()
        where TX : class
        where TY : class, TX, new()
    {
    }

    public void VoidMethodWithOptionalParams(
        string stringLiteral = "cGFyYW0=",
        string stringConstant = StringConstant,
        float floatLiteral = 0.1f,
        double doubleLiteral = 0.2,
        decimal decimalLiteral = 0.3m,
        bool trueLiteral = true,
        bool falseLiteral = false)
    {
    }

    public void VoidMethodWithExpandingParam(params string[] strings)
    {
    }

    [AutoInterfaceIgnore]
    public void IgnoredMethod()
    {
    }

    public static void StaticMethod()
    {
    }
}

[GenerateAutoInterface]
internal class MethodsTestServiceGeneric<T> : IMethodsTestServiceGeneric<T> where T : class
{
}