// 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 . 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 { private readonly ICollectionRepository _repository; protected ACollectionTests(RepositoryActivator repositories) : base(repositories) { _repository = Repositories.LibraryManager.CollectionRepository; } [Fact] public async Task CreateWithEmptySlugTest() { Collection collection = TestSample.GetNew(); collection.Slug = string.Empty; await Assert.ThrowsAsync(() => _repository.Create(collection)); } [Fact] public async Task CreateWithNumberSlugTest() { Collection collection = TestSample.GetNew(); collection.Slug = "2"; Collection ret = await _repository.Create(collection); Assert.Equal("2!", ret.Slug); } [Fact] public async Task CreateWithoutNameTest() { Collection collection = TestSample.GetNew(); collection.Name = null; await Assert.ThrowsAsync(() => _repository.Create(collection)); } [Fact] public async Task CreateWithExternalIdTest() { Collection collection = TestSample.GetNew(); collection.ExternalIDs = new[] { new MetadataID { Provider = TestSample.Get(), Link = "link", DataID = "id" }, new MetadataID { Provider = TestSample.GetNew(), Link = "new-provider-link", DataID = "new-id" } }; await _repository.Create(collection); Collection retrieved = await _repository.Get(2); await Repositories.LibraryManager.Load(retrieved, x => x.ExternalIDs); Assert.Equal(2, retrieved.ExternalIDs.Count); KAssert.DeepEqual(collection.ExternalIDs.First(), retrieved.ExternalIDs.First()); KAssert.DeepEqual(collection.ExternalIDs.Last(), retrieved.ExternalIDs.Last()); } [Fact] public async Task EditTest() { Collection value = await _repository.Get(TestSample.Get().Slug); value.Name = "New Title"; value.Images = new Dictionary { [Images.Poster] = "new-poster" }; await _repository.Edit(value, false); 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().Slug); value.ExternalIDs = new[] { new MetadataID { Provider = TestSample.Get(), Link = "link", DataID = "id" }, }; await _repository.Edit(value, false); await using DatabaseContext database = Repositories.Context.New(); Collection retrieved = await database.Collections .Include(x => x.ExternalIDs) .ThenInclude(x => x.Provider) .FirstAsync(); KAssert.DeepEqual(value, retrieved); } [Fact] public async Task AddMetadataTest() { Collection value = await _repository.Get(TestSample.Get().Slug); value.ExternalIDs = new List { new() { Provider = TestSample.Get(), Link = "link", DataID = "id" }, }; await _repository.Edit(value, false); { await using DatabaseContext database = Repositories.Context.New(); Collection retrieved = await database.Collections .Include(x => x.ExternalIDs) .ThenInclude(x => x.Provider) .FirstAsync(); KAssert.DeepEqual(value, retrieved); } value.ExternalIDs.Add(new MetadataID { Provider = TestSample.GetNew(), Link = "link", DataID = "id" }); await _repository.Edit(value, false); { await using DatabaseContext database = Repositories.Context.New(); Collection retrieved = await database.Collections .Include(x => x.ExternalIDs) .ThenInclude(x => x.Provider) .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 ret = await _repository.Search(query); KAssert.DeepEqual(value, ret.First()); } } }