using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Jellyfin.Data;
using Jellyfin.Database.Implementations.Entities;
using Jellyfin.Database.Implementations.Enums;
using Jellyfin.Plugin.WatchedTogether.Configuration;
using Jellyfin.Plugin.WatchedTogether.Services;
using MediaBrowser.Controller.Library;
using Microsoft.Extensions.Logging.Abstractions;
using Moq;
using Xunit;
namespace Jellyfin.Plugin.WatchedTogether.Tests;
///
/// Covers creating a shared account on the fly from a name typed at the login screen.
///
///
/// These tests drive through a stubbed user manager. They rely on
/// configuration, which is set up per test via
/// .
///
[Collection(nameof(PluginTestContext))]
public class DynamicGroupTests
{
private const string AliceHash = "$PBKDF2-SHA512$iterations=210000$A1A1A1A1$AAAAAAAABBBBBBBB";
private const string BobHash = "$PBKDF2-SHA512$iterations=210000$B2B2B2B2$CCCCCCCCDDDDDDDD";
private static User MakeUser(string name, string? password = null, bool disabled = false)
{
var user = new User(name, "Prov", "ResetProv");
if (password is not null)
{
user.Password = password;
}
if (disabled)
{
user.SetPermission(PermissionKind.IsDisabled, true);
}
return user;
}
private sealed record Harness(
DynamicGroupService Service,
Mock Provisioning);
private static Harness MakeService(
IReadOnlyList knownUsers,
params (string Hash, string Password)[] validPairs)
{
var userManager = new Mock();
userManager.Setup(m => m.GetUserByName(It.IsAny()))
.Returns((string n) =>
{
foreach (var u in knownUsers)
{
if (string.Equals(u.Username, n, StringComparison.OrdinalIgnoreCase))
{
return u;
}
}
return null!;
});
var provisioning = new Mock();
var createdShared = MakeUser("created-shared");
provisioning.Setup(p => p.CreateGroupAsync(
It.IsAny>(),
It.IsAny(),
It.IsAny(),
It.IsAny?>()))
.ReturnsAsync((IReadOnlyList ids, string? name, bool _, IReadOnlyList? _) =>
new SharedGroup { SharedUserId = createdShared.Id, MemberUserIds = [.. ids] });
userManager.Setup(m => m.GetUserById(createdShared.Id)).Returns(createdShared);
var service = new DynamicGroupService(
userManager.Object,
provisioning.Object,
new StubCryptoProvider(validPairs),
NullLogger.Instance);
return new Harness(service, provisioning);
}
[Fact]
public async Task TypingTwoMemberNames_WithAMemberPassword_CreatesTheAccount()
{
using var ctx = PluginTestContext.Create();
var alice = MakeUser("alice", AliceHash);
var bob = MakeUser("bob", BobHash);
var h = MakeService([alice, bob], (AliceHash, "alice-pw"));
var result = await h.Service.TryCreateFromLoginAsync("alice+bob", "alice-pw");
Assert.NotNull(result);
h.Provisioning.Verify(
p => p.CreateGroupAsync(
It.Is>(ids => ids.Count == 2),
"alice+bob",
It.IsAny(),
It.IsAny?>()),
Times.Once);
}
[Fact]
public async Task AnyNamedMembersPassword_Works()
{
using var ctx = PluginTestContext.Create();
var alice = MakeUser("alice", AliceHash);
var bob = MakeUser("bob", BobHash);
var h = MakeService([alice, bob], (BobHash, "bob-pw"));
Assert.NotNull(await h.Service.TryCreateFromLoginAsync("alice+bob", "bob-pw"));
}
[Fact]
public async Task WithoutAMatchingPassword_NothingIsCreated()
{
// Otherwise anyone who knows two usernames could conjure a shared account into existence.
using var ctx = PluginTestContext.Create();
var alice = MakeUser("alice", AliceHash);
var bob = MakeUser("bob", BobHash);
var h = MakeService([alice, bob], (AliceHash, "alice-pw"));
Assert.Null(await h.Service.TryCreateFromLoginAsync("alice+bob", "guessing"));
h.Provisioning.VerifyNoOtherCalls();
}
[Fact]
public async Task AnUnknownNamePart_IsRejected()
{
using var ctx = PluginTestContext.Create();
var alice = MakeUser("alice", AliceHash);
var h = MakeService([alice], (AliceHash, "alice-pw"));
Assert.Null(await h.Service.TryCreateFromLoginAsync("alice+nobody", "alice-pw"));
h.Provisioning.VerifyNoOtherCalls();
}
[Fact]
public async Task ADisabledMember_IsRejected()
{
using var ctx = PluginTestContext.Create();
var alice = MakeUser("alice", AliceHash);
var bob = MakeUser("bob", BobHash, disabled: true);
var h = MakeService([alice, bob], (AliceHash, "alice-pw"));
Assert.Null(await h.Service.TryCreateFromLoginAsync("alice+bob", "alice-pw"));
}
[Fact]
public async Task ASingleName_IsNotAGroup()
{
using var ctx = PluginTestContext.Create();
var alice = MakeUser("alice", AliceHash);
var h = MakeService([alice], (AliceHash, "alice-pw"));
Assert.Null(await h.Service.TryCreateFromLoginAsync("alice", "alice-pw"));
}
[Fact]
public async Task TheSameMemberTwice_IsRejected()
{
using var ctx = PluginTestContext.Create();
var alice = MakeUser("alice", AliceHash);
var h = MakeService([alice], (AliceHash, "alice-pw"));
Assert.Null(await h.Service.TryCreateFromLoginAsync("alice+alice", "alice-pw"));
}
[Fact]
public async Task AnExistingSharedAccountNamedPart_IsRejected()
{
// Shared accounts must not nest inside other shared accounts.
using var ctx = PluginTestContext.Create();
var alice = MakeUser("alice", AliceHash);
var existingShared = MakeUser("shared", BobHash);
ctx.Configuration.Groups.Add(new SharedGroup { SharedUserId = existingShared.Id });
var h = MakeService([alice, existingShared], (AliceHash, "alice-pw"));
Assert.Null(await h.Service.TryCreateFromLoginAsync("alice+shared", "alice-pw"));
}
[Fact]
public async Task WhenDisabledInConfiguration_NothingIsCreated()
{
using var ctx = PluginTestContext.Create();
ctx.Configuration.EnableDynamicGroups = false;
var alice = MakeUser("alice", AliceHash);
var bob = MakeUser("bob", BobHash);
var h = MakeService([alice, bob], (AliceHash, "alice-pw"));
Assert.Null(await h.Service.TryCreateFromLoginAsync("alice+bob", "alice-pw"));
}
[Fact]
public async Task AConfiguredSeparator_IsHonoured()
{
using var ctx = PluginTestContext.Create();
ctx.Configuration.NameSeparator = "_";
var alice = MakeUser("alice", AliceHash);
var bob = MakeUser("bob", BobHash);
var h = MakeService([alice, bob], (AliceHash, "alice-pw"));
Assert.NotNull(await h.Service.TryCreateFromLoginAsync("alice_bob", "alice-pw"));
Assert.Null(await h.Service.TryCreateFromLoginAsync("alice+bob", "alice-pw"));
}
[Fact]
public async Task ThreeOrMoreMembers_AreSupported()
{
using var ctx = PluginTestContext.Create();
var alice = MakeUser("alice", AliceHash);
var bob = MakeUser("bob", BobHash);
var carol = MakeUser("carol", BobHash);
var h = MakeService([alice, bob, carol], (AliceHash, "alice-pw"));
Assert.NotNull(await h.Service.TryCreateFromLoginAsync("alice+bob+carol", "alice-pw"));
h.Provisioning.Verify(
p => p.CreateGroupAsync(
It.Is>(ids => ids.Count == 3),
It.IsAny(),
It.IsAny(),
It.IsAny?>()),
Times.Once);
}
}