aboutsummaryrefslogtreecommitdiff
path: root/tests/test_pandora/test_errors.py
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)