using System.Linq.Expressions; using System.Net; using System.Net.Http.Json; using NUnit.Framework; using Umbraco.Cms.Api.Management.Controllers.User.Current; using Umbraco.Cms.Api.Management.ViewModels; using Umbraco.Cms.Api.Management.ViewModels.User; using Umbraco.Cms.Core.Services; using Umbraco.Cms.Core; using Umbraco.Cms.Core.Models; using Umbraco.Cms.Core.Models.TemporaryFile; namespace Umbraco.Cms.Tests.Integration.ManagementApi.User.Current; public class SetAvatarCurrentUserControllerTests : ManagementApiUserGroupTestBase { private IUserGroupService UserGroupService => GetRequiredService(); private ITemporaryFileService TemporaryFileService => GetRequiredService(); private IUserService UserService => GetRequiredService(); private Guid _avatarKey; [SetUp] public async Task SetUp() { var adminUserGroup = await UserGroupService.GetAsync(Constants.Security.AdminGroupAlias); var stringKey = Guid.NewGuid(); var model = new UserCreateModel() { Email = stringKey + "@test.com", UserName = stringKey + "@test.com", Name = stringKey.ToString(), UserGroupKeys = new HashSet { adminUserGroup.Key }, }; await UserService.CreateAsync(Constants.Security.SuperUserKey, model); await TemporaryFileService.CreateAsync(new CreateTemporaryFileModel { Key = _avatarKey, FileName = "File.png" }); } protected override Expression> MethodSelector => x => x.SetAvatar(CancellationToken.None, null); protected override UserGroupAssertionModel AdminUserGroupAssertionModel => new() { ExpectedStatusCode = HttpStatusCode.OK }; protected override UserGroupAssertionModel EditorUserGroupAssertionModel => new() { ExpectedStatusCode = HttpStatusCode.OK }; protected override UserGroupAssertionModel SensitiveDataUserGroupAssertionModel => new() { ExpectedStatusCode = HttpStatusCode.OK }; protected override UserGroupAssertionModel TranslatorUserGroupAssertionModel => new() { ExpectedStatusCode = HttpStatusCode.OK }; protected override UserGroupAssertionModel WriterUserGroupAssertionModel => new() { ExpectedStatusCode = HttpStatusCode.OK }; protected override UserGroupAssertionModel UnauthorizedUserGroupAssertionModel => new() { ExpectedStatusCode = HttpStatusCode.Unauthorized }; protected override async Task ClientRequest() { SetAvatarRequestModel setAvatarRequestModel = new() { File = new ReferenceByIdModel(_avatarKey) }; return await Client.PostAsync(Url, JsonContent.Create(setAvatarRequestModel)); } }