from view import GrokRestViewMixin
from fancyhtmlview import GrokRestViewWithFancyHtmlMixin
from StringIO import StringIO
[docs]class MockResponse(object):
def __init__(self):
self.headers = []
self.status = None
[docs] def setStatus(self, code, msg):
self.errmsg = msg
self.status = (code, msg)
[docs] def getStatus(self):
return self.status[0]
[docs]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
[docs] def get(self, key, default=None):
return self.getdata.get(key, default)
[docs]class MockContext(object):
def __init__(self, parentnode=None, id=None):
self.parentnode = parentnode
self.id = id
[docs] def getParentNode(self):
return self.parentnode
[docs]class MockRestView(GrokRestViewMixin):
def __init__(self, request=None, response=MockResponse(), context=None):
self.request = request
self.response = response
self.context = context
[docs]class MockRestViewWithFancyHtml(GrokRestViewWithFancyHtmlMixin):
def __init__(self, request=None, response=MockResponse(), context=None):
self.request = request
self.response = response
self.context = context