How to Build a Mock REST GET Endpoint Handler in Python

from http.server import BaseHTTPRequestHandler, HTTPServer
import json

# Mock API handler registry
def handle_users():
    return {"status": "ok", "data": [{"id": 1, "name": "Alice"}, {"id": 2, "name": "Bob"}]}

def handle_products():
    return {"status": "ok", "data": [{"id": 101, "name": "Laptop", "price": 999.99}]}

def handle_not_found():
    return {"status": "error", "message": "Endpoint not found"}, 404

# Mock route map (dict of {'path': handler})
routes = {
    "/api/users": handle_users,
    "/api/products": handle_products,
}

class MockHandler(BaseHTTPRequestHandler):
    def do_GET(self):
        # Look up handler by path
        handler = routes.get(self.path, handle_not_found)
        result = handler()
        
        # Support handlers returning (data, status_code) tuples
        if isinstance(result, tuple):
            data, status = result
        else:
            data, status = result, 200
        
        self.send_response(status)
        self.send_header("Content-Type", "application/json")
        self.end_headers()
        self.wfile.write(json.dumps(data).encode())

if __name__ == "__main__":
    # Test the mock server in-process
    server = HTTPServer(("localhost", 8000), MockHandler)
    print("Mock server running on port 8000")
    print("Endpoints: /api/users, /api/products")
    server.serve_forever()