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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using Jellyfin.Database.Implementations;
using Jellyfin.Server.Implementations.Item;
using MediaBrowser.Controller;
using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Library;
using MediaBrowser.Model.Configuration;
using MediaBrowser.Model.Entities;
using Microsoft.Extensions.Logging.Abstractions;
using Moq;
using Xunit;
namespace Jellyfin.Server.Implementations.Tests.Item;
public sealed class ItemPersistenceOwnedRowTests : SqliteDbTestFixture
{
private readonly ItemPersistenceService _service;
private readonly ILibraryManager? _previousLibraryManager;
private readonly IServerConfigurationManager? _previousConfigurationManager;
public ItemPersistenceOwnedRowTests()
{
// BaseItem resolves these through process-wide statics; restored in Dispose.
_previousLibraryManager = BaseItem.LibraryManager;
_previousConfigurationManager = BaseItem.ConfigurationManager;
var libraryManager = new Mock<ILibraryManager>();
libraryManager.Setup(l => l.GetCollectionFolders(It.IsAny<BaseItem>()))
.Returns([]);
BaseItem.LibraryManager = libraryManager.Object;
var configurationManager = new Mock<IServerConfigurationManager>();
configurationManager.Setup(c => c.Configuration).Returns(new ServerConfiguration());
BaseItem.ConfigurationManager = configurationManager.Object;
_service = new ItemPersistenceService(
CreateDbContextFactory(),
new Mock<IServerApplicationHost>().Object,
NullLogger<ItemPersistenceService>.Instance);
}
protected override void Dispose(bool disposing)
{
BaseItem.LibraryManager = _previousLibraryManager!;
BaseItem.ConfigurationManager = _previousConfigurationManager!;
base.Dispose(disposing);
}
[Fact]
public void SaveItems_UpdateExistingItem_ReplacesOwnedRows()
{
var id = Guid.Parse("aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa");
_service.SaveItems(
[CreateBook(id, new() { ["Imdb"] = "tt0001", ["Tmdb"] = "555" }, [MetadataField.Name])],
CancellationToken.None);
using (var ctx = CreateDbContext())
{
Assert.Equal(2, ctx.BaseItemProviders.Count(e => e.ItemId.Equals(id)));
Assert.Equal(1, ctx.BaseItemImageInfos.Count(e => e.ItemId.Equals(id)));
Assert.Equal(1, ctx.BaseItemMetadataFields.Count(e => e.ItemId.Equals(id)));
}
// Re-save with different owned rows: the update path rewrites all three tables wholesale.
_service.SaveItems(
[CreateBook(id, new() { ["Imdb"] = "tt9999" }, [MetadataField.Name, MetadataField.Genres])],
CancellationToken.None);
using (var ctx = CreateDbContext())
{
var providers = ctx.BaseItemProviders.Where(e => e.ItemId.Equals(id)).ToList();
Assert.Equal("tt9999", Assert.Single(providers).ProviderValue);
Assert.Equal(1, ctx.BaseItemImageInfos.Count(e => e.ItemId.Equals(id)));
Assert.Equal(2, ctx.BaseItemMetadataFields.Count(e => e.ItemId.Equals(id)));
}
}
[Fact]
public void SaveItems_MixedNewAndExistingBatch_ReplacesOnlyExistingOwnedRows()
{
var existing = Guid.Parse("bbbbbbbb-bbbb-bbbb-bbbb-bbbbbbbbbbbb");
var fresh = Guid.Parse("cccccccc-cccc-cccc-cccc-cccccccccccc");
_service.SaveItems([CreateBook(existing, new() { ["Imdb"] = "tt0001" }, [])], CancellationToken.None);
// One already-persisted item and one brand new item in the same batch.
_service.SaveItems(
[
CreateBook(existing, new() { ["Imdb"] = "tt0002" }, []),
CreateBook(fresh, new() { ["Tmdb"] = "777" }, [])
],
CancellationToken.None);
using var ctx = CreateDbContext();
Assert.Equal("tt0002", Assert.Single(ctx.BaseItemProviders.Where(e => e.ItemId.Equals(existing))).ProviderValue);
Assert.Equal("777", Assert.Single(ctx.BaseItemProviders.Where(e => e.ItemId.Equals(fresh))).ProviderValue);
}
private static Book CreateBook(Guid id, Dictionary<string, string> providerIds, MetadataField[] lockedFields)
{
var book = new Book
{
Id = id,
Name = "Book",
ProviderIds = providerIds,
LockedFields = lockedFields
};
book.SetImage(new ItemImageInfo { Path = "/img/primary.jpg", Type = ImageType.Primary }, 0);
return book;
}
}
|
