[py] Add test for File Upload

AutomatedTester · AutomatedTester · commit 4750e40b9c92 · 2021-02-23T22:43:29.000Z
diff --git a/py/BUILD.bazel b/py/BUILD.bazel
@@ -143,6 +143,7 @@ py_library(
     data = [
         "pytest.ini",
         "setup.cfg",
+        "test/selenium/webdriver/common/test_file.txt"
     ],
     imports = ["."],
     deps = [
diff --git a/py/test/selenium/webdriver/common/test_file.txt b/py/test/selenium/webdriver/common/test_file.txt
@@ -0,0 +1 @@
+lorem ipsum dolor sit amet
diff --git a/py/test/selenium/webdriver/common/upload_tests.py b/py/test/selenium/webdriver/common/upload_tests.py
@@ -0,0 +1,33 @@
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+from selenium.webdriver.common.by import By
+
+
+import os
+
+
+def test_can_upload_file(driver, pages):
+
+    pages.load("upload.html")
+    current_dir = os.path.dirname(os.path.realpath(__file__))
+    driver.find_element(By.ID, 'upload').send_keys(os.path.join(current_dir, "test_file.txt"))
+    driver.find_element(By.ID, 'go').click()
+    driver.switch_to.frame(driver.find_element(By.ID, "upload_target"))
+    body = driver.find_element(By.CSS_SELECTOR, "body").text
+
+    assert "test_file.txt" in body
diff --git a/py/test/selenium/webdriver/common/webserver.py b/py/test/selenium/webdriver/common/webserver.py
@@ -20,6 +20,7 @@
 
 import logging
 import os
+import re
 import socket
 import threading
 from io import open
@@ -76,6 +77,47 @@ def do_GET(self):
         except IOError:
             self.send_error(404, 'File Not Found: %s' % path)
 
+    def do_POST(self):
+        """POST method handler."""
+        try:
+            remaining_bytes = int(self.headers['content-length'])
+            contents = ""
+            line = self.rfile.readline()
+            contents += line.decode("utf-8")
+            remaining_bytes -= len(line)
+            line = self.rfile.readline()
+            contents += line.decode("utf-8")
+            remaining_bytes -= len(line)
+            fn = re.findall(r'Content-Disposition.*name="upload"; filename="(.*)"', line.decode("utf-8"))
+            if not fn:
+                self.send_error(500, f"File not found. {contents}")
+                return
+            line = self.rfile.readline()
+            remaining_bytes -= len(line)
+            contents += line.decode("utf-8")
+            line = self.rfile.readline()
+            remaining_bytes -= len(line)
+            contents += line.decode("utf-8")
+            preline = self.rfile.readline()
+            remaining_bytes -= len(preline)
+            while remaining_bytes > 0:
+                line = self.rfile.readline()
+                remaining_bytes -= len(line)
+                contents += line.decode("utf-8")
+
+            self.send_response(200)
+            self.send_header('Content-type', 'text/html')
+            self.end_headers()
+
+            self.wfile.write(
+                f"""<!doctype html>
+                {contents}
+                <script>window.top.window.onUploadDone();</script>
+                """.encode('utf-8')
+            )
+        except Exception as e:
+            self.send_error(500, f"Error found: {e}")
+
     def log_message(self, format, *args):
         """Override default to avoid trashing stderr"""
         pass
