Skip to content

Commit

Permalink
Fixed flaky tests.
Browse files Browse the repository at this point in the history
  • Loading branch information
artempushkin committed Nov 27, 2024
1 parent 2ce231a commit 01ee670
Show file tree
Hide file tree
Showing 5 changed files with 18 additions and 17 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ public async Task ActivityListeners_ControlsActivityCreation()
// check that Activity from Diagnostic listener also captured
using (DiagnosticListener listener = new(nameof(ActivityListeners_ControlsActivityCreation)))
{
Activity activity = new("DiagnosticsListenerActivity");
using Activity activity = new("DiagnosticsListenerActivity");

listener.StartActivity(activity, null);
mockStartObserver.Verify(s => s.OnStart(activity, null), Times.Once);
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -179,15 +179,14 @@ public void SendActivityMetric_WithSendParentName_ProducesMetricPointWithParentN
Listener listener = new();

string? parentName = hasParentName ? nameof(parentName) : null;
if (hasParentActivity)
{
Activity parent = new(parentName!);
parent.Start();
}
using Activity? parent = hasParentActivity ? new(parentName!) : null;
parent?.Start();

Activity activity = new(nameof(activity));
using Activity activity = new(nameof(activity));
activity.Start().Stop();

parent?.Stop();

sender.SendActivityMetric(activity);

// 3. Assert
Expand Down
4 changes: 2 additions & 2 deletions tests/Activities.UnitTests/Internal/ActivityObserverTests.cs
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ public class ActivityObserverTests
[TestMethod]
public void OnStop_CallsLogActivityStop()
{
Activity activity = new(nameof(OnStop_CallsLogActivityStop));
using Activity activity = new(nameof(OnStop_CallsLogActivityStop));
(ActivityObserver observer, _, Mock<IActivitiesEventSender> senderMock) = CreateObserver();

observer.OnStop(activity, null);
Expand Down Expand Up @@ -53,7 +53,7 @@ public void OnStart_LogsStart()
[DataRow(false, ActivityResult.ExpectedError)]
public void OnStop_LogsStop(bool isSuccesful, ActivityResult? result)
{
Activity activity = new Activity(nameof(OnStop_LogsStop))
using Activity activity = new Activity(nameof(OnStop_LogsStop))
.Start()
.SetBaggage("SomeValue", "BaggageValue")
.SetTag("SomeTag", "TagValue");
Expand Down
12 changes: 6 additions & 6 deletions tests/Activities.UnitTests/ServiceCollectionTests.cs
Original file line number Diff line number Diff line change
@@ -1,16 +1,16 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT license.

using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using System.Diagnostics;
using System;
using Microsoft.Omex.Extensions.Abstractions.Activities.Processing;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Omex.Extensions.Abstractions.Activities.Processing;
using Microsoft.Omex.Extensions.Testing.Helpers;
using Microsoft.VisualStudio.TestTools.UnitTesting;

namespace Microsoft.Omex.Extensions.Activities.UnitTests
{
Expand Down Expand Up @@ -45,7 +45,7 @@ public void AddOmexActivitySource_ActivityCreationEnabled()
{
Task task = CreateHost().RunAsync();

Activity? activity = new ActivitySource("Source")
using Activity? activity = new ActivitySource("Source")
.StartActivity(nameof(AddOmexActivitySource_HostedServicesRegistered));

NullableAssert.IsNotNull(activity, "Activity creation enabled after host started");
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -34,8 +34,10 @@ public void BuildStatelessWebService_UseUniqueServiceUrl_Failing()
[TestMethod]
public async Task BuildStatelessWebService_RegisterListeners()
{
(string name, int port) httpListener1 = ("httpListener", 20080);
(string name, int port) httpListener2 = ("httpsListener", 20443);
// Use random ports from private range.
Random random = new Random();
(string name, int port) httpListener1 = ("httpListener", random.Next(49152, 65535));
(string name, int port) httpListener2 = ("httpsListener", random.Next(49152, 65535));

SfConfigurationProviderHelper.SetPublishAddress();
SfConfigurationProviderHelper.SetPortVariable(httpListener1.name, httpListener1.port);
Expand Down

0 comments on commit 01ee670

Please sign in to comment.