blob: d9df41f323b8dc0a778a4839d345176b29f95bc9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
from unittest import TestCase
from pandora.errors import InternalServerError, PandoraException
class TestPandoraExceptionConstructionFromErrorCode(TestCase):
def test_it_returns_specific_error_class_if_possible(self):
error = PandoraException.from_code(0, "Test Message")
self.assertIsInstance(error, InternalServerError)
self.assertEqual("Test Message", error.extended_message)
self.assertEqual(0, error.code)
def test_it_returns_generic_error_if_unknown(self):
error = PandoraException.from_code(-99, "Test Message")
self.assertIsInstance(error, PandoraException)
self.assertEqual("Test Message", error.extended_message)
|