[test] add nwuseragent case for nwjs#5397

rogerwang · rogerwang · commit 3c61ca970a03 · 2017-02-21T14:47:35.000+08:00
diff --git a/test/sanity/nwuseragent/echo-user-agent.py b/test/sanity/nwuseragent/echo-user-agent.py
@@ -0,0 +1,28 @@
+#!/usr/bin/env python
+
+from BaseHTTPServer import HTTPServer, BaseHTTPRequestHandler
+import sys
+
+class RequestHandler(BaseHTTPRequestHandler):
+    
+    def do_GET(self):
+        
+        request_path = self.path
+        
+        print(self.headers['User-Agent'])
+        
+        self.send_response(200)
+        self.send_header("Set-Cookie", "foo=bar")
+
+    def log_request(self, code):
+        pass
+        
+def main():
+    port = int(sys.argv[1])
+    print('Listening on localhost:%s' % port)
+    server = HTTPServer(('', port), RequestHandler)
+    server.serve_forever()
+
+        
+if __name__ == "__main__":
+    main()
diff --git a/test/sanity/nwuseragent/package.json b/test/sanity/nwuseragent/package.json
@@ -0,0 +1,4 @@
+{
+  "name":"test-nwuseragent",
+  "main":"index.html"
+}
diff --git a/test/sanity/nwuseragent/test.py b/test/sanity/nwuseragent/test.py
@@ -0,0 +1,44 @@
+import time
+import os
+import subprocess
+
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common import utils
+
+chrome_options = Options()
+chrome_options.add_argument("nwapp=" + os.path.dirname(os.path.abspath(__file__)))
+
+testdir = os.path.dirname(os.path.abspath(__file__))
+os.chdir(testdir)
+
+port = str(utils.free_port())
+server = subprocess.Popen(['python', '-u', 'echo-user-agent.py', port], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+print server.stdout.readline()
+
+html = open('index.html', 'w')
+html.write('''
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="utf8">
+    <title>test</title>
+  </head>
+  <body>
+    <iframe id="iframe"  nwuseragent="test-agent" src="http://localhost:%s/iframe.html"></iframe>
+  </body>
+</html>
+''' % (port))
+
+html.close()
+
+driver = webdriver.Chrome(executable_path=os.environ['CHROMEDRIVER'], chrome_options=chrome_options)
+try:
+    print driver.current_url
+    time.sleep(1)
+    user_agent = server.stdout.readline()
+    print "user agent: " + user_agent
+    server.terminate()
+    assert("test-agent" in user_agent)
+finally:
+    driver.quit()

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +{
 +  "name":"test-nwuseragent",
 +  "main":"index.html"
 +}