2020-07-21 16:27:12 -07:00
|
|
|
|
// Copyright (c) Microsoft Corporation
|
|
|
|
|
|
// The Microsoft Corporation licenses this file to you under the MIT license.
|
|
|
|
|
|
// See the LICENSE file in the project root for more information.
|
|
|
|
|
|
|
|
|
|
|
|
using System;
|
|
|
|
|
|
using System.Drawing;
|
|
|
|
|
|
using System.Runtime.InteropServices;
|
|
|
|
|
|
using System.Runtime.InteropServices.ComTypes;
|
|
|
|
|
|
using System.Text;
|
|
|
|
|
|
using Common.ComInterlop;
|
2020-12-11 09:59:42 +01:00
|
|
|
|
using Microsoft.PowerToys.STATestExtension;
|
2020-09-21 19:44:12 -07:00
|
|
|
|
using Microsoft.PowerToys.ThumbnailHandler.Svg;
|
2020-12-11 09:59:42 +01:00
|
|
|
|
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
2020-07-30 17:36:39 -07:00
|
|
|
|
using Moq;
|
2020-07-21 16:27:12 -07:00
|
|
|
|
|
|
|
|
|
|
namespace SvgThumbnailProviderUnitTests
|
|
|
|
|
|
{
|
2020-12-11 09:59:42 +01:00
|
|
|
|
[STATestClass]
|
2020-07-21 16:27:12 -07:00
|
|
|
|
public class SvgThumbnailProviderTests
|
|
|
|
|
|
{
|
|
|
|
|
|
[TestMethod]
|
2020-08-21 17:23:42 -07:00
|
|
|
|
public void LoadSimpleSVGShouldReturnNonNullBitmap()
|
2020-07-21 16:27:12 -07:00
|
|
|
|
{
|
|
|
|
|
|
var svgBuilder = new StringBuilder();
|
|
|
|
|
|
svgBuilder.AppendLine("<svg viewBox=\"0 0 100 100\" xmlns=\"http://www.w3.org/2000/svg\">");
|
|
|
|
|
|
svgBuilder.AppendLine("\t<circle cx=\"50\" cy=\"50\" r=\"50\">");
|
|
|
|
|
|
svgBuilder.AppendLine("\t</circle>");
|
|
|
|
|
|
svgBuilder.AppendLine("</svg>");
|
|
|
|
|
|
|
2022-12-14 13:37:23 +01:00
|
|
|
|
SvgThumbnailProvider svgThumbnailProvider = new SvgThumbnailProvider(null);
|
2022-04-14 17:27:22 +02:00
|
|
|
|
Bitmap thumbnail = svgThumbnailProvider.GetThumbnail(svgBuilder.ToString(), 256);
|
|
|
|
|
|
|
|
|
|
|
|
Assert.IsNotNull(thumbnail);
|
|
|
|
|
|
Assert.IsTrue(thumbnail.Width > 0);
|
|
|
|
|
|
Assert.IsTrue(thumbnail.Height > 0);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-21 17:23:42 -07:00
|
|
|
|
public void CheckBlockedElementsShouldReturnNonNullBitmapIfBlockedElementsIsPresentInNestedLevel()
|
2020-07-21 16:27:12 -07:00
|
|
|
|
{
|
|
|
|
|
|
var svgBuilder = new StringBuilder();
|
|
|
|
|
|
svgBuilder.AppendLine("<svg viewBox=\"0 0 100 100\" xmlns=\"http://www.w3.org/2000/svg\">");
|
|
|
|
|
|
svgBuilder.AppendLine("\t<circle cx=\"50\" cy=\"50\" r=\"50\">");
|
|
|
|
|
|
svgBuilder.AppendLine("\t\t<script>alert(\"valid-message\")</script>");
|
|
|
|
|
|
svgBuilder.AppendLine("\t</circle>");
|
|
|
|
|
|
svgBuilder.AppendLine("</svg>");
|
|
|
|
|
|
|
2022-12-14 13:37:23 +01:00
|
|
|
|
SvgThumbnailProvider svgThumbnailProvider = new SvgThumbnailProvider(null);
|
2022-04-14 17:27:22 +02:00
|
|
|
|
Bitmap thumbnail = svgThumbnailProvider.GetThumbnail(svgBuilder.ToString(), 256);
|
2020-07-22 13:27:58 -07:00
|
|
|
|
Assert.IsTrue(thumbnail != null);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-21 17:23:42 -07:00
|
|
|
|
public void CheckNoSvgShouldReturnNullBitmap()
|
2020-07-21 16:27:12 -07:00
|
|
|
|
{
|
|
|
|
|
|
var svgBuilder = new StringBuilder();
|
|
|
|
|
|
svgBuilder.AppendLine("<p>foo</p>");
|
|
|
|
|
|
|
2022-12-14 13:37:23 +01:00
|
|
|
|
SvgThumbnailProvider svgThumbnailProvider = new SvgThumbnailProvider(null);
|
2022-04-14 17:27:22 +02:00
|
|
|
|
Bitmap thumbnail = svgThumbnailProvider.GetThumbnail(svgBuilder.ToString(), 256);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
Assert.IsTrue(thumbnail == null);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-21 17:23:42 -07:00
|
|
|
|
public void CheckNoSvgEmptyStringShouldReturnNullBitmap()
|
2020-07-21 16:27:12 -07:00
|
|
|
|
{
|
2022-12-14 13:37:23 +01:00
|
|
|
|
SvgThumbnailProvider svgThumbnailProvider = new SvgThumbnailProvider(null);
|
2022-04-14 17:27:22 +02:00
|
|
|
|
Bitmap thumbnail = svgThumbnailProvider.GetThumbnail(string.Empty, 256);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
Assert.IsTrue(thumbnail == null);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-21 17:23:42 -07:00
|
|
|
|
public void CheckNoSvgNullStringShouldReturnNullBitmap()
|
2020-07-21 16:27:12 -07:00
|
|
|
|
{
|
2022-12-14 13:37:23 +01:00
|
|
|
|
SvgThumbnailProvider svgThumbnailProvider = new SvgThumbnailProvider(null);
|
2022-04-14 17:27:22 +02:00
|
|
|
|
Bitmap thumbnail = svgThumbnailProvider.GetThumbnail(null, 256);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
Assert.IsTrue(thumbnail == null);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-21 17:23:42 -07:00
|
|
|
|
public void CheckZeroSizedThumbnailShouldReturnNullBitmap()
|
2020-07-21 16:27:12 -07:00
|
|
|
|
{
|
|
|
|
|
|
string content = "<svg></svg>";
|
2022-12-14 13:37:23 +01:00
|
|
|
|
SvgThumbnailProvider svgThumbnailProvider = new SvgThumbnailProvider(null);
|
2022-04-14 17:27:22 +02:00
|
|
|
|
Bitmap thumbnail = svgThumbnailProvider.GetThumbnail(content, 0);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
Assert.IsTrue(thumbnail == null);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-21 17:23:42 -07:00
|
|
|
|
public void CheckBlockedElementsShouldReturnBitmapHTMLWrapped()
|
2020-07-21 16:27:12 -07:00
|
|
|
|
{
|
|
|
|
|
|
var svgBuilder = new StringBuilder();
|
|
|
|
|
|
svgBuilder.AppendLine("<html>");
|
|
|
|
|
|
svgBuilder.AppendLine("<head>");
|
|
|
|
|
|
svgBuilder.AppendLine("<meta http-equiv=\"X-UA-Compatible\" content=\"IE=Edge\">");
|
|
|
|
|
|
svgBuilder.AppendLine("<meta http-equiv=\"Content-Type\" content=\"text/html\" charset=\"utf-8\">");
|
|
|
|
|
|
svgBuilder.AppendLine("</head>");
|
|
|
|
|
|
svgBuilder.AppendLine("<body>");
|
|
|
|
|
|
svgBuilder.AppendLine("<svg viewBox=\"0 0 100 100\" xmlns=\"http://www.w3.org/2000/svg\">");
|
|
|
|
|
|
svgBuilder.AppendLine("<circle cx=\"50\" cy=\"50\" r=\"50\">");
|
|
|
|
|
|
svgBuilder.AppendLine("</circle>");
|
|
|
|
|
|
svgBuilder.AppendLine("</svg>");
|
|
|
|
|
|
svgBuilder.AppendLine("</body>");
|
|
|
|
|
|
svgBuilder.AppendLine("</html>");
|
|
|
|
|
|
|
2022-12-14 13:37:23 +01:00
|
|
|
|
SvgThumbnailProvider svgThumbnailProvider = new SvgThumbnailProvider(null);
|
2022-04-14 17:27:22 +02:00
|
|
|
|
Bitmap thumbnail = svgThumbnailProvider.GetThumbnail(svgBuilder.ToString(), 256);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
Assert.IsTrue(thumbnail != null);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-21 17:23:42 -07:00
|
|
|
|
public void GetThumbnailValidStreamSVG()
|
2020-07-21 16:27:12 -07:00
|
|
|
|
{
|
2022-12-14 13:37:23 +01:00
|
|
|
|
var filePath = "HelperFiles/file1.svg";
|
2020-07-21 16:27:12 -07:00
|
|
|
|
|
2022-12-14 13:37:23 +01:00
|
|
|
|
SvgThumbnailProvider svgThumbnailProvider = new SvgThumbnailProvider(filePath);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
|
2022-12-14 13:37:23 +01:00
|
|
|
|
Bitmap bitmap = svgThumbnailProvider.GetThumbnail(256);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
|
2022-12-14 13:37:23 +01:00
|
|
|
|
Assert.IsTrue(bitmap != null);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-21 17:23:42 -07:00
|
|
|
|
public void GetThumbnailValidStreamHTML()
|
2020-07-21 16:27:12 -07:00
|
|
|
|
{
|
2022-12-14 13:37:23 +01:00
|
|
|
|
var filePath = "HelperFiles/file2.svg";
|
2020-07-21 16:27:12 -07:00
|
|
|
|
|
2022-12-14 13:37:23 +01:00
|
|
|
|
SvgThumbnailProvider svgThumbnailProvider = new SvgThumbnailProvider(filePath);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
|
2022-12-14 13:37:23 +01:00
|
|
|
|
Bitmap bitmap = svgThumbnailProvider.GetThumbnail(256);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
|
2022-12-14 13:37:23 +01:00
|
|
|
|
Assert.IsTrue(bitmap != null);
|
2020-07-21 16:27:12 -07:00
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
