[test] Add test for issue5781

wanghongjuan · rogerwang · commit 4752dd1654e3 · 2018-02-06T14:28:19.000+08:00
- Add test for issue nwjs#5781 - This case is passed on v0.21.2 and v0.27.4 after adding "nwfaketop"
diff --git a/test/sanity/issue5781-X-Frame-Options/http-server.py b/test/sanity/issue5781-X-Frame-Options/http-server.py
@@ -0,0 +1,23 @@
+import SocketServer
+import BaseHTTPServer
+import sys
+
+PORT = int(sys.argv[1])
+
+class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):
+
+    def _set_headers(s):
+        s.send_response(200)
+        s.send_header('Content-type', 'text/html')
+        s.send_header("X-Frame-Options", "DENY")
+        s.end_headers()
+
+    def do_GET(s):
+        """Respond to a GET request."""
+        s._set_headers()
+        s.wfile.write("<script type='text/javascript'> document.write('<h1 id=\"res2\">Node is ' + (typeof nw === 'undefined' ? 'DISABLED': 'ENABLED') + '</h1>');")
+        s.wfile.write("</script>")
+
+if __name__ == "__main__":
+    httpd = SocketServer.TCPServer(("", PORT), MyHandler)
+    httpd.serve_forever()
diff --git a/test/sanity/issue5781-X-Frame-Options/index.tpl b/test/sanity/issue5781-X-Frame-Options/index.tpl
@@ -0,0 +1,8 @@
+<html>
+ <body>
+  <iframe src="http://localhost:{port}" nwfaketop ></iframe>
+  <script>
+      document.write('<h1 id="res1">Node is ' + (typeof nw === 'undefined' ? 'DISABLED': 'ENABLED') + '</h1>');
+    </script>
+ </body>
+</html>
diff --git a/test/sanity/issue5781-X-Frame-Options/package.json b/test/sanity/issue5781-X-Frame-Options/package.json
@@ -0,0 +1,4 @@
+{
+  "name":"issue5781-X-Frame-Options",
+  "main":"index.html"
+}
diff --git a/test/sanity/issue5781-X-Frame-Options/test.py b/test/sanity/issue5781-X-Frame-Options/test.py
@@ -0,0 +1,42 @@
+import time
+import os
+import sys
+import subprocess
+
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from nw_util import *
+
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common import utils
+
+chrome_options = Options()
+testdir = os.path.dirname(os.path.abspath(__file__))
+chrome_options.add_argument("nwapp=" + testdir)
+os.chdir(testdir)
+
+port = str(utils.free_port())
+server = subprocess.Popen(['python', 'http-server.py', port])
+
+tpl = open('index.tpl', 'r')
+content = tpl.read().replace('{port}', port)
+tpl.close()
+
+html = open('index.html', 'w')
+html.write(content)
+html.close()
+
+driver = webdriver.Chrome(executable_path=os.environ['CHROMEDRIVER'], chrome_options=chrome_options)
+driver.implicitly_wait(5)
+try:
+    print driver.current_url
+    print 'loading http server in iframe element'
+    wait_for_element_tag(driver, "iframe")
+    assert(driver.find_element_by_tag_name("iframe") is not None)
+    driver.switch_to_frame(driver.find_element_by_tag_name("iframe"))
+    res2 = wait_for_element_id(driver, 'res2')
+    assert( "Node is DISABLED" in res2)
+    print 'page iframe is loaded'
+finally:
+    server.terminate()
+    driver.quit()

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +{
 +  "name":"issue5781-X-Frame-Options",
 +  "main":"index.html"
 +}