Migrate HealthChecks.ArangoDb tests to Testcontainers

Author: monofunc

Directory.Packages.props

@@ -103,6 +103,7 @@
     <PackageVersion Include="System.Text.RegularExpressions" Version="4.3.1" />
     <PackageVersion Include="System.Threading.Channels" Version="8.0.0" />
     <PackageVersion Include="Testcontainers" Version="$(TestcontainersVersion)" />
+    <PackageVersion Include="Testcontainers.ArangoDb" Version="$(TestcontainersVersion)" />
     <PackageVersion Include="Testcontainers.ClickHouse" Version="$(TestcontainersVersion)" />
     <PackageVersion Include="Testcontainers.Kafka" Version="$(TestcontainersVersion)" />
     <PackageVersion Include="Testcontainers.Milvus" Version="$(TestcontainersVersion)" />

test/HealthChecks.ArangoDb.Tests/ArangoDbContainerFixture.cs

@@ -0,0 +1,46 @@
+using Testcontainers.ArangoDb;
+
+namespace HealthChecks.ArangoDb.Tests;
+
+public class ArangoDbContainerFixture : IAsyncLifetime
+{
+    private const string Registry = "docker.io";
+
+    private const string Image = "library/arangodb";
+
+    private const string Tag = "3.12.5.2";
+
+    public ArangoDbContainer? Container { get; private set; }
+
+    public ArangoDbOptions GetConnectionOptions()
+    {
+        if (Container is null)
+        {
+            throw new InvalidOperationException("The test container was not initialized.");
+        }
+
+        var options = new ArangoDbOptions
+        {
+            HostUri = Container.GetTransportAddress(),
+            UserName = ArangoDbBuilder.DefaultUsername,
+            Password = ArangoDbBuilder.DefaultPassword,
+        };
+
+        return options;
+    }
+
+    public async Task InitializeAsync() => Container = await CreateContainerAsync();
+
+    public Task DisposeAsync() => Container?.DisposeAsync().AsTask() ?? Task.CompletedTask;
+
+    private async Task<ArangoDbContainer?> CreateContainerAsync()
+    {
+        var container = new ArangoDbBuilder()
+            .WithImage($"{Registry}/{Image}:{Tag}")
+            .Build();
+
+        await container.StartAsync();
+
+        return container;
+    }
+}

test/HealthChecks.ArangoDb.Tests/Functional/ArangoDbHealthCheckTests.cs

@@ -2,22 +2,18 @@
 
 namespace HealthChecks.ArangoDb.Tests.Functional;
 
-public class arangodb_healthcheck_should
+public class arangodb_healthcheck_should(ArangoDbContainerFixture arangoDbFixture) : IClassFixture<ArangoDbContainerFixture>
 {
     [Fact]
     public async Task be_healthy_if_arangodb_is_available()
     {
+        var options = arangoDbFixture.GetConnectionOptions();
+
         var webHostBuilder = new WebHostBuilder()
             .ConfigureServices(services =>
             {
                 services.AddHealthChecks()
-                 .AddArangoDb(_ => new ArangoDbOptions
-                 {
-                     HostUri = "http://localhost:8529/",
-                     Database = "_system",
-                     UserName = "root",
-                     Password = "strongArangoDbPassword"
-                 }, tags: ["arangodb"]);
+                    .AddArangoDb(_ => options, tags: ["arangodb"]);
             })
             .Configure(app =>
             {
@@ -37,25 +33,14 @@ public async Task be_healthy_if_arangodb_is_available()
     [Fact]
     public async Task be_healthy_if_multiple_arango_are_available()
     {
+        var options = arangoDbFixture.GetConnectionOptions();
+
         var webHostBuilder = new WebHostBuilder()
             .ConfigureServices(services =>
             {
                 services.AddHealthChecks()
-                    .AddArangoDb(_ => new ArangoDbOptions
-                    {
-                        HostUri = "http://localhost:8529/",
-                        Database = "_system",
-                        UserName = "root",
-                        Password = "strongArangoDbPassword"
-                    }, tags: ["arango"], name: "1")
-                    .AddArangoDb(_ => new ArangoDbOptions
-                    {
-                        HostUri = "http://localhost:8529/",
-                        Database = "_system",
-                        UserName = "root",
-                        Password = "strongArangoDbPassword",
-                        IsGenerateJwtTokenBasedOnUserNameAndPassword = true
-                    }, tags: ["arango"], name: "2");
+                    .AddArangoDb(_ => options, tags: ["arango"], name: "1")
+                    .AddArangoDb(_ => options, tags: ["arango"], name: "2");
             })
             .Configure(app =>
             {
@@ -75,17 +60,15 @@ public async Task be_healthy_if_multiple_arango_are_available()
     [Fact]
     public async Task be_unhealthy_if_arango_is_not_available()
     {
+        var options = arangoDbFixture.GetConnectionOptions();
+
+        options.Password = "invalid password";
+
         var webHostBuilder = new WebHostBuilder()
             .ConfigureServices(services =>
             {
                 services.AddHealthChecks()
-                 .AddArangoDb(_ => new ArangoDbOptions
-                 {
-                     HostUri = "http://localhost:8529/",
-                     Database = "_system",
-                     UserName = "root",
-                     Password = "invalid password"
-                 }, tags: ["arango"]);
+                 .AddArangoDb(_ => options, tags: ["arango"]);
             })
             .Configure(app =>
             {

test/HealthChecks.ArangoDb.Tests/HealthChecks.ArangoDb.Tests.csproj

@@ -4,4 +4,8 @@
     <ProjectReference Include="..\..\src\HealthChecks.ArangoDb\HealthChecks.ArangoDb.csproj" />
   </ItemGroup>
 
+  <ItemGroup>
+    <PackageReference Include="Testcontainers.ArangoDb" />
+  </ItemGroup>
+
 </Project>