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 = new[] { 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[] { 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 => { return 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 = new[] { 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[] { 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 => { return 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 = new[] { 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[] { 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 => { return o .Using(ctx => ctx.Subject.Should().BeCloseTo(ctx.Expectation, TimeSpan.FromSeconds(1)) ) .WhenTypeIs(); }); } }