using System;
using Jellyfin.Database.Implementations.Entities;
using Jellyfin.Server.Implementations.Tests.Item;
using Jellyfin.Server.Implementations.Users;
using Microsoft.EntityFrameworkCore.Diagnostics;
using Xunit;
namespace Jellyfin.Server.Implementations.Tests.Users;
///
/// Two first-ever requests for the same preferences can both find nothing and both insert; the unique index lets one
/// through. The other has to return the stored preferences instead of failing.
///
public sealed class DisplayPreferencesManagerConcurrentCreateTests : SqliteDbTestFixture
{
private const string Client = "client";
private static readonly Guid _userId = Guid.Parse("aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa");
private static readonly Guid _itemId = Guid.Parse("bbbbbbbb-bbbb-bbbb-bbbb-bbbbbbbbbbbb");
private readonly CompetingInsert _competingInsert;
private readonly DisplayPreferencesManager _manager;
public DisplayPreferencesManagerConcurrentCreateTests()
: this(new CompetingInsert())
{
}
private DisplayPreferencesManagerConcurrentCreateTests(CompetingInsert competingInsert)
: base(competingInsert)
{
_competingInsert = competingInsert;
_manager = new DisplayPreferencesManager(CreateDbContextFactory());
using var context = CreateDbContext();
context.Users.Add(new User("user", "auth-provider", "reset-provider") { Id = _userId });
context.SaveChanges();
}
[Fact]
public void GetDisplayPreferences_AnotherRequestStoresThemFirst_ReturnsTheStoredPreferences()
{
var storedId = 0;
_competingInsert.Before = () =>
{
using var context = CreateDbContext();
var stored = new DisplayPreferences(_userId, _itemId, Client);
context.DisplayPreferences.Add(stored);
context.SaveChanges();
storedId = stored.Id;
};
var preferences = _manager.GetDisplayPreferences(_userId, _itemId, Client);
Assert.NotEqual(0, storedId);
Assert.Equal(storedId, preferences.Id);
using var check = CreateDbContext();
Assert.Single(check.DisplayPreferences);
}
///
/// Stores the other request's row just before the next save, after the manager has looked for the preferences.
///
private sealed class CompetingInsert : SaveChangesInterceptor
{
public Action? Before { get; set; }
public override InterceptionResult SavingChanges(DbContextEventData eventData, InterceptionResult result)
{
// Disarmed first, because the competing save goes through this interceptor as well.
var before = Before;
Before = null;
before?.Invoke();
return result;
}
}
}