using System;
using System.Collections.Generic;
using System.Linq;
using FluentAssertions;
using Xunit;

namespace Wasmtime.Tests
{
    public class FunctionImportsFixture : ModuleFixture
    {
        protected override string ModuleFileName => "FunctionImports.wasm";
    }

    public class FunctionImportsTests : IClassFixture<FunctionImportsFixture>
    {
        public FunctionImportsTests(FunctionImportsFixture fixture)
        {
            Fixture = fixture;
        }

        private FunctionImportsFixture Fixture { get; set; }

        [Theory]
        [MemberData(nameof(GetFunctionImports))]
        public void ItHasTheExpectedFunctionImports(string importModule, string importName, ValueKind[] expectedParameters, ValueKind[] expectedResults)
        {
            var import = Fixture.Module.Imports.Functions.Where(f => f.ModuleName == importModule && f.Name == importName).FirstOrDefault();
            import.Should().NotBeNull();
            import.Parameters.Should().Equal(expectedParameters);
            import.Results.Should().Equal(expectedResults);
        }

        [Fact]
        public void ItHasTheExpectedNumberOfExportedFunctions()
        {
            GetFunctionImports().Count().Should().Be(Fixture.Module.Imports.Functions.Count);
        }

        public static IEnumerable<object[]> GetFunctionImports()
        {
            yield return new object[] {
                "",
                "no_params_no_results",
                Array.Empty<ValueKind>(),
                Array.Empty<ValueKind>()
            };

            yield return new object[] {
                "",
                "one_i32_param_no_results",
                new ValueKind[] {
                    ValueKind.Int32
                },
                Array.Empty<ValueKind>()
            };

            yield return new object[] {
                "",
                "one_i64_param_no_results",
                new ValueKind[] {
                    ValueKind.Int64
                },
                Array.Empty<ValueKind>()
            };

            yield return new object[] {
                "",
                "one_f32_param_no_results",
                new ValueKind[] {
                    ValueKind.Float32
                },
                Array.Empty<ValueKind>()
            };

            yield return new object[] {
                "",
                "one_f64_param_no_results",
                new ValueKind[] {
                    ValueKind.Float64
                },
                Array.Empty<ValueKind>()
            };

            yield return new object[] {
                "",
                "one_param_of_each_type",
                new ValueKind[] {
                    ValueKind.Int32,
                    ValueKind.Int64,
                    ValueKind.Float32,
                    ValueKind.Float64
                },
                Array.Empty<ValueKind>()
            };

            yield return new object[] {
                "",
                "no_params_one_i32_result",
                Array.Empty<ValueKind>(),
                new ValueKind[] {
                    ValueKind.Int32,
                }
            };

            yield return new object[] {
                "",
                "no_params_one_i64_result",
                Array.Empty<ValueKind>(),
                new ValueKind[] {
                    ValueKind.Int64,
                }
            };

            yield return new object[] {
                "",
                "no_params_one_f32_result",
                Array.Empty<ValueKind>(),
                new ValueKind[] {
                    ValueKind.Float32,
                }
            };

            yield return new object[] {
                "",
                "no_params_one_f64_result",
                Array.Empty<ValueKind>(),
                new ValueKind[] {
                    ValueKind.Float64,
                }
            };

            yield return new object[] {
                "",
                "one_result_of_each_type",
                Array.Empty<ValueKind>(),
                new ValueKind[] {
                    ValueKind.Int32,
                    ValueKind.Int64,
                    ValueKind.Float32,
                    ValueKind.Float64,
                }
            };

            yield return new object[] {
                "",
                "one_param_and_result_of_each_type",
                new ValueKind[] {
                    ValueKind.Int32,
                    ValueKind.Int64,
                    ValueKind.Float32,
                    ValueKind.Float64,
                },
                new ValueKind[] {
                    ValueKind.Int32,
                    ValueKind.Int64,
                    ValueKind.Float32,
                    ValueKind.Float64,
                }
            };

            yield return new object[] {
                "other",
                "function_from_module",
                Array.Empty<ValueKind>(),
                Array.Empty<ValueKind>(),
            };
        }
    }
}