P

lib/yql/tests/test_errors.py

@@ -0,0 +1,29 @@
+import json
+from unittest import TestCase
+
+from yql import NotOneError, YQLError
+
+
+class YQLErrorTest(TestCase):
+    def test_error_passed_error_string(self):
+        error = YQLError(resp='some response', content='some content')
+        self.assertEqual("some content", str(error))
+
+    def test_error_passed_object(self):
+        error = YQLError(resp='some response', content={"foo": 1})
+        self.assertEqual(repr({"foo": 1}), str(error))
+
+    def test_error_passed_json(self):
+        content = {
+            'error': {
+                'description': 'some description',
+            }
+        }
+        error = YQLError(resp='some response', content=json.dumps(content))
+        self.assertEqual("some description", str(error))
+
+
+class NotOneErrorTest(TestCase):
+    def test_is_represented_by_message_as_json(self):
+        error = NotOneError('some message')
+        self.assertEqual("some message", str(error))