Adding System.Drawing.Printing PrintDocument tests. (#20711)

src/System.Drawing.Common/tests/Printing/PrintDocumentTests.cs

@@ -0,0 +1,272 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// See the LICENSE file in the project root for more information.
+//
+// Copyright (C) 2005-2006 Novell, Inc (http://www.novell.com)
+//
+// Permission is hereby granted, free of charge, to any person obtaining
+// a copy of this software and associated documentation files (the
+// "Software"), to deal in the Software without restriction, including
+// without limitation the rights to use, copy, modify, merge, publish,
+// distribute, sublicense, and/or sell copies of the Software, and to
+// permit persons to whom the Software is furnished to do so, subject to
+// the following conditions:
+//
+// The above copyright notice and this permission notice shall be
+// included in all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+// LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+// OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+// WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+//
+
+using Xunit;
+
+namespace System.Drawing.Printing.Tests
+{
+    public class PrintDocumentTests
+    {
+        private readonly PageSettings _pageSettings = new PageSettings()
+        {
+            PaperSize = new PaperSize()
+            {
+                RawKind = (int)PaperKind.A3
+            }
+        };
+
+        [Fact]
+        public void Ctor_Default_Success()
+        {
+            using (var document = new PrintDocument())
+            {
+                Assert.Equal("document", document.DocumentName);
+                Assert.False(document.OriginAtMargins);
+                AssertDefaultPageSettings(document.DefaultPageSettings);
+            }
+        }
+
+        [Fact]
+        public void DefaultPageSettings_SetValue_ReturnsExpected()
+        {
+            using (var document = new PrintDocument())
+            {
+                document.DefaultPageSettings = null;
+                Assert.IsAssignableFrom<PageSettings>(document.DefaultPageSettings);
+
+                document.DefaultPageSettings = _pageSettings;
+                Assert.Equal(_pageSettings.PaperSize.Kind, _pageSettings.PaperSize.Kind);
+            }
+        }
+
+        [Fact]
+        public void DefaultPageSettings_Null_ReturnsExpected()
+        {
+            using (var document = new PrintDocument())
+            {
+                document.DefaultPageSettings = null;
+                AssertDefaultPageSettings(document.DefaultPageSettings);
+            }
+        }
+
+        [Theory]
+        [InlineData(null)]
+        [InlineData("")]
+        [InlineData("newDocument")]
+        public void DocumentName_SetValue_ReturnsExpected(string documentName)
+        {
+            if (documentName == null)
+            { documentName = string.Empty; }
+
+            using (var document = new PrintDocument())
+            {
+                document.DocumentName = documentName;
+                Assert.Equal(documentName, document.DocumentName);
+            }
+        }
+
+        [Theory]
+        [InlineData(true)]
+        [InlineData(false)]
+        public void OriginAtMargins_SetValue_ReturnsExpected(bool originAtMargins)
+        {
+            using (var document = new PrintDocument())
+            {
+                document.OriginAtMargins = originAtMargins;
+                Assert.Equal(originAtMargins, document.OriginAtMargins);
+            }
+        }
+
+        [Fact]
+        public void PrintController_SetValue_ReturnsExpected()
+        {
+            using (var document = new PrintDocument())
+            {
+                document.PrintController = null;
+                Assert.NotNull(document.PrintController);
+
+                var printController = new StandardPrintController();
+                document.PrintController = printController;
+                Assert.Same(printController, document.PrintController);
+            }
+        }
+
+        [Fact]
+        public void PrinterSettings_SetValue_ReturnsExpected()
+        {
+            using (var document = new PrintDocument())
+            {
+                document.PrinterSettings = null;
+                Assert.IsAssignableFrom<PrinterSettings>(document.PrinterSettings);
+
+                var printerSettings = new PrinterSettings();
+                document.PrinterSettings = printerSettings;
+                Assert.Same(printerSettings, document.PrinterSettings);
+                Assert.Equal(
+                    document.PrinterSettings.DefaultPageSettings.PaperSize.Kind,
+                    document.DefaultPageSettings.PaperSize.Kind);
+
+                document.DefaultPageSettings = _pageSettings;
+                document.PrinterSettings = printerSettings;
+                Assert.Equal(
+                    _pageSettings.PaperSize.Kind,
+                    document.DefaultPageSettings.PaperSize.Kind);
+            }
+        }
+
+        [Fact]
+        public void BeginPrint_SetValue_ReturnsExpected()
+        {
+            bool flag = false;
+            var beginPrintHandler = new PrintEventHandler((sender, e) => flag = true);
+
+            using (var document = new PrintDocument())
+            {
+                document.PrintController = new TestPrintController();
+                document.BeginPrint += beginPrintHandler;
+                document.Print();
+                Assert.True(flag);
+
+                flag = false;
+                document.BeginPrint -= beginPrintHandler;
+                document.Print();
+                Assert.False(flag);
+            }
+        }
+
+        [Fact]
+        public void EndPrint_SetValue_ReturnsExpected()
+        {
+            bool flag = false;
+            var endPrintHandler = new PrintEventHandler((sender, e) => flag = true);
+
+            using (var document = new PrintDocument())
+            {
+                document.PrintController = new TestPrintController();
+                document.EndPrint += endPrintHandler;
+                document.Print();
+                Assert.True(flag);
+
+                flag = false;
+                document.EndPrint -= endPrintHandler;
+                document.Print();
+                Assert.False(flag);
+            }
+        }
+
+        [Fact]
+        public void PrintPage_SetValue_ReturnsExpected()
+        {
+            bool flag = false;
+            var printPageHandler = new PrintPageEventHandler((sender, e) => flag = true);
+
+            using (var document = new PrintDocument())
+            {
+                document.PrintController = new TestPrintController();
+                document.PrintPage += printPageHandler;
+                document.Print();
+                Assert.True(flag);
+
+                flag = false;
+                document.PrintPage -= printPageHandler;
+                document.Print();
+                Assert.False(flag);
+            }
+        }
+
+        [Fact]
+        public void QueryPageSettings_SetValue_ReturnsExpected()
+        {
+            bool flag = false;
+            var queryPageSettingsHandler = new QueryPageSettingsEventHandler((sender, e) => flag = true);
+
+            using (var document = new PrintDocument())
+            {
+                document.PrintController = new TestPrintController();
+                document.QueryPageSettings += queryPageSettingsHandler;
+                document.Print();
+                Assert.True(flag);
+
+                flag = false;
+                document.QueryPageSettings -= queryPageSettingsHandler;
+                document.Print();
+                Assert.False(flag);
+            }
+        }
+
+        [Fact]
+        public void ToString_ReturnsExpected()
+        {
+            using (var document = new PrintDocument())
+            {
+                var expected = string.Format("[PrintDocument {0}]", document.DocumentName);
+                Assert.Equal(expected, document.ToString());
+            }
+        }
+
+        private void AssertDefaultPageSettings(PageSettings pageSettings)
+        {
+            Assert.Equal(new Rectangle(0, 0, 827, 1169), pageSettings.Bounds);
+            Assert.True(pageSettings.Color);
+            Assert.Equal(0, pageSettings.HardMarginX);
+            Assert.Equal(0, pageSettings.HardMarginY);
+            Assert.False(pageSettings.Landscape);
+            Assert.Equal(PaperKind.A4, pageSettings.PaperSize.Kind);
+            Assert.Equal(PaperSourceKind.FormSource, pageSettings.PaperSource.Kind);
+            Assert.Equal(new PointF(0f, 0f), pageSettings.PrintableArea.Location);
+            Assert.Equal(827, pageSettings.PrintableArea.Width, 0);
+            Assert.Equal(1169, pageSettings.PrintableArea.Height, 0);
+            Assert.Equal(PrinterResolutionKind.Custom, pageSettings.PrinterResolution.Kind);
+            Assert.True(pageSettings.PrinterSettings.IsDefaultPrinter);
+        }
+
+        private class TestPrintController : PrintController
+        {
+            public override Graphics OnStartPage(PrintDocument document, PrintPageEventArgs e)
+            {
+                using (var bitmap = new Bitmap(20, 20))
+                {
+                    return Graphics.FromImage(bitmap);
+                }
+            }
+
+            public override void OnStartPrint(PrintDocument document, PrintEventArgs e)
+            {
+                base.OnStartPrint(document, e);
+            }
+
+            public override void OnEndPrint(PrintDocument document, PrintEventArgs e)
+            {
+                base.OnEndPrint(document, e);
+            }
+
+            public override void OnEndPage(PrintDocument document, PrintPageEventArgs e)
+            {
+                base.OnEndPage(document, e);
+                e.Graphics.Dispose();
+            }
+        }
+    }
+}

src/System.Drawing.Common/tests/System.Drawing.Common.Tests.csproj

@@ -44,6 +44,7 @@
     <Compile Include="Imaging\ImageFormatTests.cs" />
     <Compile Include="Imaging\MetaHeaderTests.cs" />
     <Compile Include="Imaging\WmfPlaceableFileHeaderTests.cs" />
+    <Compile Include="Printing\PrintDocumentTests.cs" />
     <Compile Include="RegionTests.cs" />
     <Compile Include="SolidBrushTests.cs" />
     <Compile Include="StringFormatTests.cs" />