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); } }