Skip to content

Commit 656d7f2

Browse files
committed
Improve tests.
1 parent cc6f326 commit 656d7f2

File tree

3 files changed

+22
-16
lines changed

3 files changed

+22
-16
lines changed

TelemetryServices.OpenTelemetry.Tests/OpenTelemetryProviderBuilderExtensions_should.cs

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ public void Add_activity_source_by_telemetry_name_type()
5151
using var activity = telemetry.ActivitySource.StartActivity("TestActivity");
5252
var recordedActivities = _traceListener.Activities;
5353
Assert.NotNull(activity);
54-
Assert.Contains(recordedActivities, a => a.DisplayName == activity.DisplayName);
54+
Assert.Contains(activity, recordedActivities);
5555
});
5656
}
5757

@@ -64,10 +64,13 @@ public async Task Add_meter_by_telemetry_name_type()
6464
await _hostBuilder.RunInHost(async host =>
6565
{
6666
var telemetry = host.Services.GetRequiredService<ITelemetry<TelemetryName>>();
67-
var meter = telemetry.Meter.CreateCounter<int>("TestCounter");
68-
meter.Add(1);
67+
var counter = telemetry.Meter.CreateCounter<int>("TestCounter");
68+
counter.Add(1);
6969
var recordedMetrics = await _metricListener.WaitForMetrics();
70-
Assert.Contains(recordedMetrics, m => m.Name == meter.Name);
70+
Assert.Contains(recordedMetrics, m =>
71+
m.MeterName == telemetry.Meter.Name &&
72+
m.Name == counter.Name
73+
);
7174
});
7275
}
7376

TelemetryServices.OpenTelemetry.Tests/TelemetryServicesBuilderExtensions_should.cs

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ public void Add_activity_source_to_open_telemetry_provider()
4646
using var activity = telemetry.ActivitySource.StartActivity("TestActivity");
4747
var recordedActivities = _traceListener.Activities;
4848
Assert.NotNull(activity);
49-
Assert.Contains(recordedActivities, a => a.DisplayName == activity.DisplayName);
49+
Assert.Contains(activity, recordedActivities);
5050
});
5151
}
5252

@@ -59,10 +59,13 @@ public async Task Add_meter_to_open_telemetry_provider()
5959
await _hostBuilder.RunInHost(async host =>
6060
{
6161
var telemetry = host.Services.GetRequiredService<ITelemetry<TelemetryName>>();
62-
var meter = telemetry.Meter.CreateCounter<int>("TestCounter");
63-
meter.Add(1);
62+
var counter = telemetry.Meter.CreateCounter<int>("TestCounter");
63+
counter.Add(1);
6464
var recordedMetrics = await _metricListener.WaitForMetrics();
65-
Assert.Contains(recordedMetrics, m => m.Name == meter.Name);
65+
Assert.Contains(recordedMetrics, m =>
66+
m.MeterName == telemetry.Meter.Name &&
67+
m.Name == counter.Name
68+
);
6669
});
6770
}
6871

TelemetryServices.OpenTelemetry.Tests/TestDoubles/OpenTelemetryMetricListener.cs

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,18 +17,18 @@ public OpenTelemetryMetricListener()
1717
}
1818

1919
public async Task<IEnumerable<Metric>> WaitForMetrics(int timeoutMilliseconds = _exportIntervalMilliseconds * 10)
20-
{
21-
using var cts = new CancellationTokenSource(TimeSpan.FromMilliseconds(timeoutMilliseconds));
22-
return await WaitForMetrics(cts.Token);
23-
}
24-
25-
public async Task<IEnumerable<Metric>> WaitForMetrics(CancellationToken cancellationToken)
2620
{
2721
using var timer = new PeriodicTimer(TimeSpan.FromMilliseconds(_exportIntervalMilliseconds));
28-
while (_exporter.Metrics.IsEmpty)
22+
using var cts = new CancellationTokenSource(TimeSpan.FromMilliseconds(timeoutMilliseconds));
23+
do
2924
{
30-
await timer.WaitForNextTickAsync(cancellationToken);
25+
if (!_exporter.Metrics.IsEmpty)
26+
{
27+
return _exporter.Metrics;
28+
}
29+
await timer.WaitForNextTickAsync();
3130
}
31+
while (!cts.Token.IsCancellationRequested);
3232
return _exporter.Metrics;
3333
}
3434

0 commit comments

Comments
 (0)