使用模拟restapi的测试环境

Question

假设我有一个非常简单的web应用程序，如果现任总统是民主党人，则显示为蓝色；如果现任总统是共和党人，则显示为红色。REST API用于通过端点获取当前总裁：

/presidents/current

当前返回json对象：

^{pr2}$

因此，当我的页面加载时，我调用端点，并根据返回者显示红色或蓝色。在

我希望测试这个HTML/javascript页面，并模拟后端，以便可以在测试环境中控制API响应。例如：

def test_republican():
    # configure the response for this test that the web app will receive when it connects to this endpoint
    configure_endpoint(
        "/presidents/current", 
        jsonify(
            name="Donald Trump",
            party="Republican"
        )
    )  

    # start the web app in the browser using selenium 
    load_web_app(driver, "http://localhost:8080")  

    e = driver.find_element_by_name("background")
    assert(e.getCssValue("background-color") == "red")


def test_democrat():
    # configure the response for this test that the web app will receive when it connects to this endpoint
    configure_endpoint(
        "/presidents/current", 
        jsonify(
            name="Barack Obama",
            party="Democrat"
        )
    )    

    # start the web app in the browser using selenium 
    load_web_app(driver, "http://localhost:8080")  

    e = driver.find_element_by_name("background")
    assert(e.getCssValue("background-color") == "blue")

所以问题是我应该如何实现函数configure_endpoint（），你能推荐什么库？在

Answer 1

我会使用tornado web框架。在

import json
import functools
import operator
from tornado import ioloop, web, gen
from tornado.options import define, options

define("data_file", default='default/mock.json', type=str)

class Handler(web.RequestHandler):

    def data_received(self, chunk):
        pass

    def initialize(self, data):
        self.data = data

    @gen.coroutine
    def get(self, *args, **kwargs):
        path = self.request.path.split("/")[1:]
        path = functools.reduce(
            operator.add,
            [[k, v[0].decode("utf-8")] for k, v in         self.request.query_arguments.items()],
            path
        )

        try:
            self.write(functools.reduce(operator.getitem, path, self.data))
        except KeyError:
            self.set_status(404)


class Application(web.Application):
    def __init__(self):
        data = {}
        with open(options.data_file) as data_file:
            data = json.load(data_file)

        handlers = [
            ('(.*)', Handler, {"data": data})
        ]
        settings = dict(
            gzip=True,
            static_hash_cache=True,
        )
        web.Application.__init__(self, handlers, **settings)


    def main():
        io_loop = ioloop.IOLoop.instance()
        backend_application = Application()
        backend_application.listen(8001)
        io_loop.start()

    if __name__ == "__main__":
        main()

这是我用来模拟REST-API的代码，REST-API是一个独立的脚本，但它也可以嵌入到测试环境中。在

我定义了一个JSON文件，它定义了不同的路径组件和应该返回的内容。像这样：

^{pr2}$

我把这个保存到mock.json文件并使用参数mock_rest.py --data-file="./mock.json"调用脚本。在

我希望这能给你一个起点和一个很好的例子。在