DocumentOrganizer/Diligent.WebAPI.Tests/Helpers.cs

using Diligent.WebAPI.Data;
using Diligent.WebAPI.Data.Entities;
using Microsoft.EntityFrameworkCore;

namespace Diligent.WebAPI.Tests
{
    public static class Helpers<T> where T : class
    {
        public static async Task<DatabaseContext> GetDatabaseContext(List<T> applicants)
        {
            var options = new DbContextOptionsBuilder<DatabaseContext>()
                .UseInMemoryDatabase(databaseName: Guid.NewGuid().ToString())
                .Options;
            var databaseContext = new DatabaseContext(options);
            databaseContext.Database.EnsureCreated();
            if (!await databaseContext.Set<T>().AnyAsync())
            {
                await databaseContext.Set<T>().AddRangeAsync(applicants);
                await databaseContext.SaveChangesAsync();
            }
            return databaseContext;
        }
        public static async Task<DatabaseContext> GetDatabaseContextWithRelation(List<T> users, List<RefreshToken> tokens)
        {
            var options = new DbContextOptionsBuilder<DatabaseContext>()
                .UseInMemoryDatabase(databaseName: Guid.NewGuid().ToString())
                .Options;
            var databaseContext = new DatabaseContext(options);
            databaseContext.Database.EnsureCreated();
            if (!await databaseContext.Set<T>().AnyAsync() || !await databaseContext.Set<RefreshToken>().AnyAsync())
            {
                await databaseContext.Set<T>().AddRangeAsync(users);
                await databaseContext.Set<RefreshToken>().AddRangeAsync(tokens);
                await databaseContext.SaveChangesAsync();
            }
            return databaseContext;
        }
    }
}