using System.Text;
using HtmlAgilityPack;
using Microsoft.Extensions.DependencyInjection;
using Moq;
using NUnit.Framework;
using Umbraco.Cms.Core;
using Umbraco.Cms.Core.Models.PublishedContent;
using Umbraco.Cms.Core.Models.TemporaryFile;
using Umbraco.Cms.Core.PropertyEditors;
using Umbraco.Cms.Core.PublishedCache;
using Umbraco.Cms.Core.Routing;
using Umbraco.Cms.Core.Services;
using Umbraco.Cms.Core.Web;
using Umbraco.Cms.Tests.Common.Testing;
using Umbraco.Cms.Tests.Integration.Testing;
namespace Umbraco.Cms.Tests.Integration.Umbraco.Infrastructure.PropertyEditors;
[TestFixture]
[UmbracoTest(Database = UmbracoTestOptions.Database.NewSchemaPerTest)]
internal sealed class RichTextEditorPastedImagesTests : UmbracoIntegrationTest
{
private static readonly Guid GifFileKey = Guid.Parse("E625C7FA-6CA7-4A01-92CD-FB5C6F89973D");
private static readonly Guid SvgFileKey = Guid.Parse("0E3A7DFE-DF09-4C3B-881C-E1B815A4502F");
protected override void ConfigureTestServices(IServiceCollection services)
{
// mock out the temporary file service so we don't have to read/write files from/to disk
var temporaryFileServiceMock = new Mock();
temporaryFileServiceMock
.Setup(t => t.GetAsync(GifFileKey))
.Returns(Task.FromResult(new TemporaryFileModel
{
AvailableUntil = DateTime.UtcNow.AddDays(1),
FileName = "the-pixel.gif",
Key = GifFileKey,
OpenReadStream = () => new MemoryStream(Convert.FromBase64String("R0lGODlhAQABAIAAAAAAAP///yH5BAEAAAAALAAAAAABAAEAAAIBRAA7"))
}));
temporaryFileServiceMock
.Setup(t => t.GetAsync(SvgFileKey))
.Returns(Task.FromResult(new TemporaryFileModel
{
AvailableUntil = DateTime.UtcNow.AddDays(1),
FileName = "the-vector.svg",
Key = SvgFileKey,
OpenReadStream = () => new MemoryStream(Encoding.UTF8.GetBytes(@""))
}));
services.AddUnique(temporaryFileServiceMock.Object);
// the integration tests do not really play nice with published content, so we need to mock a fair bit in order to generate media URLs
var publishedMediaTypeMock = new Mock();
publishedMediaTypeMock.SetupGet(c => c.ItemType).Returns(PublishedItemType.Media);
var publishedMediaMock = new Mock();
publishedMediaMock.SetupGet(m => m.ContentType).Returns(publishedMediaTypeMock.Object);
var publishedMediaCacheMock = new Mock();
publishedMediaCacheMock.Setup(mc => mc.GetById(It.IsAny())).Returns(publishedMediaMock.Object);
var umbracoContextMock = new Mock();
umbracoContextMock.SetupGet(c => c.Media).Returns(publishedMediaCacheMock.Object);
var umbracoContext = umbracoContextMock.Object;
var umbracoContextAccessor = new Mock();
umbracoContextAccessor.Setup(ca => ca.TryGetUmbracoContext(out umbracoContext)).Returns(true);
services.AddUnique(umbracoContextAccessor.Object);
var publishedUrlProviderMock = new Mock();
publishedUrlProviderMock
.Setup(pu => pu.GetMediaUrl(It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny()))
.Returns("the-media-url");
services.AddUnique(publishedUrlProviderMock.Object);
}
[Test]
public async Task Can_Handle_Temp_Gif_Image()
{
var html = $"
";
var subject = Services.GetRequiredService();
var result = await subject.FindAndPersistPastedTempImagesAsync(html, Guid.Empty, Constants.Security.SuperUserKey);
AssertContainsMedia(result, Constants.Conventions.MediaTypes.Image);
}
[Test]
public async Task Can_Handle_Temp_Svg_Image()
{
var html = $"
";
var subject = Services.GetRequiredService();
var result = await subject.FindAndPersistPastedTempImagesAsync(html, Guid.Empty, Constants.Security.SuperUserKey);
AssertContainsMedia(result, Constants.Conventions.MediaTypes.VectorGraphicsAlias);
}
[Test]
public async Task Ignores_Non_Existing_Temp_Image()
{
var key = Guid.NewGuid();
var html = $"
";
var subject = Services.GetRequiredService();
var result = await subject.FindAndPersistPastedTempImagesAsync(html, Guid.Empty, Constants.Security.SuperUserKey);
Assert.AreEqual(html, result);
}
[Test]
public async Task Can_Handle_Multiple_Temp_Images()
{
var html = $"
";
var subject = Services.GetRequiredService();
var result = await subject.FindAndPersistPastedTempImagesAsync(html, Guid.Empty, Constants.Security.SuperUserKey);
var htmlDoc = new HtmlDocument();
htmlDoc.LoadHtml(result);
var imageNodes = htmlDoc.DocumentNode.SelectNodes("//img");
Assert.AreEqual(2, imageNodes.Count);
var udis = imageNodes.Select(imageNode => UdiParser.Parse(imageNode.Attributes["data-udi"].Value)).OfType().ToArray();
Assert.AreEqual(2, udis.Length);
Assert.AreNotEqual(udis.First().Guid, udis.Last().Guid);
var mediaService = Services.GetRequiredService();
Assert.Multiple(() =>
{
Assert.IsNotNull(mediaService.GetById(udis.First().Guid));
Assert.IsNotNull(mediaService.GetById(udis.Last().Guid));
});
}
[Test]
public async Task Does_Not_Create_Duplicates_Of_The_Same_Temp_Image()
{
var html = $"
";
var subject = Services.GetRequiredService();
var result = await subject.FindAndPersistPastedTempImagesAsync(html, Guid.Empty, Constants.Security.SuperUserKey);
var htmlDoc = new HtmlDocument();
htmlDoc.LoadHtml(result);
var imageNodes = htmlDoc.DocumentNode.SelectNodes("//img");
Assert.AreEqual(2, imageNodes.Count);
var udis = imageNodes.Select(imageNode => UdiParser.Parse(imageNode.Attributes["data-udi"].Value)).OfType().ToArray();
Assert.AreEqual(2, udis.Length);
Assert.AreEqual(udis.First().Guid, udis.Last().Guid);
var mediaService = Services.GetRequiredService();
Assert.IsNotNull(mediaService.GetById(udis.First().Guid));
}
private void AssertContainsMedia(string result, string expectedMediaTypeAlias)
{
Assert.IsFalse(result.Contains("data-tmpimg"));
var htmlDoc = new HtmlDocument();
htmlDoc.LoadHtml(result);
var imageNode = htmlDoc.DocumentNode.SelectNodes("//img").FirstOrDefault();
Assert.IsNotNull(imageNode);
Assert.IsTrue(imageNode.Attributes.Contains("src"));
Assert.AreEqual("the-media-url", imageNode.Attributes["src"].Value);
Assert.IsTrue(imageNode.Attributes.Contains("data-udi"));
Assert.IsTrue(UdiParser.TryParse(imageNode.Attributes["data-udi"].Value, out GuidUdi udi));
Assert.AreEqual(Constants.UdiEntityType.Media, udi.EntityType);
var media = Services.GetRequiredService().GetById(udi.Guid);
Assert.IsNotNull(media);
Assert.AreEqual(expectedMediaTypeAlias, media.ContentType.Alias);
}
}