DocSpace-buildtools/common/Tests/ASC.Webhooks.Tests/BaseSetUp.cs

// (c) Copyright Ascensio System SIA 2010-2022
//
// This program is a free software product.
// You can redistribute it and/or modify it under the terms
// of the GNU Affero General Public License (AGPL) version 3 as published by the Free Software
// Foundation. In accordance with Section 7(a) of the GNU AGPL its Section 15 shall be amended
// to the effect that Ascensio System SIA expressly excludes the warranty of non-infringement of
// any third-party rights.
//
// This program is distributed WITHOUT ANY WARRANTY, without even the implied warranty
// of MERCHANTABILITY or FITNESS FOR A PARTICULAR  PURPOSE. For details, see
// the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
//
// You can contact Ascensio System SIA at Lubanas st. 125a-25, Riga, Latvia, EU, LV-1021.
//
// The  interactive user interfaces in modified source and object code versions of the Program must
// display Appropriate Legal Notices, as required under Section 5 of the GNU AGPL version 3.
//
// Pursuant to Section 7(b) of the License you must retain the original Product logo when
// distributing the program. Pursuant to Section 7(e) we decline to grant you any rights under
// trademark law for use of our trademarks.
//
// All the Product's GUI elements, including illustrations and icon sets, as well as technical writing
// content are licensed under the terms of the Creative Commons Attribution-ShareAlike 4.0
// International. See the License terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode

using System;
using System.IO;
using System.Net;
using System.Net.Http;
using System.Reflection;
using System.Threading.Tasks;

using ASC.Api.Core;
using ASC.Api.Core.Extensions;
using ASC.Common;
using ASC.Common.Caching;
using ASC.Common.Logging;
using ASC.Common.Utils;
using ASC.Core.Common.EF;
using ASC.Core.Common.EF.Context;
using ASC.Webhooks.Core;
using ASC.Webhooks.Core.EF.Context;

using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;

using NUnit.Framework;

namespace ASC.Webhooks.Tests;

public class BaseSetUp
{
    private IServiceProvider _serviceProvider;
    protected IHost _host;
    protected WebhookSender _webhookSender;
    protected RequestHistory _requestHistory;
    protected Settings _settings;
    protected IHttpClientFactory _httpClientFactory;
    protected string _testConnection = "Server=localhost;Database=onlyoffice_test;User ID=dev;Password=dev;Pooling=true;Character Set=utf8;AutoEnlist=false;SSL Mode=none;AllowPublicKeyRetrieval=True";
    protected static int _port = 8867;

    [OneTimeSetUp]
    public async Task CreateDb()
    {
        var args = new string[] {
            "--pathToConf", Path.Combine( "..", "..", "..", "..", "..", "..", "config"),
            "--ConnectionStrings:default:connectionString", _testConnection,
            "--migration:enabled", "true"};

        await StartHost(args);

        Migrate(_serviceProvider, "ASC.Migrations.MySql");
        Migrate(_serviceProvider, Assembly.GetExecutingAssembly().GetName().Name);
    }

    [OneTimeTearDown]
    public async Task DropDb()
    {
        var scope = _serviceProvider.CreateScope();
        var context = scope.ServiceProvider.GetService<IDbContextFactory<WebhooksDbContext>>().CreateDbContext();
        context.Database.EnsureDeleted();
        await _host.StopAsync();
    }

    private async Task StartHost(string[] args)
    {
        IConfiguration configuration = null;
        _host = await Host.CreateDefaultBuilder(args)
            .ConfigureWebHostDefaults(webBuilder =>
            {
                webBuilder
                .UseKestrel(options =>
                {
                    options.Listen(IPAddress.Loopback, _port);
                })
                .ConfigureAppConfiguration((hostingContext, config) =>
                {
                    configuration = config.Build();
                    var path = configuration["pathToConf"];
                    if (!Path.IsPathRooted(path))
                    {
                        path = Path.GetFullPath(CrossPlatform.PathCombine(hostingContext.HostingEnvironment.ContentRootPath, path));
                    }
                    config.SetBasePath(path);
                    config
                    .AddJsonFile("appsettings.json")
                    .AddCommandLine(args);
                })
                    .ConfigureServices(services =>
                {
                    services.AddControllers();
                    services.AddMemoryCache();
                    services.AddHttpClient();

                    services.AddScoped<EFLoggerFactory>();
                    services.AddBaseDbContextPool<WebhooksDbContext>();
                    services.AddBaseDbContextPool<TenantDbContext>();
                    services.AddBaseDbContextPool<UserDbContext>();
                    services.AddBaseDbContextPool<CoreDbContext>();

                    services.AddDistributedCache(configuration);
                    services.AddDistributedTaskQueue();
                    services.AddCacheNotify(configuration);
                    services.AddAutoMapper(BaseStartup.GetAutoMapperProfileAssemblies());


                    var dIHelper = new DIHelper();
                    dIHelper.Configure(services);
                    dIHelper.TryAdd<DbWorker>();
                    dIHelper.TryAdd<TestController>();
                    dIHelper.TryAdd<WebhookSender>();
                    dIHelper.TryAdd(typeof(ICacheNotify<>), typeof(KafkaCacheNotify<>));
                })
                .Configure(app =>
                {
                    app.UseRouting();
                    app.UseEndpoints(endpoints =>
                    {
                        endpoints.MapControllers();
                    });
                });
            })
            .StartAsync();

        _serviceProvider = _host.Services;
        _webhookSender = _serviceProvider.GetService<WebhookSender>();
        _requestHistory = _serviceProvider.GetService<RequestHistory>();
        _settings = _serviceProvider.GetService<Settings>();
        _httpClientFactory = _serviceProvider.GetService<IHttpClientFactory>();
    }

    private void Migrate(IServiceProvider serviceProvider, string testAssembly = null)
    {

        if (!string.IsNullOrEmpty(testAssembly))
        {
            var configuration = serviceProvider.GetService<IConfiguration>();
            configuration["testAssembly"] = testAssembly;
        }

        using var db = serviceProvider.GetService<IDbContextFactory<WebhooksDbContext>>().CreateDbContext();
        db.Database.Migrate();
    }
}