BSSFramework/src/AutomationCore/Framework.AutomationCore.ServiceEnvironment/IntegrationTestBase.cs at a26249332149d22e5b27b5ba330dcac89a8115d4 · Luxoft/BSSFramework · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
using Framework.AutomationCore.Enums;
using Framework.AutomationCore.ServiceEnvironment.Services;
using Framework.AutomationCore.ServiceProviderPool;
using Framework.AutomationCore.Utils.DatabaseUtils;

using Microsoft.Extensions.DependencyInjection;

using Anch.SecuritySystem.Testing;

namespace Framework.AutomationCore;

public abstract class IntegrationTestBase(IServiceProviderPool rootServiceProviderPool)
    : RootServiceProviderContainer.RootServiceProviderContainer(rootServiceProviderPool.Get())
{
    public virtual void Initialize()
    {
        this.ReattachDatabase();
        this.ClearNotifications();
        this.ClearIntegrationEvents();
        this.ResetServices();
    }

    public virtual void Cleanup()
    {
        try
        {
            this.DropDatabaseAfterTest();
            this.CleanupTestEnvironment();
        }
        finally
        {
            this.ReleaseServiceProvider();
        }
    }

    protected virtual void ResetServices()
    {
        this.RootServiceProvider.GetService<IntegrationTestTimeProvider>()?.Reset();
        this.RootServiceProvider.GetService<RootImpersonateServiceState>()?.Reset();
    }

    protected virtual void DropDatabaseAfterTest()
    {
        if (this.AutomationFrameworkSettings.UseLocalDb || this.AutomationFrameworkSettings.TestRunMode == TestRunMode.DefaultRunModeOnEmptyDatabase)
        {
            AssemblyInitializeAndCleanupBase.RunAction("Drop Database", this.DatabaseContext.Drop);
        }
    }

    protected virtual void ReleaseServiceProvider()
        => rootServiceProviderPool.Release(this.RootServiceProvider);

    protected virtual void ReattachDatabase()
    {
        switch (this.AutomationFrameworkSettings.TestRunMode)
        {
            case TestRunMode.DefaultRunModeOnEmptyDatabase:
            case TestRunMode.RestoreDatabaseUsingAttach:
                AssemblyInitializeAndCleanupBase.RunAction("Drop Database", this.DatabaseContext.Drop);
                AssemblyInitializeAndCleanupBase.RunAction("Restore Databases", this.DatabaseContext.AttachDatabase);
                break;
        }
    }

    protected virtual void CleanupTestEnvironment()
    {
    }

    /// <summary>
    /// Отчистка списка нотифицаций
    /// </summary>
    public virtual void ClearNotifications()
    {
    }

    /// <summary>
    /// Отчистка интеграционных евентов
    /// </summary>
    public virtual void ClearIntegrationEvents()
    {
    }
}