AspNetCore.Diagnostics.HealthChecks/test/HealthChecks.Network.Tests/Fixtures/DockerMailServerContainerFixture.cs at 8bd2a31e0509de9d2ba5bd2be1330aba297da039 · Xabaril/AspNetCore.Diagnostics.HealthChecks · 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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
using DotNet.Testcontainers.Builders;
using DotNet.Testcontainers.Containers;

namespace HealthChecks.Network.Tests.Fixtures;

public class DockerMailServerContainerFixture : IAsyncLifetime
{
    protected const string Registry = "docker.io";

    protected const string Image = "mailserver/docker-mailserver";

    protected const string Tag = "15.1.0";

    protected const int ExplicitTlsSmtpPort = 587;

    protected const int ImplicitTlsSmtpPort = 465;

    protected const int ExplicitTlsImapPort = 143;

    protected const int ImplicitTlsImapPort = 993;

    protected const string Email = "user@healthchecks.com";

    protected const string Password = "password";

    public IContainer? Container { get; private set; }

    public (string Host, int ExplicitTlsPort, int ImplicitTlsPort, string Username, string Password) GetSmtpConnectionProperties()
    {
        if (Container is null)
        {
            throw new InvalidOperationException("The test container was not initialized.");
        }

        return (
            Container.Hostname,
            Container.GetMappedPublicPort(ExplicitTlsSmtpPort),
            Container.GetMappedPublicPort(ImplicitTlsSmtpPort),
            Email,
            Password
        );
    }

    public (string Host, int ExplicitTlsPort, int ImplicitTlsPort, string Username, string Password) GetImapConnectionProperties()
    {
        if (Container is null)
        {
            throw new InvalidOperationException("The test container was not initialized.");
        }

        return (
            Container.Hostname,
            Container.GetMappedPublicPort(ExplicitTlsImapPort),
            Container.GetMappedPublicPort(ImplicitTlsImapPort),
            Email,
            Password
        );
    }

    public async Task InitializeAsync() => Container = await CreateContainerAsync();

    public Task DisposeAsync() => Container?.DisposeAsync().AsTask() ?? Task.CompletedTask;

    protected virtual ContainerBuilder Configure(ContainerBuilder builder) => builder;

    private async Task<IContainer> CreateContainerAsync()
    {
        var waitStrategy = Wait
            .ForUnixContainer()
            .UntilCommandIsCompleted("setup", "email", "add", Email, Password)
            .UntilMessageIsLogged(".+ is up and running")
            .UntilMessageIsLogged(".+daemon started.+")
            .UntilPortIsAvailable(ImplicitTlsSmtpPort)
            .UntilPortIsAvailable(ExplicitTlsSmtpPort);

        var builder = new ContainerBuilder()
            .WithImage($"{Registry}/{Image}:{Tag}")
            .WithEnvironment("OVERRIDE_HOSTNAME", "mail.beatpulse.com")
            .WithEnvironment("POSTFIX_INET_PROTOCOLS", "ipv4")
            .WithEnvironment("DOVECOT_INET_PROTOCOLS", "ipv4")
            .WithEnvironment("ENABLE_CLAMAV", "0")
            .WithEnvironment("ENABLE_AMAVIS", "0")
            .WithEnvironment("ENABLE_RSPAMD", "0")
            .WithEnvironment("ENABLE_OPENDKIM", "0")
            .WithEnvironment("ENABLE_OPENDMARC", "0")
            .WithEnvironment("ENABLE_POLICYD_SPF", "0")
            .WithEnvironment("ENABLE_SPAMASSASSIN", "0")
            .WithPortBinding(ExplicitTlsSmtpPort, true)
            .WithPortBinding(ImplicitTlsSmtpPort, true)
            .WithPortBinding(ExplicitTlsImapPort, true)
            .WithPortBinding(ImplicitTlsImapPort, true)
            .WithWaitStrategy(waitStrategy);

        builder = Configure(builder);

        var container = builder.Build();

        await container.StartAsync();

        return container;
    }
}

public class SecureDockerMailServerContainerFixture : DockerMailServerContainerFixture
{
    protected override ContainerBuilder Configure(ContainerBuilder builder)
    {
        var certsPath = new DirectoryInfo(Path.Combine(
            Directory.GetCurrentDirectory(),
            "Resources",
            "docker-mailserver",
            "certs"));

        return builder
            .WithEnvironment("SSL_TYPE", "manual")
            .WithResourceMapping(certsPath, "/tmp/docker-mailserver/certs")
            .WithEnvironment("SSL_CERT_PATH", "/tmp/docker-mailserver/certs/public.crt")
            .WithEnvironment("SSL_KEY_PATH", "/tmp/docker-mailserver/certs/private.key");
    }
}