Source code for bzz.mock

# -*- coding: utf-8 -*-

# This file is part of bzz.

# Licensed under the MIT license:
# Copyright (c) 2014 Bernardo Heynemann

import tornado.web
from collections import OrderedDict

import bzz.core as core

class MockedRoutesHandler(tornado.web.RequestHandler):
    def __init__(self, *args, **kwargs):
        self.handler_methods = kwargs.pop('handler_methods')
        super(MockedRoutesHandler, self).__init__(*args, **kwargs)

    def prepare(self):
        if self.request.method in self.handler_methods or '*' in self.handler_methods:
            response = self.handler_methods.get(self.request.method, None) or self.handler_methods.get('*')
            status = response.get('status', 200)
            body = response.get('body', '')

            if hasattr(body, '__call__'):
                body = body(self.request)

            if status >= 400:
                raise tornado.web.HTTPError(status, body)

            if 'cookies' in response:
                for cookie, value in response['cookies'].items():
                    self.set_cookie(cookie, value)


[docs]class MockHive(object): @classmethod
[docs] def routes_for(cls, routes_tuple): """ Returns a tuples list of paths, tornado ready """ routes = OrderedDict() for route in routes_tuple: if not route[1] in routes: routes[route[1]] = {} if not route[0] in routes[route[1]]: routes[route[1]][route[0]] = {} if len(route) > 2: result = route[2] else: result = {'body': '', 'status': 200} routes[route[1]][route[0]] = result result = core.RouteList() for route, methods in routes.items(): route_tuple = (route, MockedRoutesHandler, dict(handler_methods=methods)) result.append(route_tuple) return result