test: add base class for armory infrastructure tests

CarlosPavajeau · Oct 20, 2021 · 48354ac · 48354ac
1 parent d1e3889
commit 48354ac
Show file tree

Hide file tree

Showing 2 changed files with 61 additions and 0 deletions.
diff --git a/test/Armory.Test/Armory.Test.csproj b/test/Armory.Test/Armory.Test.csproj
@@ -11,6 +11,12 @@
       <PrivateAssets>all</PrivateAssets>
       <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
     </PackageReference>
+    <PackageReference Include="Microsoft.EntityFrameworkCore" Version="5.0.11"/>
+    <PackageReference Include="Microsoft.EntityFrameworkCore.Design" Version="5.0.11">
+      <PrivateAssets>all</PrivateAssets>
+      <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
+    </PackageReference>
+    <PackageReference Include="Microsoft.EntityFrameworkCore.InMemory" Version="5.0.11"/>
     <PackageReference Include="Microsoft.NET.Test.Sdk" Version="16.11.0"/>
     <PackageReference Include="NUnit" Version="3.13.2"/>
     <PackageReference Include="NUnit3TestAdapter" Version="4.0.0"/>

diff --git a/test/Armory.Test/ArmoryContextInfrastructureTestCase.cs b/test/Armory.Test/ArmoryContextInfrastructureTestCase.cs
@@ -0,0 +1,55 @@
+using System;
+using System.Linq;
+using Armory.Api;
+using Armory.Shared.Helpers;
+using Armory.Shared.Infrastructure.Persistence.EntityFramework;
+using Armory.Shared.Test.Infrastructure;
+using MediatR;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.Extensions.Configuration;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace Armory.Test
+{
+    public class ArmoryContextInfrastructureTestCase : InfrastructureTestCase<Startup>, IDisposable
+    {
+        public void Dispose()
+        {
+            Finish();
+        }
+
+        protected override void Setup()
+        {
+        }
+
+        protected override Action<IServiceCollection> Services()
+        {
+            return services =>
+            {
+                var descriptor =
+                    services.SingleOrDefault(d => d.ServiceType == typeof(DbContextOptions<ArmoryDbContext>));
+                if (descriptor is not null)
+                {
+                    services.Remove(descriptor);
+                }
+
+                var configuration = new ConfigurationBuilder()
+                    .SetBasePath(AppContext.BaseDirectory)
+                    .AddJsonFile("appsettings.json")
+                    .Build();
+
+
+                services.AddAutoMapper(AssemblyHelper.GetInstance(Assemblies.Armory));
+                services.AddMediatR(AssemblyHelper.GetInstance(Assemblies.Armory));
+                services.AddAutoMapper(typeof(Startup));
+
+                var serviceProvider = services.AddEntityFrameworkInMemoryDatabase().BuildServiceProvider();
+                services.AddDbContext<ArmoryDbContext>(options =>
+                {
+                    options.UseInMemoryDatabase(Guid.NewGuid().ToString());
+                    options.UseInternalServiceProvider(serviceProvider);
+                });
+            };
+        }
+    }
+}