-
-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathPostgreSqlDatabase.cs
41 lines (34 loc) · 1.31 KB
/
PostgreSqlDatabase.cs
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
using Microsoft.Extensions.Hosting;
using Npgsql;
using Respawn;
using Testcontainers.PostgreSql;
namespace TestExamplesDotnet.PostgreSql;
public sealed class PostgreSqlDatabase : IDatabase
{
private readonly IDatabaseInitializer _databaseInitializer;
private readonly RespawnerOptions _respawnerOptions;
private Respawner? _respawner;
private bool _initialized;
public string ConnectionString { get; }
public PostgreSqlDatabase(PostgreSqlContainer container, IDatabaseInitializer databaseInitializer, RespawnerOptions respawnerOptions)
{
_databaseInitializer = databaseInitializer;
_respawnerOptions = respawnerOptions;
ConnectionString = $"Host=127.0.0.1;Port={container.GetMappedPublicPort(5432)};Database={databaseInitializer.GetUniqueDataBaseName()};Username=postgres;Password=postgres;Include Error Detail=true";
}
public void EnsureInitialized(IHost host)
{
if (!_initialized)
{
_databaseInitializer.Initialize(host);
_initialized = true;
}
}
public async Task Clean()
{
await using var conn = new NpgsqlConnection(ConnectionString);
await conn.OpenAsync();
_respawner ??= await Respawner.CreateAsync(conn, _respawnerOptions);
await _respawner.ResetAsync(conn);
}
}