public async Task CanPublishCloudEvent()
{
EventGridPublisherClientOptions options = InstrumentClientOptions(new EventGridPublisherClientOptions());
EventGridPublisherClient client = InstrumentClient(
new EventGridPublisherClient(
new Uri(TestEnvironment.CloudEventTopicHost),
new AzureKeyCredential(TestEnvironment.CloudEventTopicKey),
options));
List <CloudEvent> eventsList = new List <CloudEvent>();
for (int i = 0; i < 10; i++)
{
eventsList.Add(
new CloudEvent
{
Type = "record",
Source = new Uri("http://localHost"),
Id = Recording.Random.NewGuid().ToString(),
Time = Recording.Now,
Data = new TestPayload("name", i)
});
}
await client.SendCloudNativeCloudEventsAsync(eventsList);
}
public async Task SetsTraceParentExtension(bool inclTraceparent, bool inclTracestate)
{
var mockTransport = new MockTransport(new MockResponse(200));
var options = new EventGridPublisherClientOptions
{
Transport = mockTransport
};
EventGridPublisherClient client =
new EventGridPublisherClient(
new Uri("http://localHost"),
new AzureKeyCredential("fakeKey"),
options);
var activity = new Activity($"{nameof(EventGridPublisherClient)}.{nameof(EventGridPublisherClient.SendEvents)}");
activity.SetW3CFormat();
activity.Start();
List <CloudEvent> inputEvents = new List <CloudEvent>();
for (int i = 0; i < 10; i++)
{
var cloudEvent =
new CloudEvent
{
Subject = "record",
Source = new Uri("http://localHost"),
Id = Guid.NewGuid().ToString(),
Time = DateTime.Now,
Type = "test"
};
if (inclTraceparent && inclTracestate && i % 2 == 0)
{
cloudEvent.SetAttributeFromString("traceparent", "traceparentValue");
}
if (inclTracestate && i % 2 == 0)
{
cloudEvent.SetAttributeFromString("tracestate", "param:value");
}
inputEvents.Add(cloudEvent);
}
await client.SendCloudNativeCloudEventsAsync(inputEvents);
activity.Stop();
List <CloudEvent> endEvents = DeserializeRequest(mockTransport.SingleRequest);
IEnumerator <CloudEvent> inputEnum = inputEvents.GetEnumerator();
foreach (CloudEvent cloudEvent in endEvents)
{
inputEnum.MoveNext();
var inputAttributes = inputEnum.Current.GetPopulatedAttributes().Select(pair => pair.Key.Name).ToList();
if (inputAttributes.Contains(TraceParentHeaderName) &&
inputAttributes.Contains(TraceStateHeaderName))
{
Assert.AreEqual(
inputEnum.Current[TraceParentHeaderName],
cloudEvent[TraceParentHeaderName]);
Assert.AreEqual(
inputEnum.Current[TraceStateHeaderName],
cloudEvent[TraceStateHeaderName]);
}
else if (inputAttributes.Contains(TraceParentHeaderName))
{
Assert.AreEqual(
inputEnum.Current[TraceParentHeaderName],
cloudEvent[TraceParentHeaderName]);
}
else if (inputAttributes.Contains(TraceStateHeaderName))
{
Assert.AreEqual(
inputEnum.Current[TraceStateHeaderName],
cloudEvent[TraceStateHeaderName]);
}
else
{
Assert.AreEqual(
activity.Id,
cloudEvent[TraceParentHeaderName]);
}
}
}