simpletestingserver.py

#!/usr/bin/env python3
from http.server import BaseHTTPRequestHandler
from http.server import HTTPServer
import cgi

PORT = 8003
FILE_PREFIX = "."

if __name__ == "__main__":
    try:
        import argparse

        parser = argparse.ArgumentParser(description = 'A simple fake server for testing your API client.')
        parser.add_argument('-p', '--port', type = int, dest = "PORT",
                           help = 'the port to run the server on; defaults to 8003')
        parser.add_argument('--path', type = str, dest = "PATH",
                           help = 'the folder to find the json files')

        args = parser.parse_args()

        if args.PORT:
            PORT = args.PORT
        if args.PATH:
            FILE_PREFIX = args.PATH

    except TypeError:
        # Could not successfully import argparse or something
        pass


class JSONRequestHandler(BaseHTTPRequestHandler):

    def do_GET(self):

        # send response code:
        self.send_response(200)
        # send headers:
        self.send_header("Content-type", "application/json")
        # send a blank line to end headers:
        self.wfile.write("\n")

        try:
            output = open(FILE_PREFIX + "/" + self.path[1:] + ".json", 'r').read()
        except IOError:
            output = "{'error': 'Could not find file " + self.path[1:] + ".json'" + "}"
        self.wfile.write(output)

    def do_POST(self):
        if self.path == "/success":
            response_code = 200
        elif self.path == "/error":
            response_code = 500
        else:
            try:
                response_code = int(self.path[1:])
            except ValueError:
                response_code = 201
        try:
            self.send_response(response_code)
            self.wfile.write('Content-Type: application/json\n')
            self.wfile.write('Client: %s\n' % str(self.client_address))
            self.wfile.write('User-agent: %s\n' % str(self.headers['user-agent']))
            self.wfile.write('Path: %s\n' % self.path)
            self.end_headers()

            form = cgi.FieldStorage(
                    fp = self.rfile,
                    headers = self.headers,
                    environ = {'REQUEST_METHOD':'POST',
                               'CONTENT_TYPE':self.headers['Content-Type'],
                               })

            self.wfile.write('{\n')
            first_key = True
            for field in form.keys():
                if not first_key:
                    self.wfile.write(',\n')
                else:
                    self.wfile.write('\n')
                    first_key = False
                self.wfile.write('"%s":"%s"' % (field, form[field].value))
            self.wfile.write('\n}')
                            
        except OSError as err:
            print("OS error: {0}".format(err))
            self.send_response(500)        
        else:
            self.send_response(500)


server = HTTPServer(("localhost", PORT), JSONRequestHandler)
server.serve_forever()