Use ContainerFixture for BigQueryContainerTest

0xced · Nov 23, 2024 · 3fe8dc1 · 3fe8dc1
1 parent a404df6
commit 3fe8dc1
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 18 deletions.
diff --git a/tests/Testcontainers.BigQuery.Tests/BigQueryContainerTest.cs b/tests/Testcontainers.BigQuery.Tests/BigQueryContainerTest.cs
@@ -1,19 +1,7 @@
 namespace Testcontainers.BigQuery;
 
-public sealed class BigQueryContainerTest : IAsyncLifetime
+public sealed class BigQueryContainerTest(ContainerFixture<BigQueryBuilder, BigQueryContainer> fixture) : IClassFixture<ContainerFixture<BigQueryBuilder, BigQueryContainer>>
 {
-    private readonly BigQueryContainer _bigQueryContainer = new BigQueryBuilder().Build();
-
-    public Task InitializeAsync()
-    {
-        return _bigQueryContainer.StartAsync();
-    }
-
-    public Task DisposeAsync()
-    {
-        return _bigQueryContainer.DisposeAsync().AsTask();
-    }
-
     [Fact]
     [Trait(nameof(DockerCli.DockerPlatform), nameof(DockerCli.DockerPlatform.Linux))]
     public async Task ExecuteQueryReturnsInsertRow()
@@ -26,7 +14,7 @@ public async Task ExecuteQueryReturnsInsertRow()
         var utcNowWithoutMilliseconds = new DateTime(utcNow.Year, utcNow.Month, utcNow.Day, utcNow.Hour, utcNow.Minute, utcNow.Second, DateTimeKind.Utc);
 
         var bigQueryClientBuilder = new BigQueryClientBuilder();
-        bigQueryClientBuilder.BaseUri = _bigQueryContainer.GetEmulatorEndpoint();
+        bigQueryClientBuilder.BaseUri = fixture.Container.GetEmulatorEndpoint();
         bigQueryClientBuilder.ProjectId = BigQueryBuilder.DefaultProjectId;
         bigQueryClientBuilder.Credential = new Credential();
 

diff --git a/tests/Testcontainers.BigQuery.Tests/Testcontainers.BigQuery.Tests.csproj b/tests/Testcontainers.BigQuery.Tests/Testcontainers.BigQuery.Tests.csproj
@@ -13,6 +13,7 @@
     </ItemGroup>
     <ItemGroup>
         <ProjectReference Include="../../src/Testcontainers.BigQuery/Testcontainers.BigQuery.csproj"/>
+        <ProjectReference Include="../../src/Testcontainers.Xunit/Testcontainers.Xunit.csproj" />
         <ProjectReference Include="../Testcontainers.Commons/Testcontainers.Commons.csproj"/>
     </ItemGroup>
 </Project>
diff --git a/tests/Testcontainers.BigQuery.Tests/Usings.cs b/tests/Testcontainers.BigQuery.Tests/Usings.cs
@@ -1,9 +1,10 @@
 global using System;
 global using System.Linq;
+global using System.Threading;
 global using System.Threading.Tasks;
 global using DotNet.Testcontainers.Commons;
-global using Google.Cloud.BigQuery.V2;
-global using Xunit;
-global using System.Threading;
 global using Google.Apis.Auth.OAuth2;
-global using Google.Apis.Http;
+global using Google.Apis.Http;
+global using Google.Cloud.BigQuery.V2;
+global using Testcontainers.Xunit;
+global using Xunit;