mfgames-gallium-cil/tests/MfGames.Gallium.Tests/EnumerableEntityTests.cs

using System.Linq;

using MfGames.Gallium;

using Xunit;

namespace MfGames.Gallium.Tests;

public class EnumerableEntityTests
{
    [Fact]
    public void ForComponentsC1()
    {
        Entity[] entities =
        {
            new Entity().Add("1")
                .Add(new TestComponent1()),
            new Entity().Add("2")
                .Add(new TestComponent2()),
            new Entity().Add("3")
                .Add(new TestComponent1())
        };

        Assert.Equal(
            new[] { "1!", "2", "3!" },
            entities.SelectEntity<TestComponent1>(
                    (
                        e,
                        _) => e.Set(e.Get<string>() + "!"))
                .Select(x => x.Get<string>())
                .ToArray());
    }

    [Fact]
    public void ForComponentsC2()
    {
        Entity[] entities =
        {
            new Entity().Add("1")
                .Add(new TestComponent1()),
            new Entity().Add("2")
                .Add(new TestComponent2()),
            new Entity().Add("3")
                .Add(new TestComponent1())
                .Add(new TestComponent2())
        };

        Assert.Equal(
            new[] { "1", "2", "3!" },
            entities.SelectEntity<TestComponent1, TestComponent2>(
                    (
                        e,
                        _,
                        _) => e.Set(e.Get<string>() + "!"))
                .Select(x => x.Get<string>())
                .ToArray());
    }

    [Fact]
    public void ForComponentsC3()
    {
        Entity[] entities =
        {
            new Entity().Add("1")
                .Add(new TestComponent1()),
            new Entity().Add("2")
                .Add(new TestComponent2())
                .Add<ITestComponent3>(new TestComponent3b()),
            new Entity().Add("3")
                .Add<ITestComponent3>(new TestComponent3a())
                .Add(new TestComponent1())
                .Add(new TestComponent2())
        };

        Assert.Equal(
            new[] { "1", "2", "3-TestComponent3a" },
            entities
                .SelectEntity<TestComponent1, TestComponent2, ITestComponent3>(
                    (
                        e,
                        _,
                        _,
                        t) => e.Set(
                        e.Get<string>()
                        + "-"
                        + t.GetType()
                            .Name))
                .Select(x => x.Get<string>())
                .ToArray());
    }

    [Fact]
    public void HasComponentsC1()
    {
        Entity[] entities =
        {
            new Entity().Add("1")
                .Add(new TestComponent1()),
            new Entity().Add("2")
                .Add(new TestComponent2()),
            new Entity().Add("3")
                .Add(new TestComponent1())
        };

        Assert.Equal(
            new[] { "1", "3" },
            entities.WhereEntityHas<TestComponent1>()
                .Select(x => x.Get<string>())
                .ToArray());
    }

    [Fact]
    public void HasComponentsC2()
    {
        Entity[] entities =
        {
            new Entity().Add("1")
                .Add(new TestComponent1()),
            new Entity().Add("2")
                .Add(new TestComponent2())
                .Add(new TestComponent1()),
            new Entity().Add("3")
                .Add(new TestComponent1())
        };

        Assert.Equal(
            new[] { "2" },
            entities.WhereEntityHasAll<TestComponent1, TestComponent2>()
                .Select(x => x.Get<string>())
                .ToArray());
    }

    [Fact]
    public void HasComponentsC3()
    {
        Entity[] entities =
        {
            new Entity().Add("1")
                .Add(new TestComponent1())
                .Add<ITestComponent3>(new TestComponent3b())
                .Add(new TestComponent2()),
            new Entity().Add("2")
                .Add(new TestComponent2())
                .Add(new TestComponent1()),
            new Entity().Add("3")
                .Add<ITestComponent3>(new TestComponent3a())
        };

        Assert.Equal(
            new[] { "1" },
            entities
                .WhereEntityHasAll<TestComponent1, TestComponent2,
                    ITestComponent3>()
                .Select(x => x.Get<string>())
                .ToArray());
    }

    [Fact]
    public void NotComponentsC1()
    {
        Entity[] entities =
        {
            new Entity().Add("1")
                .Add(new TestComponent1()),
            new Entity().Add("2")
                .Add(new TestComponent2()),
            new Entity().Add("3")
                .Add(new TestComponent1())
        };

        Assert.Equal(
            new[] { "2" },
            entities.WhereEntityNotHas<TestComponent1>()
                .Select(x => x.Get<string>())
                .ToArray());
    }

    [Fact]
    public void NotComponentsC2()
    {
        Entity[] entities =
        {
            new Entity().Add("1")
                .Add(new TestComponent1()),
            new Entity().Add("2")
                .Add(new TestComponent2())
                .Add(new TestComponent1()),
            new Entity().Add("3")
        };

        Assert.Equal(
            new[] { "1", "3" },
            entities.WhereEntityNotHasAll<TestComponent1, TestComponent2>()
                .Select(x => x.Get<string>())
                .ToArray());
    }

    [Fact]
    public void NotComponentsC3()
    {
        Entity[] entities =
        {
            new Entity().Add("1")
                .Add(new TestComponent1()),
            new Entity().Add("2")
                .Add(new TestComponent1())
                .Add(new TestComponent2())
                .Add<ITestComponent3>(new TestComponent3b()),
            new Entity().Add("3")
                .Add<ITestComponent3>(new TestComponent3a())
        };

        Assert.Equal(
            new[] { "1", "3" },
            entities
                .WhereEntityNotHasAll<TestComponent1, TestComponent2,
                    ITestComponent3>()
                .Select(x => x.Get<string>())
                .ToArray());
    }
}