def test_empty_request(self): app = Microdot() mock_socket.clear_requests() fd = mock_socket.FakeStream(b'\n') mock_socket._requests.append(fd) self._add_shutdown(app) app.run() self.assertTrue(fd.response.startswith(b'HTTP/1.0 400 N/A\r\n')) self.assertIn(b'Content-Length: 11\r\n', fd.response) self.assertIn(b'Content-Type: text/plain\r\n', fd.response) self.assertTrue(fd.response.endswith(b'\r\n\r\nBad request'))
def test_get_request(self): app = Microdot() @app.route('/') def index(req): return 'foo' @app.route('/async') async def index2(req): return 'foo-async' client = TestClient(app) res = self._run(client.get('/')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.headers['Content-Length'], '3') self.assertEqual(res.text, 'foo') self.assertEqual(res.body, b'foo') self.assertEqual(res.json, None) res = self._run(client.get('/async')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.headers['Content-Length'], '9') self.assertEqual(res.text, 'foo-async') self.assertEqual(res.body, b'foo-async') self.assertEqual(res.json, None)
def test_streaming(self): app = Microdot() @app.route('/') def index(req): class stream(): def __init__(self): self.i = 0 self.data = ['foo', b'bar'] def __aiter__(self): return self async def __anext__(self): if self.i >= len(self.data): raise StopAsyncIteration data = self.data[self.i] self.i += 1 return data return stream() client = TestClient(app) res = self._run(client.get('/')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, 'foobar')
def test_post_request(self): app = Microdot() @app.route('/') def index(req): return 'foo' @app.route('/', methods=['POST']) def index_post(req): return Response('bar') @app.route('/async', methods=['POST']) async def index_post2(req): return Response('bar-async') client = TestClient(app) res = self._run(client.post('/')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.headers['Content-Length'], '3') self.assertEqual(res.text, 'bar') self.assertEqual(res.body, b'bar') self.assertEqual(res.json, None) res = self._run(client.post('/async')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.headers['Content-Length'], '9') self.assertEqual(res.text, 'bar-async') self.assertEqual(res.body, b'bar-async') self.assertEqual(res.json, None)
def test_post_request(self): app = Microdot() @app.route('/') def index(req): return 'foo' @app.route('/', methods=['POST']) def index_post(req): return Response('bar') @app.route('/async', methods=['POST']) async def index_post2(req): return Response('bar-async') mock_socket.clear_requests() fd = mock_socket.add_request('POST', '/') fd2 = mock_socket.add_request('POST', '/async') self.assertRaises(IndexError, app.run) self.assertTrue(fd.response.startswith(b'HTTP/1.0 200 OK\r\n')) self.assertIn(b'Content-Length: 3\r\n', fd.response) self.assertIn(b'Content-Type: text/plain\r\n', fd.response) self.assertTrue(fd.response.endswith(b'\r\n\r\nbar')) self.assertTrue(fd2.response.startswith(b'HTTP/1.0 200 OK\r\n')) self.assertIn(b'Content-Length: 9\r\n', fd2.response) self.assertIn(b'Content-Type: text/plain\r\n', fd2.response) self.assertTrue(fd2.response.endswith(b'\r\n\r\nbar-async'))
def test_method_decorators(self): app = Microdot() @app.get('/get') def get(req): return 'GET' @app.post('/post') async def post(req): return 'POST' @app.put('/put') def put(req): return 'PUT' @app.patch('/patch') async def patch(req): return 'PATCH' @app.delete('/delete') def delete(req): return 'DELETE' client = TestClient(app) methods = ['GET', 'POST', 'PUT', 'PATCH', 'DELETE'] for method in methods: res = self._run( getattr(client, method.lower())('/' + method.lower())) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, method)
def test_400_handler(self): self._mock() app = Microdot() @app.errorhandler(400) async def handle_404(req): return '400' mock_socket.clear_requests() fd = mock_socket.FakeStream(b'\n') mock_socket._requests.append(fd) self._add_shutdown(app) app.run() self.assertTrue(fd.response.startswith(b'HTTP/1.0 200 OK\r\n')) self.assertIn(b'Content-Length: 3\r\n', fd.response) self.assertIn(b'Content-Type: text/plain\r\n', fd.response) self.assertTrue(fd.response.endswith(b'\r\n\r\n400')) self._unmock()
def test_render_template_in_app_async(self): app = MicrodotAsync() @app.route('/') async def index(req): return render_template('hello.utemplate.txt', name='foo') req = _run( RequestAsync.create(app, get_async_request_fd('GET', '/'), 'addr')) res = _run(app.dispatch_request(req)) self.assertEqual(res.status_code, 200) async def get_result(): result = [] async for chunk in res.body_iter(): result.append(chunk) return result result = _run(get_result()) self.assertEqual(result, ['Hello, ', 'foo', '!\n'])
def test_binary_payload(self): app = Microdot() @app.post('/') def index(req): return req.body client = TestClient(app) res = self._run(client.post('/', body=b'foo')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, 'foo')
def test_headers(self): app = Microdot() @app.route('/') def index(req): return req.headers.get('X-Foo') client = TestClient(app) res = self._run(client.get('/', headers={'X-Foo': 'bar'})) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, 'bar')
def test_500(self): app = Microdot() @app.route('/') def index(req): return 1 / 0 client = TestClient(app) res = self._run(client.get('/')) self.assertEqual(res.status_code, 500) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, 'Internal server error')
def test_413(self): app = Microdot() @app.route('/') def index(req): return 'foo' client = TestClient(app) res = self._run(client.post('/foo', body='x' * 17000)) self.assertEqual(res.status_code, 413) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, 'Payload too large')
def test_405(self): app = Microdot() @app.route('/foo') def index(req): return 'foo' client = TestClient(app) res = self._run(client.post('/foo')) self.assertEqual(res.status_code, 405) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, 'Not found')
def test_abort(self): app = Microdot() @app.route('/') def index(req): abort(406, 'Not acceptable') return 'foo' client = TestClient(app) res = self._run(client.get('/')) self.assertEqual(res.status_code, 406) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, 'Not acceptable')
def test_cookies(self): app = Microdot() @app.route('/') def index(req): return req.cookies['one'] + req.cookies['two'] + \ req.cookies['three'] client = TestClient(app, cookies={'one': '1', 'two': '2'}) res = self._run(client.get('/', headers={'Cookie': 'three=3'})) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, '123')
def test_before_after_request(self): app = Microdot() @app.before_request def before_request(req): if req.path == '/bar': @req.after_request async def after_request(req, res): res.headers['X-Two'] = '2' return res return 'bar', 202 req.g.message = 'baz' @app.after_request def after_request_one(req, res): res.headers['X-One'] = '1' @app.after_request async def after_request_two(req, res): res.set_cookie('foo', 'bar') return res @app.route('/bar') def bar(req): return 'foo' @app.route('/baz') def baz(req): return req.g.message client = TestClient(app) res = self._run(client.get('/bar')) self.assertEqual(res.status_code, 202) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.headers['Set-Cookie'], ['foo=bar']) self.assertEqual(res.headers['X-One'], '1') self.assertEqual(res.headers['X-Two'], '2') self.assertEqual(res.text, 'bar') self.assertEqual(client.cookies['foo'], 'bar') res = self._run(client.get('/baz')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.headers['Set-Cookie'], ['foo=bar']) self.assertEqual(res.headers['X-One'], '1') self.assertFalse('X-Two' in res.headers) self.assertEqual(res.headers['Content-Length'], '3') self.assertEqual(res.text, 'baz')
def test_404(self): app = Microdot() @app.route('/') def index(req): return 'foo' mock_socket.clear_requests() fd = mock_socket.add_request('GET', '/foo') self.assertRaises(IndexError, app.run) self.assertTrue(fd.response.startswith(b'HTTP/1.0 404 N/A\r\n')) self.assertIn(b'Content-Length: 9\r\n', fd.response) self.assertIn(b'Content-Type: text/plain\r\n', fd.response) self.assertTrue(fd.response.endswith(b'\r\n\r\nNot found'))
def test_500(self): app = Microdot() @app.route('/') def index(req): return 1 / 0 mock_socket.clear_requests() fd = mock_socket.add_request('GET', '/') self.assertRaises(IndexError, app.run) self.assertTrue(fd.response.startswith(b'HTTP/1.0 500 N/A\r\n')) self.assertIn(b'Content-Length: 21\r\n', fd.response) self.assertIn(b'Content-Type: text/plain\r\n', fd.response) self.assertTrue(fd.response.endswith(b'\r\n\r\nInternal server error'))
def test_binary_response(self): app = Microdot() @app.route('/bin') def index(req): return b'\xff\xfe', {'Content-Type': 'application/octet-stream'} client = TestClient(app) res = self._run(client.get('/bin')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'application/octet-stream') self.assertEqual(res.text, None) self.assertEqual(res.json, None) self.assertEqual(res.body, b'\xff\xfe')
def test_500_handler(self): app = Microdot() @app.route('/') def index(req): return 1 / 0 @app.errorhandler(500) def handle_500(req): return '501', 501 client = TestClient(app) res = self._run(client.get('/')) self.assertEqual(res.status_code, 501) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, '501')
def test_exception_handler(self): app = Microdot() @app.route('/') def index(req): return 1 / 0 @app.errorhandler(ZeroDivisionError) async def handle_div_zero(req, exc): return '501', 501 client = TestClient(app) res = self._run(client.get('/')) self.assertEqual(res.status_code, 501) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, '501')
def test_405_handler(self): app = Microdot() @app.route('/foo') def index(req): return 'foo' @app.errorhandler(405) async def handle_405(req): return '405', 405 client = TestClient(app) res = self._run(client.patch('/foo')) self.assertEqual(res.status_code, 405) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, '405')
def test_413_handler(self): app = Microdot() @app.route('/') def index(req): return 'foo' @app.errorhandler(413) async def handle_413(req): return '413', 400 client = TestClient(app) res = self._run(client.post('/foo', body='x' * 17000)) self.assertEqual(res.status_code, 400) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, '413')
def test_abort_handler(self): app = Microdot() @app.route('/') def index(req): abort(406) return 'foo' @app.errorhandler(406) def handle_500(req): return '406', 406 client = TestClient(app) res = self._run(client.get('/')) self.assertEqual(res.status_code, 406) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, '406')
def test_before_after_request(self): app = Microdot() @app.before_request def before_request(req): if req.path == '/bar': return 'bar', 202 req.g.message = 'baz' @app.after_request def after_request_one(req, res): res.headers['X-One'] = '1' @app.after_request async def after_request_two(req, res): res.set_cookie('foo', 'bar') return res @app.route('/bar') def bar(req): return 'foo' @app.route('/baz') def baz(req): return req.g.message mock_socket.clear_requests() fd = mock_socket.add_request('GET', '/bar') self.assertRaises(IndexError, app.run) self.assertTrue(fd.response.startswith(b'HTTP/1.0 202 N/A\r\n')) self.assertIn(b'X-One: 1\r\n', fd.response) self.assertIn(b'Set-Cookie: foo=bar\r\n', fd.response) self.assertIn(b'Content-Length: 3\r\n', fd.response) self.assertIn(b'Content-Type: text/plain\r\n', fd.response) self.assertTrue(fd.response.endswith(b'\r\n\r\nbar')) mock_socket.clear_requests() fd = mock_socket.add_request('GET', '/baz') self.assertRaises(IndexError, app.run) self.assertTrue(fd.response.startswith(b'HTTP/1.0 200 OK\r\n')) self.assertIn(b'X-One: 1\r\n', fd.response) self.assertIn(b'Set-Cookie: foo=bar\r\n', fd.response) self.assertIn(b'Content-Length: 3\r\n', fd.response) self.assertIn(b'Content-Type: text/plain\r\n', fd.response) self.assertTrue(fd.response.endswith(b'\r\n\r\nbaz'))
def test_exception_handler(self): app = Microdot() @app.route('/') def index(req): return 1 / 0 @app.errorhandler(ZeroDivisionError) async def handle_div_zero(req, exc): return '501', 501 mock_socket.clear_requests() fd = mock_socket.add_request('GET', '/') self.assertRaises(IndexError, app.run) self.assertTrue(fd.response.startswith(b'HTTP/1.0 501 N/A\r\n')) self.assertIn(b'Content-Length: 3\r\n', fd.response) self.assertIn(b'Content-Type: text/plain\r\n', fd.response) self.assertTrue(fd.response.endswith(b'\r\n\r\n501'))
def test_404_handler(self): app = Microdot() @app.route('/') def index(req): return 'foo' @app.errorhandler(404) async def handle_404(req): return '404' mock_socket.clear_requests() fd = mock_socket.add_request('GET', '/foo') self.assertRaises(IndexError, app.run) self.assertTrue(fd.response.startswith(b'HTTP/1.0 200 OK\r\n')) self.assertIn(b'Content-Length: 3\r\n', fd.response) self.assertIn(b'Content-Type: text/plain\r\n', fd.response) self.assertTrue(fd.response.endswith(b'\r\n\r\n404'))
def test_tuple_responses(self): app = Microdot() @app.route('/body') def one(req): return 'one' @app.route('/body-status') def two(req): return 'two', 202 @app.route('/body-headers') def three(req): return '<p>three</p>', {'Content-Type': 'text/html'} @app.route('/body-status-headers') def four(req): return '<p>four</p>', 202, {'Content-Type': 'text/html'} client = TestClient(app) res = self._run(client.get('/body')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, 'one') res = self._run(client.get('/body-status')) self.assertEqual(res.status_code, 202) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, 'two') res = self._run(client.get('/body-headers')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/html') self.assertEqual(res.text, '<p>three</p>') res = self._run(client.get('/body-status-headers')) self.assertEqual(res.status_code, 202) self.assertEqual(res.headers['Content-Type'], 'text/html') self.assertEqual(res.text, '<p>four</p>')
def test_json_response(self): app = Microdot() @app.route('/dict') async def json_dict(req): return {'foo': 'bar'} @app.route('/list') def json_list(req): return ['foo', 'bar'] client = TestClient(app) res = self._run(client.get('/dict')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'application/json') self.assertEqual(res.json, {'foo': 'bar'}) res = self._run(client.get('/list')) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'application/json') self.assertEqual(res.json, ['foo', 'bar'])
def test_json_payload(self): app = Microdot() @app.post('/dict') def json_dict(req): print(req.headers) return req.json.get('foo') @app.post('/list') def json_list(req): return req.json[0] client = TestClient(app) res = self._run(client.post('/dict', body={'foo': 'bar'})) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, 'bar') res = self._run(client.post('/list', body=['foo', 'bar'])) self.assertEqual(res.status_code, 200) self.assertEqual(res.headers['Content-Type'], 'text/plain') self.assertEqual(res.text, 'foo')