AZEN-SGG/Kavita
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Unit tests for ChapterTags

Browse source
This commit is contained in:
Amelia 2025-06-20 19:11:59 +02:00
parent b03b5bfdfc
commit 7147fbf959
1 changed files with 184 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

184
API.Tests/Repository/TagRepositoryTests.cs Normal file
View file

@ -0,0 +1,184 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using API.DTOs.Metadata.Browse;
using API.Entities;
using API.Entities.Enums;
using API.Entities.Metadata;
using API.Helpers;
using API.Helpers.Builders;
using Xunit;
namespace API.Tests.Repository;
public class TagRepositoryTests : AbstractDbTest
{
private static readonly Tag SharedSeriesChaptersTag = new TagBuilder("Shared Series Chapter Tag").Build();
private static readonly Tag SharedSeriesTag = new TagBuilder("Shared Series Tag").Build();
private static readonly Tag SharedChaptersTag = new TagBuilder("Shared Chapters Tag").Build();
private static readonly Tag Lib0SeriesChaptersTag = new TagBuilder("Lib0 Series Chapter Tag").Build();
private static readonly Tag Lib0SeriesTag = new TagBuilder("Lib0 Series Tag").Build();
private static readonly Tag Lib0ChaptersTag = new TagBuilder("Lib0 Chapters Tag").Build();
private static readonly Tag Lib1SeriesChaptersTag = new TagBuilder("Lib1 Series Chapter Tag").Build();
private static readonly Tag Lib1SeriesTag = new TagBuilder("Lib1 Series Tag").Build();
private static readonly Tag Lib1ChaptersTag = new TagBuilder("Lib1 Chapters Tag").Build();
private static readonly Tag Lib1ChapterAgeTag = new TagBuilder("Lib1 Chapter Age Tag").Build();
private static readonly List<Tag> AllTags =
[
SharedSeriesChaptersTag, SharedSeriesTag, SharedChaptersTag,
Lib0SeriesChaptersTag, Lib0SeriesTag, Lib0ChaptersTag,
Lib1SeriesChaptersTag, Lib1SeriesTag, Lib1ChaptersTag, Lib1ChapterAgeTag
];
/**
* Access to lib0, lib1
*/
private AppUser _fullAccess;
/**
* Access to lib1
*/
private AppUser _restrictedAccess;
/**
* Access to lib1, and Teen
*/
private AppUser _restrictedAgeAccess;
protected override async Task ResetDb()
{
Context.Tag.RemoveRange(Context.Tag);
await Context.SaveChangesAsync();
}
private async Task SeedDb()
{
// Create users with different access levels
_fullAccess = new AppUserBuilder("amelia", "amelia@example.com").Build();
_restrictedAccess = new AppUserBuilder("mila", "mila@example.com").Build();
_restrictedAgeAccess = new AppUserBuilder("eva", "eva@example.com").Build();
_restrictedAgeAccess.AgeRestriction = AgeRating.Teen;
_restrictedAgeAccess.AgeRestrictionIncludeUnknowns = true;
Context.Users.Add(_fullAccess);
Context.Users.Add(_restrictedAccess);
Context.Users.Add(_restrictedAgeAccess);
await Context.SaveChangesAsync();
Context.Tag.Add(SharedSeriesChaptersTag);
Context.Tag.Add(SharedSeriesTag);
Context.Tag.Add(SharedChaptersTag);
Context.Tag.Add(Lib0SeriesChaptersTag);
Context.Tag.Add(Lib0SeriesTag);
Context.Tag.Add(Lib0ChaptersTag);
Context.Tag.Add(Lib1SeriesChaptersTag);
Context.Tag.Add(Lib1SeriesTag);
Context.Tag.Add(Lib1ChaptersTag);
await Context.SaveChangesAsync();
var lib0 = new LibraryBuilder("lib0")
.WithSeries(new SeriesBuilder("lib0-s0")
.WithMetadata(new SeriesMetadata
{
Tags = [SharedSeriesChaptersTag, SharedSeriesTag, Lib0SeriesChaptersTag, Lib0SeriesTag]
})
.WithVolume(new VolumeBuilder("1")
.WithChapter(new ChapterBuilder("1")
.WithTags([SharedSeriesChaptersTag, SharedChaptersTag, Lib0SeriesChaptersTag, Lib0ChaptersTag])
.Build())
.WithChapter(new ChapterBuilder("2")
.WithTags([SharedSeriesChaptersTag, SharedChaptersTag, Lib1SeriesChaptersTag, Lib1ChaptersTag])
.Build())
.Build())
.Build())
.Build();
var lib1 = new LibraryBuilder("lib1")
.WithSeries(new SeriesBuilder("lib1-s0")
.WithMetadata(new SeriesMetadata
{
Tags = [SharedSeriesChaptersTag, SharedSeriesTag, Lib1SeriesChaptersTag, Lib1SeriesTag]
})
.WithVolume(new VolumeBuilder("1")
.WithChapter(new ChapterBuilder("1")
.WithTags([SharedSeriesChaptersTag, SharedChaptersTag, Lib1SeriesChaptersTag, Lib1ChaptersTag])
.Build())
.WithChapter(new ChapterBuilder("2")
.WithTags([SharedSeriesChaptersTag, SharedChaptersTag, Lib1SeriesChaptersTag, Lib1ChaptersTag, Lib1ChapterAgeTag])
.WithAgeRating(AgeRating.Mature17Plus)
.Build())
.Build())
.Build())
.Build();
await Context.SaveChangesAsync();
_fullAccess.Libraries.Add(lib0);
_fullAccess.Libraries.Add(lib1);
_restrictedAccess.Libraries.Add(lib1);
_restrictedAgeAccess.Libraries.Add(lib1);
await Context.SaveChangesAsync();
}
private Predicate<BrowseTagDto> ContainsTagCheck(Tag tag)
{
return t => t.Id == tag.Id;
}
[Fact]
public async Task GetBrowseableTag()
{
await SeedDb();
var fullAccessTags = await UnitOfWork.TagRepository.GetBrowseableTag(_fullAccess.Id, new UserParams());
Assert.Equal(AllTags.Count, fullAccessTags.TotalCount);
foreach (var tag in AllTags)
{
Assert.Contains(fullAccessTags, ContainsTagCheck(tag));
}
Assert.Equal(2, fullAccessTags.First(dto => dto.Id == SharedSeriesChaptersTag.Id).SeriesCount);
Assert.Equal(4, fullAccessTags.First(dto => dto.Id == SharedSeriesChaptersTag.Id).ChapterCount);
Assert.Equal(1, fullAccessTags.First(dto => dto.Id == Lib0SeriesTag.Id).SeriesCount);
var restrictedAccessTags = await UnitOfWork.TagRepository.GetBrowseableTag(_restrictedAccess.Id, new UserParams());
// Should see: 3 shared + 4 library 1 specific = 7 tags
Assert.Equal(7, restrictedAccessTags.TotalCount);
// Verify Library 1 and shared tags are present
Assert.Contains(restrictedAccessTags, ContainsTagCheck(SharedSeriesChaptersTag));
Assert.Contains(restrictedAccessTags, ContainsTagCheck(SharedSeriesTag));
Assert.Contains(restrictedAccessTags, ContainsTagCheck(SharedChaptersTag));
Assert.Contains(restrictedAccessTags, ContainsTagCheck(Lib1SeriesChaptersTag));
Assert.Contains(restrictedAccessTags, ContainsTagCheck(Lib1SeriesTag));
Assert.Contains(restrictedAccessTags, ContainsTagCheck(Lib1ChaptersTag));
// Verify Count is correctly limited
Assert.Equal(1, restrictedAccessTags.First(dto => dto.Id == SharedSeriesChaptersTag.Id).SeriesCount);
Assert.Equal(2, restrictedAccessTags.First(dto => dto.Id == SharedSeriesChaptersTag.Id).ChapterCount);
Assert.Equal(1, restrictedAccessTags.First(dto => dto.Id == Lib1SeriesTag.Id).SeriesCount);
var restrictedAgeAccessTags = await UnitOfWork.TagRepository.GetBrowseableTag(_restrictedAgeAccess.Id, new UserParams());
// Should see: 3 shared + 3 library 1 specific = 6 tags
Assert.Equal(6, restrictedAgeAccessTags.TotalCount);
Assert.Contains(restrictedAgeAccessTags, ContainsTagCheck(SharedSeriesChaptersTag));
Assert.Contains(restrictedAgeAccessTags, ContainsTagCheck(SharedSeriesTag));
Assert.Contains(restrictedAgeAccessTags, ContainsTagCheck(Lib1ChaptersTag));
Assert.DoesNotContain(restrictedAgeAccessTags, ContainsTagCheck(Lib1ChapterAgeTag));
// Verify Count is correctly limited
Assert.Equal(1, restrictedAgeAccessTags.First(dto => dto.Id == SharedSeriesChaptersTag.Id).SeriesCount);
Assert.Equal(1, restrictedAgeAccessTags.First(dto => dto.Id == SharedSeriesChaptersTag.Id).ChapterCount);
Assert.Equal(1, restrictedAgeAccessTags.First(dto => dto.Id == Lib1SeriesTag.Id).SeriesCount);
}
}