Kyoo/back/tests/Kyoo.Tests/Database/SpecificTests/CollectionsTests.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.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Kyoo.Abstractions.Controllers;
using Kyoo.Abstractions.Models;
using Kyoo.Postgresql;
using Microsoft.EntityFrameworkCore;
using Xunit;
using Xunit.Abstractions;

namespace Kyoo.Tests.Database
{
	namespace PostgreSQL
	{
		[Collection(nameof(Postgresql))]
		public class CollectionTests : ACollectionTests
		{
			public CollectionTests(PostgresFixture postgres, ITestOutputHelper output)
				: base(new RepositoryActivator(output, postgres)) { }
		}
	}

	public abstract class ACollectionTests : RepositoryTests<Collection>
	{
		private readonly IRepository<Collection> _repository;

		protected ACollectionTests(RepositoryActivator repositories)
			: base(repositories)
		{
			_repository = Repositories.LibraryManager.Collections;
		}

		[Fact]
		public async Task CreateWithEmptySlugTest()
		{
			Collection collection = TestSample.GetNew<Collection>();
			collection.Slug = string.Empty;
			await Assert.ThrowsAsync<ArgumentException>(() => _repository.Create(collection));
		}

		[Fact]
		public async Task CreateWithNumberSlugTest()
		{
			Collection collection = TestSample.GetNew<Collection>();
			collection.Slug = "2";
			Collection ret = await _repository.Create(collection);
			Assert.Equal("2!", ret.Slug);
		}

		[Fact]
		public async Task CreateWithExternalIdTest()
		{
			Collection collection = TestSample.GetNew<Collection>();
			collection.ExternalId = new Dictionary<string, MetadataId>
			{
				["1"] = new() { Link = "link", DataId = "id" },
				["2"] = new() { Link = "new-provider-link", DataId = "new-id" }
			};
			await _repository.Create(collection);

			Collection retrieved = await _repository.Get(2.AsGuid());
			Assert.Equal(2, retrieved.ExternalId.Count);
			KAssert.DeepEqual(collection.ExternalId.First(), retrieved.ExternalId.First());
			KAssert.DeepEqual(collection.ExternalId.Last(), retrieved.ExternalId.Last());
		}

		[Fact]
		public async Task EditTest()
		{
			Collection value = await _repository.Get(TestSample.Get<Collection>().Slug);
			value.Name = "New Title";
			value.Poster = new Image("new-poster");
			await _repository.Edit(value);

			await using DatabaseContext database = Repositories.Context.New();
			Collection retrieved = await database.Collections.FirstAsync();

			KAssert.DeepEqual(value, retrieved);
		}

		[Fact]
		public async Task EditMetadataTest()
		{
			Collection value = await _repository.Get(TestSample.Get<Collection>().Slug);
			value.ExternalId = new Dictionary<string, MetadataId>
			{
				["test"] = new() { Link = "link", DataId = "id" },
			};
			await _repository.Edit(value);

			await using DatabaseContext database = Repositories.Context.New();
			Collection retrieved = await database.Collections.FirstAsync();

			KAssert.DeepEqual(value, retrieved);
		}

		[Fact]
		public async Task AddMetadataTest()
		{
			Collection value = await _repository.Get(TestSample.Get<Collection>().Slug);
			value.ExternalId = new Dictionary<string, MetadataId>
			{
				["toto"] = new() { Link = "link", DataId = "id" },
			};
			await _repository.Edit(value);

			{
				await using DatabaseContext database = Repositories.Context.New();
				Collection retrieved = await database.Collections.FirstAsync();

				KAssert.DeepEqual(value, retrieved);
			}

			value.ExternalId.Add("test", new MetadataId { Link = "link", DataId = "id" });
			await _repository.Edit(value);

			{
				await using DatabaseContext database = Repositories.Context.New();
				Collection retrieved = await database.Collections.FirstAsync();

				KAssert.DeepEqual(value, retrieved);
			}
		}

		[Theory]
		[InlineData("test")]
		[InlineData("super")]
		[InlineData("title")]
		[InlineData("TiTlE")]
		[InlineData("SuPeR")]
		public async Task SearchTest(string query)
		{
			Collection value = new() { Slug = "super-test", Name = "This is a test title", };
			await _repository.Create(value);
			ICollection<Collection> ret = await _repository.Search(query);
			KAssert.DeepEqual(value, ret.First());
		}
	}
}