1 files changed, 40 insertions, 0 deletions
diff --git a/test/wpt/tests/fetch/stale-while-revalidate/resources/stale-image.py b/test/wpt/tests/fetch/stale-while-revalidate/resources/stale-image.py
new file mode 100644
index 0000000..36e6fc0
--- /dev/null
+++ b/test/wpt/tests/fetch/stale-while-revalidate/resources/stale-image.py
@@ -0,0 +1,40 @@
+import os.path
+
+from wptserve.utils import isomorphic_decode
+
+def main(request, response):
+
+    token = request.GET.first(b"token", None)
+    is_query = request.GET.first(b"query", None) != None
+    with request.server.stash.lock:
+      value = request.server.stash.take(token)
+      count = 0
+      if value != None:
+        count = int(value)
+      if is_query:
+        if count < 2:
+          request.server.stash.put(token, count)
+      else:
+        count = count + 1
+        request.server.stash.put(token, count)
+
+    if is_query:
+      headers = [(b"Count", count)]
+      content = b""
+      return 200, headers, content
+    else:
+      filename = u"green-16x16.png"
+      if count > 1:
+        filename = u"green-256x256.png"
+
+      path = os.path.join(os.path.dirname(isomorphic_decode(__file__)), u"../../../images", filename)
+      body = open(path, "rb").read()
+
+      response.add_required_headers = False
+      response.writer.write_status(200)
+      response.writer.write_header(b"content-length", len(body))
+      response.writer.write_header(b"Cache-Control", b"private, max-age=0, stale-while-revalidate=60")
+      response.writer.write_header(b"content-type", b"image/png")
+      response.writer.end_headers()
+
+      response.writer.write(body)