public async Task AgentlessTestEventTest()
{
var sender = new Mock <ICIAgentlessWriterSender>();
var agentlessWriter = new CIAgentlessWriter(sender.Object);
var span = new Span(new SpanContext(TraceId.CreateFromUlong(1), 1), DateTimeOffset.UtcNow);
span.Type = SpanTypes.Test;
span.SetTag(TestTags.Type, TestTags.TypeTest);
var expectedPayload = new Ci.Agent.Payloads.CITestCyclePayload();
expectedPayload.TryProcessEvent(new TestEvent(span));
var expectedBytes = expectedPayload.ToArray();
byte[] finalPayload = null;
sender.Setup(x => x.SendPayloadAsync(It.IsAny <Ci.Agent.Payloads.EventsPayload>()))
.Returns <Ci.Agent.Payloads.EventsPayload>(payload =>
{
finalPayload = payload.ToArray();
return(Task.CompletedTask);
});
var trace = new[] { span };
agentlessWriter.WriteTrace(new ArraySegment <Span>(trace));
await agentlessWriter.FlushTracesAsync(); // Force a flush to make sure the trace is written to the API
Assert.True(finalPayload.SequenceEqual(expectedBytes));
}
public async Task SlowSenderTest()
{
var flushTcs = new TaskCompletionSource <bool>();
var sender = new Mock <ICIAgentlessWriterSender>();
var agentlessWriter = new CIAgentlessWriter(sender.Object);
var lstPayloads = new List <byte[]>();
sender.Setup(x => x.SendPayloadAsync(It.IsAny <Ci.Agent.Payloads.EventsPayload>()))
.Returns <Ci.Agent.Payloads.EventsPayload>(payload =>
{
lstPayloads.Add(payload.ToArray());
return(flushTcs.Task);
});
var span = new Span(new SpanContext(TraceId.CreateFromUlong(1), 1), DateTimeOffset.UtcNow);
var expectedPayload = new Ci.Agent.Payloads.CITestCyclePayload();
expectedPayload.TryProcessEvent(new SpanEvent(span));
expectedPayload.TryProcessEvent(new SpanEvent(span));
expectedPayload.TryProcessEvent(new SpanEvent(span));
var expectedBytes = expectedPayload.ToArray();
agentlessWriter.WriteEvent(new SpanEvent(span));
agentlessWriter.WriteEvent(new SpanEvent(span));
agentlessWriter.WriteEvent(new SpanEvent(span));
var firstFlush = agentlessWriter.FlushTracesAsync();
agentlessWriter.WriteEvent(new SpanEvent(span));
agentlessWriter.WriteEvent(new SpanEvent(span));
agentlessWriter.WriteEvent(new SpanEvent(span));
var secondFlush = agentlessWriter.FlushTracesAsync();
flushTcs.TrySetResult(true);
agentlessWriter.WriteEvent(new SpanEvent(span));
agentlessWriter.WriteEvent(new SpanEvent(span));
agentlessWriter.WriteEvent(new SpanEvent(span));
var thirdFlush = agentlessWriter.FlushTracesAsync();
await Task.WhenAll(firstFlush, secondFlush, thirdFlush);
// We expect 3 batches.
Assert.Equal(3, lstPayloads.Count);
foreach (var payloadBytes in lstPayloads)
{
Assert.True(payloadBytes.SequenceEqual(expectedBytes));
}
}
