tests: allow fake Kong server to receive PUT requests

1 files changed, 9 insertions, 0 deletions

diff --git a/tests/kong.py b/tests/kong.py
index a1f9cec..c39542f 100755
--- a/tests/kong.py
+++ b/tests/kong.py
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 
 from http.server import HTTPServer, CGIHTTPRequestHandler
+import random
 
 class SimpleHTTPRequestHandler(CGIHTTPRequestHandler):
     def do_GET(self):
@@ -17,6 +18,14 @@ class SimpleHTTPRequestHandler(CGIHTTPRequestHandler):
         self.end_headers()
         self.wfile.write(b'Hello, world!')
 
+    def do_PUT(self):
+        if bool(random.getrandbits(1)):
+            self.send_response(200)
+        else:
+            self.send_response(201)
+        self.end_headers()
+        self.wfile.write(b'Hello, world!')
+
 
 httpd = HTTPServer(('127.0.0.1', 8000), SimpleHTTPRequestHandler)
 httpd.serve_forever()