125 lines
2.9 KiB
Python
125 lines
2.9 KiB
Python
import unittest
|
|
import webob
|
|
from webob.dec import wsgify
|
|
import webtest
|
|
|
|
from wsme import *
|
|
from wsme.controller import scan_api, WSRoot
|
|
|
|
|
|
class DummyProtocol(object):
|
|
name = 'dummy'
|
|
content_types = ['', None]
|
|
|
|
def __init__(self):
|
|
self.hits = 0
|
|
|
|
def accept(self, req):
|
|
return True
|
|
|
|
def extract_path(self, request):
|
|
return ['touch']
|
|
|
|
def read_arguments(self, funcdef, request):
|
|
self.lastreq = request
|
|
self.hits += 1
|
|
return {}
|
|
|
|
def encode_result(self, funcdef, result):
|
|
return str(result)
|
|
|
|
def encode_error(self, infos):
|
|
return str(infos)
|
|
|
|
|
|
def serve_ws(req, root):
|
|
return root._handle_request(req)
|
|
|
|
|
|
class TestController(unittest.TestCase):
|
|
def test_expose(self):
|
|
class MyWS(WSRoot):
|
|
@expose(int)
|
|
def getint(self):
|
|
return 1
|
|
|
|
assert MyWS.getint._wsme_definition.return_type == int
|
|
|
|
def test_validate(self):
|
|
class MyWS(object):
|
|
@expose(int)
|
|
@validate(int, int, int)
|
|
def add(self, a, b, c=0):
|
|
return a + b + c
|
|
|
|
args = MyWS.add._wsme_definition.arguments
|
|
|
|
assert args[0].name == 'a'
|
|
assert args[0].datatype == int
|
|
assert args[0].mandatory
|
|
assert args[0].default is None
|
|
|
|
assert args[1].name == 'b'
|
|
assert args[1].datatype == int
|
|
assert args[1].mandatory
|
|
assert args[1].default is None
|
|
|
|
assert args[2].name == 'c'
|
|
assert args[2].datatype == int
|
|
assert not args[2].mandatory
|
|
assert args[2].default == 0
|
|
|
|
def test_register_protocol(self):
|
|
import wsme.controller
|
|
wsme.controller.register_protocol(DummyProtocol)
|
|
assert wsme.controller.registered_protocols['dummy'] == DummyProtocol
|
|
|
|
r = WSRoot()
|
|
assert len(r.protocols) == 0
|
|
|
|
r.addprotocol('dummy')
|
|
assert r.protocols['dummy']
|
|
|
|
r = WSRoot(['dummy'])
|
|
assert r.protocols['dummy']
|
|
|
|
def test_scan_api(self):
|
|
class NS(object):
|
|
@expose(int)
|
|
@validate(int, int)
|
|
def multiply(self, a, b):
|
|
return a * b
|
|
|
|
class MyRoot(WSRoot):
|
|
ns = NS()
|
|
|
|
r = MyRoot()
|
|
|
|
api = [i for i in scan_api(r)]
|
|
assert len(api) == 1
|
|
fd = api[0]
|
|
assert fd.path == ['ns']
|
|
assert fd.name == 'multiply'
|
|
|
|
def test_handle_request(self):
|
|
class MyRoot(WSRoot):
|
|
@expose()
|
|
def touch(self):
|
|
pass
|
|
|
|
p = DummyProtocol()
|
|
r = MyRoot(protocols=[p])
|
|
|
|
app = webtest.TestApp(
|
|
wsgify(r._handle_request))
|
|
|
|
res = app.get('/')
|
|
|
|
assert p.lastreq.path == '/'
|
|
assert p.hits == 1
|
|
|
|
res = app.get('/touch?wsmeproto=dummy')
|
|
|
|
assert p.lastreq.path == '/touch'
|
|
assert p.hits == 2
|