Source code for restfulgrok.mock

from view import GrokRestViewMixin
from fancyhtmlview import GrokRestViewWithFancyHtmlMixin
from StringIO import StringIO


class MockResponse(object):
    def __init__(self):
        self.headers = []
        self.status = None

    def setHeader(self, header, value):
        self.headers.append((header, value))

    def setStatus(self, code, msg):
        self.errmsg = msg
        self.status = (code, msg)

    def getStatus(self):
        return self.status[0]


class MockRequest(object):
    def __init__(self, method='GET', body='', getdata={},
                 headers={'Accept': 'application/json'}):
        self.method = method
        self.body = body
        self.getdata = {} #'BODY': self.body}
        self.getdata.update(getdata)
        self.headers = {}
        self.stdin = StringIO(body)
        for header, value in headers.iteritems():
            self.headers[header.lower()] = value

    def get(self, key, default=None):
        return self.getdata.get(key, default)

    def getHeader(self, header):
        return self.headers[header.lower()]


class MockContext(object):
    def __init__(self, parentnode=None, id=None):
        self.parentnode = parentnode
        self.id = id

    def getParentNode(self):
        return self.parentnode


class MockRestView(GrokRestViewMixin):
    def __init__(self, request=None, response=MockResponse(), context=None):
        self.request = request
        self.response = response
        self.context = context

class MockRestViewWithFancyHtml(GrokRestViewWithFancyHtmlMixin):
    def __init__(self, request=None, response=MockResponse(), context=None):
        self.request = request
        self.response = response
        self.context = context