Implemented Page.setContent tests (#99)

Meir017 · kblok · commit 5db4883d3852 · 2018-03-23T09:37:09.000-03:00
diff --git a/lib/PuppeteerSharp.Tests/Page/SetContentTests.cs b/lib/PuppeteerSharp.Tests/Page/SetContentTests.cs
@@ -0,0 +1,47 @@
+﻿using System.Threading.Tasks;
+using Xunit;
+
+namespace PuppeteerSharp.Tests.Page
+{
+    [Collection("PuppeteerLoaderFixture collection")]
+    public class SetContentTests : PuppeteerBaseTest
+    {
+        const string ExpectedOutput = "<html><head></head><body><div>hello</div></body></html>";
+
+        [Fact]
+        public async Task ShouldWork()
+        {
+            var page = await Browser.NewPageAsync();
+
+            await page.SetContentAsync("<div>hello</div>");
+            var result = await page.GetContentAsync();
+
+            Assert.Equal(ExpectedOutput, result);
+        }
+
+        [Fact]
+        public async Task ShouldWorkWithDoctype()
+        {
+            var page = await Browser.NewPageAsync();
+            const string doctype = "<!DOCTYPE html>";
+
+            await page.SetContentAsync($"{doctype}<div>hello</div>");
+            var result = await page.GetContentAsync();
+
+            Assert.Equal($"{doctype}{ExpectedOutput}", result);
+        }
+
+        [Fact]
+        public async Task ShouldWorkWithHtml4Doctype()
+        {
+            var page = await Browser.NewPageAsync();
+            const string doctype = "<!DOCTYPE html PUBLIC \" -//W3C//DTD HTML 4.01//EN\" " +
+                "\"http://www.w3.org/TR/html4/strict.dtd\">";
+
+            await page.SetContentAsync($"{doctype}<div>hello</div>");
+            var result = await page.GetContentAsync();
+
+            Assert.Equal($"{doctype}{ExpectedOutput}", result);
+        }
+    }
+}
diff --git a/lib/PuppeteerSharp/Frame.cs b/lib/PuppeteerSharp/Frame.cs
@@ -116,6 +116,27 @@ internal Task<ElementHandle> AddScriptTag(dynamic options)
             throw new NotImplementedException();
         }
 
+        internal async Task<string> GetContentAsync()
+        {
+            return await EvaluateFunctionAsync<string>(@"() => {
+                let retVal = '';
+                if (document.doctype)
+                    retVal = new XMLSerializer().serializeToString(document.doctype);
+                if (document.documentElement)
+                    retVal += document.documentElement.outerHTML;
+                return retVal;
+            }");
+        }
+
+        internal async Task SetContentAsync(string html)
+        {
+            await EvaluateFunctionAsync(@"html => {
+                document.open();
+                document.write(html);
+                document.close();
+            }", html);
+        }
+        
         internal async Task<string> GetTitleAsync() => await EvaluateExpressionAsync<string>("document.title");
 
         internal void OnLifecycleEvent(string loaderId, string name)
diff --git a/lib/PuppeteerSharp/Page.cs b/lib/PuppeteerSharp/Page.cs
@@ -262,6 +262,10 @@ await page.SetViewport(new ViewPortOptions
             return page;
         }
 
+        public async Task<string> GetContentAsync() => await _frameManager.MainFrame.GetContentAsync();
+
+        public async Task SetContentAsync(string html) => await _frameManager.MainFrame.SetContentAsync(html);
+
         public async Task<dynamic> GoToAsync(string url, Dictionary<string, string> options = null)
         {
             var referrer = _networkManager.ExtraHTTPHeaders?.GetValueOrDefault("referer");

Original file line number	Diff line number	Diff line change
`@@ -262,6 +262,10 @@ await page.SetViewport(new ViewPortOptions`
`262`	`262`	`return page;`
`263`	`263`	`}`
`264`	`264`
	`265`	`+ public async Task<string> GetContentAsync() => await _frameManager.MainFrame.GetContentAsync();`
	`266`	`+`
	`267`	`+ public async Task SetContentAsync(string html) => await _frameManager.MainFrame.SetContentAsync(html);`
	`268`	`+`
`265`	`269`	`public async Task<dynamic> GoToAsync(string url, Dictionary<string, string> options = null)`
`266`	`270`	`{`
`267`	`271`	`var referrer = _networkManager.ExtraHTTPHeaders?.GetValueOrDefault("referer");`