summaryrefslogtreecommitdiff
path: root/service/test/unit
diff options
context:
space:
mode:
Diffstat (limited to 'service/test/unit')
-rw-r--r--service/test/unit/resources/test_helpers.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/service/test/unit/resources/test_helpers.py b/service/test/unit/resources/test_helpers.py
new file mode 100644
index 00000000..a17ce755
--- /dev/null
+++ b/service/test/unit/resources/test_helpers.py
@@ -0,0 +1,30 @@
+import unittest
+import re
+
+from pixelated.resources import respond_json, respond_json_deferred
+from test.unit.resources import DummySite
+from twisted.web.test.requesthelper import DummyRequest
+
+
+class TestHelpers(unittest.TestCase):
+
+ def setUp(self):
+ pass
+
+ def test_respond_json_should_populate_response(self):
+ request = DummyRequest([''])
+ body = respond_json({"test": "yep"}, request)
+
+ self.assertEqual(200, request.code)
+ self.assertEqual(b"{\"test\": \"yep\"}", body)
+ self.assertEqual([b"application/json"],
+ request.responseHeaders.getRawHeaders("Content-Type"))
+
+ def test_respond_json_deferred_should_populate_response(self):
+ request = DummyRequest([''])
+ body = respond_json_deferred({"test": "yep"}, request)
+
+ self.assertEqual(200, request.code)
+ self.assertEqual(b"{\"test\": \"yep\"}", request.written[0])
+ self.assertEqual([b"application/json"],
+ request.responseHeaders.getRawHeaders("Content-Type"))