diff options
Diffstat (limited to 'tests/test_server.py')
| -rw-r--r-- | tests/test_server.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tests/test_server.py b/tests/test_server.py index bf571a6..3340398 100644 --- a/tests/test_server.py +++ b/tests/test_server.py @@ -91,6 +91,13 @@ class Server(unittest.TestCase): h = {**HOST, **ORIGIN, "Content-Type": "application/json"} self.assertEqual(self.req("POST", "/api/demo/boom", h, '{"x": null}')[0], 400) + def test_art_route(self): + from unittest import mock + with mock.patch.object(srv, "art", return_value=(b"PNGDATA", "image/png")): + status, body = self.req("GET", "/api/art", HOST) + self.assertEqual(status, 200) + self.assertEqual(body, b"PNGDATA") + if __name__ == "__main__": unittest.main() |
