aboutsummaryrefslogtreecommitdiffstats
path: root/tests/test_server.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_server.py')
-rw-r--r--tests/test_server.py7
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()