DiscordChatExporter/DiscordChatExporter.Cli.Tests/Specs/FilterSpecs.cs

using System.IO;
using System.Linq;
using System.Threading.Tasks;
using CliFx.Infrastructure;
using DiscordChatExporter.Cli.Commands;
using DiscordChatExporter.Cli.Tests.Fixtures;
using DiscordChatExporter.Cli.Tests.Infra;
using DiscordChatExporter.Cli.Tests.TestData;
using DiscordChatExporter.Core.Exporting;
using DiscordChatExporter.Core.Exporting.Filtering;
using FluentAssertions;
using JsonExtensions;
using Xunit;

namespace DiscordChatExporter.Cli.Tests.Specs;

public record FilterSpecs(TempOutputFixture TempOutput) : IClassFixture<TempOutputFixture>
{
    [Fact]
    public async Task Messages_filtered_by_text_only_include_messages_that_contain_that_text()
    {
        // Arrange
        var filePath = TempOutput.GetTempFilePath();

        // Act
        await new ExportChannelsCommand
        {
            TokenValue = Secrets.DiscordToken,
            IsBotToken = Secrets.IsDiscordTokenBot,
            ChannelIds = new[] { ChannelIds.FilterTestCases },
            ExportFormat = ExportFormat.Json,
            OutputPath = filePath,
            MessageFilter = MessageFilter.Parse("some text")
        }.ExecuteAsync(new FakeConsole());

        var data = await File.ReadAllTextAsync(filePath);
        var document = Json.Parse(data);

        // Assert
        document
            .GetProperty("messages")
            .EnumerateArray()
            .Select(j => j.GetProperty("content").GetString())
            .Should()
            .ContainSingle("Some random text");
    }

    [Fact]
    public async Task Messages_filtered_by_author_only_include_messages_sent_by_that_author()
    {
        // Arrange
        var filePath = TempOutput.GetTempFilePath();

        // Act
        await new ExportChannelsCommand
        {
            TokenValue = Secrets.DiscordToken,
            IsBotToken = Secrets.IsDiscordTokenBot,
            ChannelIds = new[] { ChannelIds.FilterTestCases },
            ExportFormat = ExportFormat.Json,
            OutputPath = filePath,
            MessageFilter = MessageFilter.Parse("from:Tyrrrz")
        }.ExecuteAsync(new FakeConsole());

        var data = await File.ReadAllTextAsync(filePath);
        var document = Json.Parse(data);

        // Assert
        document
            .GetProperty("messages")
            .EnumerateArray()
            .Select(j => j.GetProperty("author").GetProperty("name").GetString())
            .Should()
            .AllBe("Tyrrrz");
    }

    [Fact]
    public async Task Messages_filtered_by_content_only_include_messages_that_have_that_content()
    {
        // Arrange
        var filePath = TempOutput.GetTempFilePath();

        // Act
        await new ExportChannelsCommand
        {
            TokenValue = Secrets.DiscordToken,
            IsBotToken = Secrets.IsDiscordTokenBot,
            ChannelIds = new[] { ChannelIds.FilterTestCases },
            ExportFormat = ExportFormat.Json,
            OutputPath = filePath,
            MessageFilter = MessageFilter.Parse("has:image")
        }.ExecuteAsync(new FakeConsole());

        var data = await File.ReadAllTextAsync(filePath);
        var document = Json.Parse(data);

        // Assert
        document
            .GetProperty("messages")
            .EnumerateArray()
            .Select(j => j.GetProperty("content").GetString())
            .Should()
            .ContainSingle("This has image");
    }

    [Fact]
    public async Task Messages_filtered_by_mention_only_include_messages_that_have_that_mention()
    {
        // Arrange
        var filePath = TempOutput.GetTempFilePath();

        // Act
        await new ExportChannelsCommand
        {
            TokenValue = Secrets.DiscordToken,
            IsBotToken = Secrets.IsDiscordTokenBot,
            ChannelIds = new[] { ChannelIds.FilterTestCases },
            ExportFormat = ExportFormat.Json,
            OutputPath = filePath,
            MessageFilter = MessageFilter.Parse("mentions:Tyrrrz")
        }.ExecuteAsync(new FakeConsole());

        var data = await File.ReadAllTextAsync(filePath);
        var document = Json.Parse(data);

        // Assert
        document
            .GetProperty("messages")
            .EnumerateArray()
            .Select(j => j.GetProperty("content").GetString())
            .Should()
            .ContainSingle("This has mention");
    }
}