Kyoo/back/tests/Kyoo.Tests/Database/TestContext.cs

// Kyoo - A portable and vast media library solution.
// Copyright (c) Kyoo.
//
// See AUTHORS.md and LICENSE file in the project root for full license information.
//
// Kyoo is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// any later version.
//
// Kyoo is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Kyoo. If not, see <https://www.gnu.org/licenses/>.

using System;
using System.Data.Common;
using System.Threading.Tasks;
using Kyoo.Postgresql;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
using Npgsql;
using Xunit;
using Xunit.Abstractions;

namespace Kyoo.Tests
{
	[CollectionDefinition(nameof(Postgresql))]
	public class PostgresCollection : ICollectionFixture<PostgresFixture> { }

	public sealed class PostgresFixture : IDisposable
	{
		private readonly DbContextOptions<DatabaseContext> _options;

		public string Template { get; }

		public string Connection => PostgresTestContext.GetConnectionString(Template);

		public PostgresFixture()
		{
			string id = Guid.NewGuid().ToString().Replace('-', '_');
			Template = $"kyoo_template_{id}";

			_options = new DbContextOptionsBuilder<DatabaseContext>().UseNpgsql(Connection).Options;

			using PostgresContext context = new(_options, null);
			context.Database.Migrate();

			using NpgsqlConnection conn = (NpgsqlConnection)context.Database.GetDbConnection();
			conn.Open();
			conn.ReloadTypes();

			TestSample.FillDatabase(context);
			conn.Close();
		}

		public void Dispose()
		{
			using PostgresContext context = new(_options, null);
			context.Database.EnsureDeleted();
		}
	}

	public sealed class PostgresTestContext : TestContext
	{
		private readonly string _database;
		private readonly DbContextOptions<DatabaseContext> _context;

		public PostgresTestContext(PostgresFixture template, ITestOutputHelper output)
		{
			string id = Guid.NewGuid().ToString().Replace('-', '_');
			_database = $"kyoo_test_{id}";

			using (NpgsqlConnection connection = new(template.Connection))
			{
				connection.Open();
				using NpgsqlCommand cmd =
					new(
						$"CREATE DATABASE {_database} WITH TEMPLATE {template.Template}",
						connection
					);
				cmd.ExecuteNonQuery();
			}

			_context = new DbContextOptionsBuilder<DatabaseContext>()
				.UseNpgsql(GetConnectionString(_database))
				.UseLoggerFactory(
					LoggerFactory.Create(x =>
					{
						x.ClearProviders();
						x.AddXunit(output);
					})
				)
				.EnableSensitiveDataLogging()
				.EnableDetailedErrors()
				.Options;
		}

		public static string GetConnectionString(string database)
		{
			string server = Environment.GetEnvironmentVariable("POSTGRES_HOST") ?? "127.0.0.1";
			string port = Environment.GetEnvironmentVariable("POSTGRES_PORT") ?? "5432";
			string username = Environment.GetEnvironmentVariable("POSTGRES_USER") ?? "KyooUser";
			string password =
				Environment.GetEnvironmentVariable("POSTGRES_PASSWORD") ?? "KyooPassword";
			return $"Server={server};Port={port};Database={database};User ID={username};Password={password};Include Error Detail=true";
		}

		public override void Dispose()
		{
			using DatabaseContext db = New();
			db.Database.EnsureDeleted();
		}

		public override async ValueTask DisposeAsync()
		{
			await using DatabaseContext db = New();
			await db.Database.EnsureDeletedAsync();
		}

		public override DatabaseContext New()
		{
			return new PostgresContext(_context, null);
		}

		public override DbConnection NewConnection()
		{
			return new NpgsqlConnection(GetConnectionString(_database));
		}
	}

	/// <summary>
	/// Class responsible to fill and create in memory databases for unit tests.
	/// </summary>
	public abstract class TestContext : IDisposable, IAsyncDisposable
	{
		/// <summary>
		/// Add an arbitrary data to the test context.
		/// </summary>
		public void Add<T>(T obj)
			where T : class
		{
			using DatabaseContext context = New();
			context.Set<T>().Add(obj);
			context.SaveChanges();
		}

		/// <summary>
		/// Add an arbitrary data to the test context.
		/// </summary>
		public async Task AddAsync<T>(T obj)
			where T : class
		{
			await using DatabaseContext context = New();
			await context.Set<T>().AddAsync(obj);
			await context.SaveChangesAsync();
		}

		/// <summary>
		/// Get a new database context connected to a in memory Sqlite database.
		/// </summary>
		/// <returns>A valid DatabaseContext</returns>
		public abstract DatabaseContext New();

		public abstract DbConnection NewConnection();

		public abstract void Dispose();

		public abstract ValueTask DisposeAsync();
	}
}