2020-08-18 01:00:56 +08: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 System.Windows.Forms;
|
2020-09-17 03:24:07 +08:00
|
|
|
|
using Microsoft.PowerToys.PreviewHandler.Svg;
|
2020-08-18 01:00:56 +08:00
|
|
|
|
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
|
|
|
|
using Moq;
|
|
|
|
|
using PreviewHandlerCommon;
|
|
|
|
|
|
2020-08-22 08:23:42 +08:00
|
|
|
|
namespace SvgPreviewHandlerUnitTests
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
[TestClass]
|
|
|
|
|
public class SvgPreviewControlTests
|
|
|
|
|
{
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlShouldAddExtendedBrowserControlWhenDoPreviewCalled()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.DoPreview(GetMockStream("<svg></svg>"));
|
|
|
|
|
|
|
|
|
|
// Assert
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(1, svgPreviewControl.Controls.Count);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
Assert.IsInstanceOfType(svgPreviewControl.Controls[0], typeof(WebBrowserExt));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlShouldSetDocumentStreamWhenDoPreviewCalled()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.DoPreview(GetMockStream("<svg></svg>"));
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.IsNotNull(((WebBrowser)svgPreviewControl.Controls[0]).DocumentStream);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlShouldDisableWebBrowserContextMenuWhenDoPreviewCalled()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.DoPreview(GetMockStream("<svg></svg>"));
|
|
|
|
|
|
|
|
|
|
// Assert
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(false, ((WebBrowser)svgPreviewControl.Controls[0]).IsWebBrowserContextMenuEnabled);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlShouldFillDockForWebBrowserWhenDoPreviewCalled()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.DoPreview(GetMockStream("<svg></svg>"));
|
|
|
|
|
|
|
|
|
|
// Assert
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(DockStyle.Fill, ((WebBrowser)svgPreviewControl.Controls[0]).Dock);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlShouldSetScriptErrorsSuppressedPropertyWhenDoPreviewCalled()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.DoPreview(GetMockStream("<svg></svg>"));
|
|
|
|
|
|
|
|
|
|
// Assert
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(true, ((WebBrowser)svgPreviewControl.Controls[0]).ScriptErrorsSuppressed);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlShouldSetScrollBarsEnabledPropertyWhenDoPreviewCalled()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.DoPreview(GetMockStream("<svg></svg>"));
|
|
|
|
|
|
|
|
|
|
// Assert
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(true, ((WebBrowser)svgPreviewControl.Controls[0]).ScrollBarsEnabled);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlShouldDisableAllowNavigationWhenDoPreviewCalled()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.DoPreview(GetMockStream("<svg></svg>"));
|
|
|
|
|
|
|
|
|
|
// Assert
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(false, ((WebBrowser)svgPreviewControl.Controls[0]).AllowNavigation);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlShouldAddValidInfoBarIfSvgPreviewThrows()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
var mockStream = new Mock<IStream>();
|
|
|
|
|
mockStream
|
|
|
|
|
.Setup(x => x.Read(It.IsAny<byte[]>(), It.IsAny<int>(), It.IsAny<IntPtr>()))
|
|
|
|
|
.Throws(new Exception());
|
|
|
|
|
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.DoPreview(mockStream.Object);
|
|
|
|
|
var textBox = svgPreviewControl.Controls[0] as RichTextBox;
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.IsFalse(string.IsNullOrWhiteSpace(textBox.Text));
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(1, svgPreviewControl.Controls.Count);
|
|
|
|
|
Assert.AreEqual(DockStyle.Top, textBox.Dock);
|
|
|
|
|
Assert.AreEqual(Color.LightYellow, textBox.BackColor);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
Assert.IsTrue(textBox.Multiline);
|
|
|
|
|
Assert.IsTrue(textBox.ReadOnly);
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(RichTextBoxScrollBars.None, textBox.ScrollBars);
|
|
|
|
|
Assert.AreEqual(BorderStyle.None, textBox.BorderStyle);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlInfoBarWidthShouldAdjustWithParentControlWidthChangesIfSvgPreviewThrows()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
var mockStream = new Mock<IStream>();
|
|
|
|
|
mockStream
|
|
|
|
|
.Setup(x => x.Read(It.IsAny<byte[]>(), It.IsAny<int>(), It.IsAny<IntPtr>()))
|
|
|
|
|
.Throws(new Exception());
|
|
|
|
|
svgPreviewControl.DoPreview(mockStream.Object);
|
|
|
|
|
var textBox = svgPreviewControl.Controls[0] as RichTextBox;
|
|
|
|
|
var incrementParentControlWidth = 5;
|
|
|
|
|
var initialParentWidth = svgPreviewControl.Width;
|
|
|
|
|
var initialTextBoxWidth = textBox.Width;
|
|
|
|
|
var finalParentWidth = initialParentWidth + incrementParentControlWidth;
|
|
|
|
|
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.Width += incrementParentControlWidth;
|
|
|
|
|
|
|
|
|
|
// Assert
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(initialTextBoxWidth, initialParentWidth);
|
|
|
|
|
Assert.AreEqual(textBox.Width, finalParentWidth);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlShouldAddTextBoxIfBlockedElementsArePresent()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
var svgBuilder = new StringBuilder();
|
|
|
|
|
svgBuilder.AppendLine("<svg width =\"200\" height=\"200\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">");
|
|
|
|
|
svgBuilder.AppendLine("\t<script>alert(\"hello\")</script>");
|
|
|
|
|
svgBuilder.AppendLine("</svg>");
|
|
|
|
|
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.DoPreview(GetMockStream(svgBuilder.ToString()));
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.IsInstanceOfType(svgPreviewControl.Controls[0], typeof(RichTextBox));
|
|
|
|
|
Assert.IsInstanceOfType(svgPreviewControl.Controls[1], typeof(WebBrowserExt));
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(2, svgPreviewControl.Controls.Count);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlShouldNotAddTextBoxIfNoBlockedElementsArePresent()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
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>");
|
|
|
|
|
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.DoPreview(GetMockStream(svgBuilder.ToString()));
|
|
|
|
|
|
|
|
|
|
// Assert
|
|
|
|
|
Assert.IsInstanceOfType(svgPreviewControl.Controls[0], typeof(WebBrowserExt));
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(1, svgPreviewControl.Controls.Count);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
[TestMethod]
|
2020-08-22 08:23:42 +08:00
|
|
|
|
public void SvgPreviewControlInfoBarWidthShouldAdjustWithParentControlWidthChangesIfBlockedElementsArePresent()
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
// Arrange
|
|
|
|
|
using (var svgPreviewControl = new SvgPreviewControl())
|
|
|
|
|
{
|
|
|
|
|
var svgBuilder = new StringBuilder();
|
|
|
|
|
svgBuilder.AppendLine("<svg width =\"200\" height=\"200\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">");
|
|
|
|
|
svgBuilder.AppendLine("\t<script>alert(\"hello\")</script>");
|
|
|
|
|
svgBuilder.AppendLine("</svg>");
|
|
|
|
|
svgPreviewControl.DoPreview(GetMockStream(svgBuilder.ToString()));
|
|
|
|
|
var textBox = svgPreviewControl.Controls[0] as RichTextBox;
|
|
|
|
|
var incrementParentControlWidth = 5;
|
|
|
|
|
var initialParentWidth = svgPreviewControl.Width;
|
|
|
|
|
var initialTextBoxWidth = textBox.Width;
|
|
|
|
|
var finalParentWidth = initialParentWidth + incrementParentControlWidth;
|
|
|
|
|
|
|
|
|
|
// Act
|
|
|
|
|
svgPreviewControl.Width += incrementParentControlWidth;
|
|
|
|
|
|
|
|
|
|
// Assert
|
2020-10-08 04:12:59 +08:00
|
|
|
|
Assert.AreEqual(initialTextBoxWidth, initialParentWidth);
|
|
|
|
|
Assert.AreEqual(textBox.Width, finalParentWidth);
|
2020-08-18 01:00:56 +08:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2020-08-22 08:23:42 +08:00
|
|
|
|
private static IStream GetMockStream(string streamData)
|
2020-08-18 01:00:56 +08:00
|
|
|
|
{
|
|
|
|
|
var mockStream = new Mock<IStream>();
|
|
|
|
|
var streamBytes = Encoding.UTF8.GetBytes(streamData);
|
|
|
|
|
|
|
|
|
|
var streamMock = new Mock<IStream>();
|
|
|
|
|
var firstCall = true;
|
|
|
|
|
streamMock
|
|
|
|
|
.Setup(x => x.Read(It.IsAny<byte[]>(), It.IsAny<int>(), It.IsAny<IntPtr>()))
|
|
|
|
|
.Callback<byte[], int, IntPtr>((buffer, countToRead, bytesReadPtr) =>
|
|
|
|
|
{
|
|
|
|
|
if (firstCall)
|
|
|
|
|
{
|
|
|
|
|
Array.Copy(streamBytes, 0, buffer, 0, streamBytes.Length);
|
|
|
|
|
Marshal.WriteInt32(bytesReadPtr, streamBytes.Length);
|
|
|
|
|
firstCall = false;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
Marshal.WriteInt32(bytesReadPtr, 0);
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
return streamMock.Object;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|