Skip to content

Commit 5eb5b81

Browse files
committed
test(streaming): scope SQS Aspire environments
1 parent 757870b commit 5eb5b81

1 file changed

Lines changed: 13 additions & 10 deletions

File tree

test/Extensions/Orleans.AWS.Tests/Streaming/SqsStreamingResourceTests.cs

Lines changed: 13 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -127,16 +127,19 @@ public async Task WithSqsStreaming_ReturnsOrleansServiceAndActivatesSiloAndClien
127127
var client = builder.AddContainer("client", "unused").WithReference(orleans.AsClient());
128128
await using var application = builder.Build();
129129

130-
using var siloEnvironment = new EnvironmentVariableScope(
131-
await GetEnvironmentAsync(application.Services, silo.Resource));
132-
using var siloHost = Host.CreateApplicationBuilder().UseOrleans().Build();
133-
AssertActivatedProvider(siloHost.Services, partitionCount: 2, fifoQueue: true);
134-
135-
siloEnvironment.Dispose();
136-
using var clientEnvironment = new EnvironmentVariableScope(
137-
await GetEnvironmentAsync(application.Services, client.Resource));
138-
using var clientHost = Host.CreateApplicationBuilder().UseOrleansClient().Build();
139-
AssertActivatedProvider(clientHost.Services, partitionCount: 2, fifoQueue: true);
130+
using (new EnvironmentVariableScope(
131+
await GetEnvironmentAsync(application.Services, silo.Resource)))
132+
{
133+
using var siloHost = Host.CreateApplicationBuilder().UseOrleans().Build();
134+
AssertActivatedProvider(siloHost.Services, partitionCount: 2, fifoQueue: true);
135+
}
136+
137+
using (new EnvironmentVariableScope(
138+
await GetEnvironmentAsync(application.Services, client.Resource)))
139+
{
140+
using var clientHost = Host.CreateApplicationBuilder().UseOrleansClient().Build();
141+
AssertActivatedProvider(clientHost.Services, partitionCount: 2, fifoQueue: true);
142+
}
140143
}
141144

142145
[Theory]

0 commit comments

Comments
 (0)