From ec34da60a124eb107b1d433d6070283bf947f48b Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Sun, 14 Oct 2018 11:57:15 -0400 Subject: [PATCH] more tests --- tests/test_responder.py | 47 ++++++++++++++++++++++++++++++++++++++--- 1 file changed, 44 insertions(+), 3 deletions(-) diff --git a/tests/test_responder.py b/tests/test_responder.py index 81a200e..84645b9 100644 --- a/tests/test_responder.py +++ b/tests/test_responder.py @@ -183,6 +183,47 @@ def test_async_function(api, session): assert r.text == content -# def test_media_parsing(api, session): -# dump = {'hello': 'sam'} -# @api.route('/') +def test_media_parsing(api, session): + dump = {"hello": "sam"} + + @api.route("/") + def route(req, resp): + resp.media = dump + + r = session.get(api.url_for(route)) + assert r.json() == dump + + r = session.get(api.url_for(route), headers={"Accept": "application/x-yaml"}) + assert r.text == "{hello: sam}\n" + + +def test_background(api, session): + @api.route("/") + def route(req, resp): + @api.background.task + def task(): + import time + + time.sleep(3) + + task() + api.text = "ok" + + r = session.get(api.url_for(route)) + assert r.ok + + +def test_multiple_routes(api, session): + @api.route("/1") + def route1(req, resp): + resp.text = "1" + + @api.route("/2") + def route2(req, resp): + resp.text = "2" + + r = session.get(api.url_for(route1)) + assert r.text == "1" + + r = session.get(api.url_for(route2)) + assert r.text == "2"