From b1dac4b9fbaea8d5b9938625f3fd97d2c03270e0 Mon Sep 17 00:00:00 2001 From: Thomas Haller Date: Wed, 17 May 2023 17:09:51 +0200 Subject: test-cloue-meta-mock: add and use helper to send response code --- tools/test-cloud-meta-mock.py | 22 ++++++++++------------ 1 file changed, 10 insertions(+), 12 deletions(-) diff --git a/tools/test-cloud-meta-mock.py b/tools/test-cloud-meta-mock.py index 2b8d34b7fe..4135745d11 100755 --- a/tools/test-cloud-meta-mock.py +++ b/tools/test-cloud-meta-mock.py @@ -33,39 +33,37 @@ class MockCloudMDRequestHandler(BaseHTTPRequestHandler): def log_message(self, format, *args): pass + def _response_and_end(self, code): + self.send_response(code) + self.end_headers() + def do_GET(self): path = self.path.encode("ascii") if path in self.server._resources: - self.send_response(200) - self.end_headers() + self._response_and_end(200) self.wfile.write(self.server._resources[path]) else: - self.send_response(404) - self.end_headers() + self._response_and_end(404) def do_PUT(self): path = self.path.encode("ascii") if path == b"/latest/api/token": - self.send_response(200) - self.end_headers() + self._response_and_end(200) self.wfile.write( b"AQAAALH-k7i18JMkK-ORLZQfAa7nkNjQbKwpQPExNHqzk1oL_7eh-A==" ) else: length = int(self.headers["content-length"]) self.server._resources[path] = self.rfile.read(length) - self.send_response(201) - self.end_headers() + self._response_and_end(201) def do_DELETE(self): path = self.path.encode("ascii") if path in self.server._resources: del self.server._resources[path] - self.send_response(204) - self.end_headers() + self._response_and_end(204) else: - self.send_response(404) - self.end_headers() + self._response_and_end(404) class SocketHTTPServer(HTTPServer): -- cgit v1.2.1