cs-codex-dist-tests/Tests/FrameworkTests/OverwatchTranscriptTests/TranscriptTests.cs

230 lines
8.8 KiB
C#
Raw Normal View History

2024-10-10 13:31:38 +00:00
using Logging;
using Newtonsoft.Json;
using NUnit.Framework;
using OverwatchTranscript;
using System.IO.Compression;
namespace FrameworkTests.OverwatchTranscriptTests
{
[TestFixture]
public class TranscriptTests
{
private const string TranscriptFilename = "testtranscript.owts";
private const string HeaderKey = "testHeader";
private const string HeaderData = "abcdef";
private const string EventData0 = "12345";
private const string EventData1 = "678";
private const string EventData2 = "90";
private const string EventData3 = "-=";
private readonly DateTime t0 = DateTime.UtcNow;
private readonly DateTime t1 = DateTime.UtcNow.AddMinutes(1);
private readonly DateTime t2 = DateTime.UtcNow.AddMinutes(3);
[Test]
[Combinatorial]
public void WriteAndRun()
{
WriteTranscript();
AssertFileContent();
ReadTranscript();
File.Delete(TranscriptFilename);
}
private void WriteTranscript()
{
var log = new ConsoleLog();
var writer = Transcript.NewWriter(log);
writer.AddHeader(HeaderKey, new TestHeader
{
HeaderData = HeaderData
});
writer.Add(t0, new MyEvent
{
EventData = EventData0
});
writer.Add(t2, new MyEvent
{
EventData = EventData3
});
writer.Add(t1, new MyEvent
{
EventData = EventData1
});
writer.Add(t1, new MyEvent
{
EventData = EventData2
});
if (File.Exists(TranscriptFilename)) File.Delete(TranscriptFilename);
writer.Write(TranscriptFilename);
}
private void ReadTranscript()
{
var reader = Transcript.NewReader(TranscriptFilename);
var header = reader.GetHeader<TestHeader>(HeaderKey);
Assert.That(header.HeaderData, Is.EqualTo(HeaderData));
Assert.That(reader.Header.NumberOfMoments, Is.EqualTo(3));
Assert.That(reader.Header.NumberOfEvents, Is.EqualTo(4));
Assert.That(reader.Header.EarliestUtc, Is.EqualTo(t0));
Assert.That(reader.Header.LatestUtc, Is.EqualTo(t2));
var moments = new List<ActivateMoment>();
var events = new List<ActivateEvent<MyEvent>>();
reader.AddMomentHandler(moments.Add);
reader.AddEventHandler<MyEvent>(events.Add);
var timeout = 10;
while (moments.Count < 3 && events.Count < 4)
{
Thread.Sleep(10);
reader.Next();
timeout--;
if (timeout == 0) Assert.Fail("Events not received.");
}
reader.Next();
Assert.That(moments.Count, Is.EqualTo(3));
Assert.That(events.Count, Is.EqualTo(4));
AssertMoment(moments[0], utc: t0, duration: t1 - t0, index: 0);
AssertMoment(moments[1], utc: t1, duration: t2 - t1, index: 1);
AssertMoment(moments[2], utc: t2, duration: null, index: 2);
AssertEvent(events, utc: t0, duration: t1 - t0, index: 0, data: EventData0);
AssertEvent(events, utc: t1, duration: t2 - t1, index: 1, data: EventData2);
AssertEvent(events, utc: t1, duration: t2 - t1, index: 1, data: EventData1);
AssertEvent(events, utc: t2, duration: null, index: 2, data: EventData3);
reader.Close();
}
private void AssertMoment(ActivateMoment m, DateTime utc, TimeSpan? duration, int index)
{
Assert.That(m.Utc, Is.EqualTo(utc));
Assert.That(m.Duration, Is.EqualTo(duration));
Assert.That(m.Index, Is.EqualTo(index));
}
private void AssertEvent(List<ActivateEvent<MyEvent>> events, DateTime utc, TimeSpan? duration, int index, string data)
{
var e = events.SingleOrDefault(e => e.Moment.Utc == utc && e.Payload.EventData == data);
if (e == null) Assert.Fail("Event not found");
Assert.That(e.Moment.Utc, Is.EqualTo(utc));
Assert.That(e.Moment.Duration, Is.EqualTo(duration));
Assert.That(e.Moment.Index, Is.EqualTo(index));
Assert.That(e.Payload.EventData, Is.EqualTo(data));
}
private void AssertFileContent()
{
using var zip = ZipFile.OpenRead(TranscriptFilename);
Assert.That(zip.Entries.Count, Is.EqualTo(2));
foreach (var entry in zip.Entries)
{
if (entry.Name == "transcript.json")
{
var transcript = ZipEntryJson<OverwatchTranscript.OverwatchTranscript>(entry);
AssertTranscript(transcript);
}
else
{
var moments = ZipEntryToMoments(entry);
AssertMoments(moments);
}
}
}
private void AssertTranscript(OverwatchTranscript.OverwatchTranscript transcript)
{
Assert.That(transcript.Header.Common.NumberOfMoments, Is.EqualTo(3));
Assert.That(transcript.Header.Common.NumberOfEvents, Is.EqualTo(4));
Assert.That(transcript.Header.Common.EarliestUtc, Is.EqualTo(t0));
Assert.That(transcript.Header.Common.LatestUtc, Is.EqualTo(t2));
Assert.That(transcript.Header.Entries.Length, Is.EqualTo(1));
Assert.That(transcript.Header.Entries[0].Key, Is.EqualTo(HeaderKey));
Assert.That(transcript.Header.Entries[0].Value, Is.EqualTo("{\"HeaderData\":\"abcdef\"}"));
Assert.That(transcript.MomentReferences.Length, Is.EqualTo(1));
Assert.That(transcript.MomentReferences[0].NumberOfMoments, Is.EqualTo(3));
Assert.That(transcript.MomentReferences[0].NumberOfEvents, Is.EqualTo(4));
Assert.That(transcript.MomentReferences[0].EarliestUtc, Is.EqualTo(t0));
Assert.That(transcript.MomentReferences[0].LatestUtc, Is.EqualTo(t2));
}
private void AssertMoments(OverwatchMoment[] moments)
{
Assert.That(moments.Length, Is.EqualTo(3));
Assert.That(moments[0].Utc, Is.EqualTo(t0));
Assert.That(moments[0].Events.Length, Is.EqualTo(1));
Assert.That(moments[0].Events[0].Type, Is.EqualTo("FrameworkTests.OverwatchTranscriptTests.MyEvent"));
Assert.That(moments[0].Events[0].Payload, Is.EqualTo("{\"EventData\":\"12345\"}"));
Assert.That(moments[1].Utc, Is.EqualTo(t1));
Assert.That(moments[1].Events.Length, Is.EqualTo(2));
Assert.That(moments[1].Events[0].Type, Is.EqualTo("FrameworkTests.OverwatchTranscriptTests.MyEvent"));
Assert.That(moments[1].Events[1].Type, Is.EqualTo("FrameworkTests.OverwatchTranscriptTests.MyEvent"));
// output order is not guaranteed:
var payloads = moments[1].Events.Select(e => e.Payload).ToArray();
CollectionAssert.AreEquivalent(new[]
{
"{\"EventData\":\"90\"}",
"{\"EventData\":\"678\"}"
}, payloads);
Assert.That(moments[2].Utc, Is.EqualTo(t2));
Assert.That(moments[2].Events.Length, Is.EqualTo(1));
Assert.That(moments[2].Events[0].Type, Is.EqualTo("FrameworkTests.OverwatchTranscriptTests.MyEvent"));
Assert.That(moments[2].Events[0].Payload, Is.EqualTo("{\"EventData\":\"-=\"}"));
}
private T ZipEntryJson<T>(ZipArchiveEntry? entry)
{
if (entry == null) Assert.Fail("entry is null");
using var stream = entry!.Open();
using var reader = new StreamReader(stream);
var json = reader.ReadToEnd();
var result = JsonConvert.DeserializeObject<T>(json);
if (result == null) Assert.Fail("didn't deserialize");
return result!;
}
private OverwatchMoment[] ZipEntryToMoments(ZipArchiveEntry? entry)
{
var result = new List<OverwatchMoment>();
if (entry == null) Assert.Fail("entry is null");
using var stream = entry!.Open();
using var reader = new StreamReader(stream);
var line = reader.ReadLine();
while (!string.IsNullOrEmpty(line))
{
var moment = JsonConvert.DeserializeObject<OverwatchMoment>(line);
if (moment != null) result.Add(moment);
line = reader.ReadLine();
}
return result.ToArray();
}
}
public class TestHeader
{
public string HeaderData { get; set; } = string.Empty;
}
public class MyEvent
{
public string EventData { get; set; } = string.Empty;
}
}