summaryrefslogtreecommitdiffstats
path: root/testing/condprofile/condprof/tests/fakegeckodriver.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 17:32:43 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 17:32:43 +0000
commit6bf0a5cb5034a7e684dcc3500e841785237ce2dd (patch)
treea68f146d7fa01f0134297619fbe7e33db084e0aa /testing/condprofile/condprof/tests/fakegeckodriver.py
parentInitial commit. (diff)
downloadthunderbird-upstream.tar.xz
thunderbird-upstream.zip
Adding upstream version 1:115.7.0.upstream/1%115.7.0upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rwxr-xr-xtesting/condprofile/condprof/tests/fakegeckodriver.py144
1 files changed, 144 insertions, 0 deletions
diff --git a/testing/condprofile/condprof/tests/fakegeckodriver.py b/testing/condprofile/condprof/tests/fakegeckodriver.py
new file mode 100755
index 0000000000..ab40401aaf
--- /dev/null
+++ b/testing/condprofile/condprof/tests/fakegeckodriver.py
@@ -0,0 +1,144 @@
+#!/usr/bin/env python3
+import argparse
+import json
+from http.server import BaseHTTPRequestHandler, HTTPServer
+from uuid import uuid4
+
+_SESSIONS = {}
+
+
+class Window:
+ def __init__(self, handle, title="about:blank"):
+ self.handle = handle
+ self.title = title
+
+ def visit_url(self, url):
+ print("Visiting %s" % url)
+ # XXX todo, load the URL for real
+ self.url = url
+
+
+class Session:
+ def __init__(self, uuid):
+ self.session_id = uuid
+ self.autoinc = 0
+ self.windows = {}
+ self.active_handle = self.new_window()
+
+ def visit(self, url):
+ self.windows[self.active_handle].visit_url(url)
+
+ def new_window(self):
+ w = Window(self.autoinc)
+ self.windows[w.handle] = w
+ self.autoinc += 1
+ return w.handle
+
+
+class RequestHandler(BaseHTTPRequestHandler):
+ def _set_headers(self, status=200):
+ self.send_response(status)
+ self.send_header("Content-type", "application/json")
+ self.end_headers()
+
+ def _send_response(self, status=200, data=None):
+ if data is None:
+ data = {}
+ data = json.dumps(data).encode("utf8")
+ self._set_headers(status)
+ self.wfile.write(data)
+
+ def _parse_path(self):
+ path = self.path.lstrip("/")
+ sections = path.split("/")
+ session = None
+ action = []
+ if len(sections) > 1:
+ session_id = sections[1]
+ if session_id in _SESSIONS:
+ session = _SESSIONS[session_id]
+ action = sections[2:]
+ return session, action
+
+ def do_GET(self):
+ print("GET " + self.path)
+ if self.path == "/status":
+ return self._send_response(data={"ready": "OK"})
+
+ session, action = self._parse_path()
+ if action == ["window", "handles"]:
+ data = {"value": list(session.windows.keys())}
+ return self._send_response(data=data)
+
+ if action == ["moz", "context"]:
+ data = {"value": "chrome"}
+ return self._send_response(data=data)
+
+ return self._send_response(status=404)
+
+ def do_POST(self):
+ print("POST " + self.path)
+ content_length = int(self.headers["Content-Length"])
+ post_data = json.loads(self.rfile.read(content_length))
+
+ # new session
+ if self.path == "/session":
+ uuid = str(uuid4())
+ _SESSIONS[uuid] = Session(uuid)
+ return self._send_response(data={"sessionId": uuid})
+
+ session, action = self._parse_path()
+ if action == ["url"]:
+ session.visit(post_data["url"])
+ return self._send_response()
+
+ if action == ["window", "new"]:
+ if session is None:
+ return self._send_response(404)
+ handle = session.new_window()
+ return self._send_response(data={"handle": handle, "type": "tab"})
+
+ if action == ["timeouts"]:
+ return self._send_response()
+
+ if action == ["execute", "async"]:
+ return self._send_response(data={"logs": []})
+
+ # other commands not supported yet, we just return 200s
+ return self._send_response()
+
+ def do_DELETE(self):
+ return self._send_response()
+ session, action = self._parse_path()
+ if session is not None:
+ del _SESSIONS[session.session_id]
+ return self._send_response()
+ return self._send_response(status=404)
+
+
+VERSION = """\
+geckodriver 0.24.0 ( 2019-01-28)
+
+The source code of this program is available from
+testing/geckodriver in https://hg.mozilla.org/mozilla-central.
+
+This program is subject to the terms of the Mozilla Public License 2.0.
+You can obtain a copy of the license at https://mozilla.org/MPL/2.0/.\
+"""
+
+
+if __name__ == "__main__":
+ parser = argparse.ArgumentParser(description="FakeGeckodriver")
+ parser.add_argument("--log", type=str, default=None)
+ parser.add_argument("--port", type=int, default=4444)
+ parser.add_argument("--marionette-port", type=int, default=2828)
+ parser.add_argument("--version", action="store_true", default=False)
+ parser.add_argument("--verbose", "-v", action="count")
+ args = parser.parse_args()
+
+ if args.version:
+ print(VERSION)
+ else:
+ HTTPServer.allow_reuse_address = True
+ server = HTTPServer(("127.0.0.1", args.port), RequestHandler)
+ server.serve_forever()