new tests

Author: Oleksii Sokol

ManagedCode.Database.Tests/TestContainers/CosmosTestContainer.cs

@@ -1,40 +1,107 @@
 using System;
+using System.Linq;
 using System.Net.Http;
 using System.Threading.Tasks;
+using Docker.DotNet;
+using Docker.DotNet.Models;
 using DotNet.Testcontainers.Builders;
 using DotNet.Testcontainers.Containers;
 using ManagedCode.Database.Core;
 using ManagedCode.Database.Cosmos;
 using ManagedCode.Database.Tests.Common;
 using Microsoft.Azure.Cosmos;
+using Xunit;
 
 namespace ManagedCode.Database.Tests.TestContainers;
 
-public class CosmosTestContainer : ITestContainer<string, TestCosmosItem>
+[CollectionDefinition(nameof(CosmosTestContainer))]
+public class CosmosTestContainer : ITestContainer<string, TestCosmosItem>,
+    ICollectionFixture<CosmosTestContainer>, IDisposable
 {
-    private static int _port = 20000;
-
-    private readonly CosmosDatabase _database;
-    private readonly TestcontainersContainer _cosmosContainer;
+    private readonly TestcontainersContainer _cosmosTestContainer;
+    private CosmosDatabase _database;
+    private DockerClient _dockerClient;
+    private const string containerName = "cosmosContainer";
+    private const ushort privatePort = 8081;
+    private bool containerExsist = false;
+    private string containerId;
 
     public CosmosTestContainer()
     {
-        var port = ++_port;
-
-        _cosmosContainer = new TestcontainersBuilder<TestcontainersContainer>()
-            .WithImage("mcr.microsoft.com/cosmosdb/linux/azure-cosmos-emulator:latest")
-            .WithPortBinding(port, 8081)
-            .WithEnvironment("AZURE_COSMOS_EMULATOR_PARTITION_COUNT", "30")
+        _cosmosTestContainer = new TestcontainersBuilder<TestcontainersContainer>()
+            .WithImage("mcr.microsoft.com/cosmosdb/linux/azure-cosmos-emulator")
+            .WithName(containerName)
+            .WithExposedPort(8081)
+            .WithExposedPort(10251)
+            .WithExposedPort(10252)
+            .WithExposedPort(10253)
+            .WithExposedPort(10254)
+            .WithExposedPort(10255)
+            .WithPortBinding(8081, 8081)
+            .WithPortBinding(10251, 10251)
+            .WithPortBinding(10252, 10252)
+            .WithPortBinding(10253, 10253)
+            .WithPortBinding(10254, 10254)
+            .WithPortBinding(8081, 8081)
+            .WithEnvironment("AZURE_COSMOS_EMULATOR_PARTITION_COUNT", "1")
+            .WithEnvironment("AZURE_COSMOS_EMULATOR_IP_ADDRESS_OVERRIDE", "127.0.0.1")
             .WithEnvironment("AZURE_COSMOS_EMULATOR_ENABLE_DATA_PERSISTENCE", "false")
-            .WithWaitStrategy(Wait.ForUnixContainer())
+            .WithCleanUp(false)
+            .WithWaitStrategy(Wait.ForUnixContainer()
+                .UntilPortIsAvailable(privatePort))
             .Build();
 
+        _dockerClient = new DockerClientConfiguration().CreateClient();
+    }
+
+    public IDatabaseCollection<string, TestCosmosItem> Collection =>
+        _database.GetCollection<TestCosmosItem>();
+
+    public string GenerateId()
+    {
+        return $"{Guid.NewGuid():N}";
+    }
+
+    public async Task InitializeAsync()
+    {
+        ushort publicPort = privatePort;
+
+        try
+        {
+            await _cosmosTestContainer.StartAsync();
+
+            containerExsist = false;
+        }
+        catch (Exception ex) //TODO catch name already using exception
+        {
+            containerExsist = true;
+        }
+
+        if (!containerExsist)
+        {
+            publicPort = _cosmosTestContainer.GetMappedPublicPort(privatePort);
+            containerId = _cosmosTestContainer.Id;
+        }
+        else
+        {
+            var listContainers = await _dockerClient.Containers.ListContainersAsync(new ContainersListParameters());
+
+            ContainerListResponse containerListResponse = listContainers.FirstOrDefault(container => container.Names.Contains($"/{containerName}"));
+
+            if (containerListResponse != null)
+            {
+                publicPort = containerListResponse.Ports.Single(port => port.PrivatePort == privatePort).PublicPort;
+
+                containerId = containerListResponse.ID;
+            }
+        }
+        
         _database = new CosmosDatabase(new CosmosOptions
         {
             ConnectionString =
-                $"AccountEndpoint=https://localhost:{port};AccountKey=C2y6yDjf5/R+ob0N8A7Cgv30VRDJIWEHLM+4QDU5DE2nQ9nDuVTqobD4b8mGGyPMbIZnqyMsEcaGQy67XIw/Jw==",
+                $"AccountEndpoint=https://localhost:{publicPort}/;AccountKey=C2y6yDjf5/R+ob0N8A7Cgv30VRDJIWEHLM+4QDU5DE2nQ9nDuVTqobD4b8mGGyPMbIZnqyMsEcaGQy67XIw/Jw==",
             DatabaseName = "database",
-            CollectionName = "testContainer",
+            CollectionName = $"testContainer",
             AllowTableCreation = true,
             CosmosClientOptions = new CosmosClientOptions()
             {
@@ -51,25 +118,27 @@ public CosmosTestContainer()
                 ConnectionMode = ConnectionMode.Gateway
             },
         });
-    }
 
-    public IDatabaseCollection<string, TestCosmosItem> Collection =>
-        _database.GetCollection<TestCosmosItem>();
-
-    public string GenerateId()
-    {
-        return $"{Guid.NewGuid():N}";
-    }
 
-    public async Task InitializeAsync()
-    {
-        await _cosmosContainer.StartAsync();
         await _database.InitializeAsync();
     }
 
     public async Task DisposeAsync()
     {
+        // await _database.DeleteAsync();
         await _database.DisposeAsync();
-        await _cosmosContainer.StopAsync();
+
+        /*     _testOutputHelper.WriteLine($"Cosmos container State:{_cosmosContainer.State}");
+             _testOutputHelper.WriteLine("=STOP=");*/
+    }
+
+    public async void Dispose()
+    {
+
+        await _dockerClient.Containers.RemoveContainerAsync(containerId,
+              new ContainerRemoveParameters
+              {
+                  Force = true
+              });
     }
-}
+}