using System; using System.IO; using System.Linq; using System.Threading.Tasks; using CliFx.Infrastructure; using DiscordChatExporter.Cli.Commands; using DiscordChatExporter.Cli.Tests.Infra; using DiscordChatExporter.Cli.Tests.Utils; using DiscordChatExporter.Core.Discord; using DiscordChatExporter.Core.Exporting; using FluentAssertions; using JsonExtensions; using Xunit; namespace DiscordChatExporter.Cli.Tests.Specs; public class DateRangeSpecs { [Fact] public async Task I_can_filter_the_export_to_only_include_messages_sent_after_the_specified_date() { // Arrange var after = new DateTimeOffset(2021, 07, 24, 0, 0, 0, TimeSpan.Zero); using var file = TempFile.Create(); // Act await new ExportChannelsCommand { Token = Secrets.DiscordToken, ChannelIds = [ChannelIds.DateRangeTestCases], ExportFormat = ExportFormat.Json, OutputPath = file.Path, After = Snowflake.FromDate(after), }.ExecuteAsync(new FakeConsole()); // Assert var timestamps = Json.Parse(await File.ReadAllTextAsync(file.Path)) .GetProperty("messages") .EnumerateArray() .Select(j => j.GetProperty("timestamp").GetDateTimeOffset()) .ToArray(); timestamps.All(t => t > after).Should().BeTrue(); timestamps .Should() .BeEquivalentTo( [ new DateTimeOffset(2021, 07, 24, 13, 49, 13, TimeSpan.Zero), new DateTimeOffset(2021, 07, 24, 14, 52, 38, TimeSpan.Zero), new DateTimeOffset(2021, 07, 24, 14, 52, 39, TimeSpan.Zero), new DateTimeOffset(2021, 07, 24, 14, 52, 40, TimeSpan.Zero), new DateTimeOffset(2021, 09, 08, 14, 26, 35, TimeSpan.Zero), ], o => o.Using(ctx => ctx.Subject.Should().BeCloseTo(ctx.Expectation, TimeSpan.FromSeconds(1)) ) .WhenTypeIs() ); } [Fact] public async Task I_can_filter_the_export_to_only_include_messages_sent_before_the_specified_date() { // Arrange var before = new DateTimeOffset(2021, 07, 24, 0, 0, 0, TimeSpan.Zero); using var file = TempFile.Create(); // Act await new ExportChannelsCommand { Token = Secrets.DiscordToken, ChannelIds = [ChannelIds.DateRangeTestCases], ExportFormat = ExportFormat.Json, OutputPath = file.Path, Before = Snowflake.FromDate(before), }.ExecuteAsync(new FakeConsole()); // Assert var timestamps = Json.Parse(await File.ReadAllTextAsync(file.Path)) .GetProperty("messages") .EnumerateArray() .Select(j => j.GetProperty("timestamp").GetDateTimeOffset()) .ToArray(); timestamps.All(t => t < before).Should().BeTrue(); timestamps .Should() .BeEquivalentTo( [ new DateTimeOffset(2021, 07, 19, 13, 34, 18, TimeSpan.Zero), new DateTimeOffset(2021, 07, 19, 15, 58, 48, TimeSpan.Zero), new DateTimeOffset(2021, 07, 19, 17, 23, 58, TimeSpan.Zero), ], o => o.Using(ctx => ctx.Subject.Should().BeCloseTo(ctx.Expectation, TimeSpan.FromSeconds(1)) ) .WhenTypeIs() ); } [Fact] public async Task I_can_filter_the_export_to_only_include_messages_sent_between_the_specified_dates() { // Arrange var after = new DateTimeOffset(2021, 07, 24, 0, 0, 0, TimeSpan.Zero); var before = new DateTimeOffset(2021, 08, 01, 0, 0, 0, TimeSpan.Zero); using var file = TempFile.Create(); // Act await new ExportChannelsCommand { Token = Secrets.DiscordToken, ChannelIds = [ChannelIds.DateRangeTestCases], ExportFormat = ExportFormat.Json, OutputPath = file.Path, Before = Snowflake.FromDate(before), After = Snowflake.FromDate(after), }.ExecuteAsync(new FakeConsole()); // Assert var timestamps = Json.Parse(await File.ReadAllTextAsync(file.Path)) .GetProperty("messages") .EnumerateArray() .Select(j => j.GetProperty("timestamp").GetDateTimeOffset()) .ToArray(); timestamps.All(t => t < before && t > after).Should().BeTrue(); timestamps .Should() .BeEquivalentTo( [ new DateTimeOffset(2021, 07, 24, 13, 49, 13, TimeSpan.Zero), new DateTimeOffset(2021, 07, 24, 14, 52, 38, TimeSpan.Zero), new DateTimeOffset(2021, 07, 24, 14, 52, 39, TimeSpan.Zero), new DateTimeOffset(2021, 07, 24, 14, 52, 40, TimeSpan.Zero), ], o => o.Using(ctx => ctx.Subject.Should().BeCloseTo(ctx.Expectation, TimeSpan.FromSeconds(1)) ) .WhenTypeIs() ); } [Fact] public async Task Export_file_is_created_even_when_nothing_to_export() { var long_in_the_past = new DateTimeOffset(1921, 08, 01, 0, 0, 0, TimeSpan.Zero); // Arrange var before = long_in_the_past; using var file = TempFile.Create(); // Act await new ExportChannelsCommand { Token = Secrets.DiscordToken, ChannelIds = [ChannelIds.DateRangeTestCases], ExportFormat = ExportFormat.Json, OutputPath = file.Path, Before = Snowflake.FromDate(before), }.ExecuteAsync(new FakeConsole()); // Assert var timestamps = Json.Parse(await File.ReadAllTextAsync(file.Path)) .GetProperty("messages") .EnumerateArray() .Select(j => j.GetProperty("timestamp").GetDateTimeOffset()) .ToArray(); timestamps.Should().BeEmpty(); } }