diff options
Diffstat (limited to 'testing/web-platform/tests/html/semantics/embedded-content/the-img-element/resources/image-and-stash.py')
-rw-r--r-- | testing/web-platform/tests/html/semantics/embedded-content/the-img-element/resources/image-and-stash.py | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/testing/web-platform/tests/html/semantics/embedded-content/the-img-element/resources/image-and-stash.py b/testing/web-platform/tests/html/semantics/embedded-content/the-img-element/resources/image-and-stash.py new file mode 100644 index 0000000000..d16a3e591d --- /dev/null +++ b/testing/web-platform/tests/html/semantics/embedded-content/the-img-element/resources/image-and-stash.py @@ -0,0 +1,44 @@ +# This is a simple implementation of a server-side stash that supports two +# operations: +# - increment: +# Increments a value in the stash keyed by a given uuid, and returns an +# image file +# - read: +# Returns the value in the stash keyed by a given uuid or 0 otherwise. +# This is a read-only operation, and does not remove the value from the +# stash as-is the default WPT stash behavior: +# https://web-platform-tests.org/tools/wptserve/docs/stash.html. + +import os + +from wptserve.utils import isomorphic_decode + +def main(request, response): + if b"increment" in request.GET: + uuid = request.GET[b"increment"] + + # First, increment the stash value keyed by `uuid`, and write it back to the + # stash. Writing it back to the stash is necessary since `take()` actually + # removes the value whereas we want to increment it. + stash_value = request.server.stash.take(uuid) + if stash_value is None: + stash_value = 0 + request.server.stash.put(uuid, stash_value + 1) + + # Return a basic image. + response_headers = [(b"Content-Type", b"image/png")] + image_path = os.path.join(os.path.dirname(isomorphic_decode(__file__)), u"image.png") + return (200, response_headers, open(image_path, mode='rb').read()) + + elif b"read" in request.GET: + uuid = request.GET[b"read"] + stash_value = request.server.stash.take(uuid) + + if stash_value is None: + stash_value = 0 + # Write the stash value keyed by `uuid` back to the stash. This is necessary + # because `take()` actually removes it, but we want a read-only read. + request.server.stash.put(uuid, stash_value); + return (200, [], str(stash_value)) + + return (404 , [], "Not found") |