47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
import error_handling as er
|
||
|
|
|
||
|
|
|
||
|
|
class FileLike(object):
|
||
|
|
def __init__(self):
|
||
|
|
self.is_open = False
|
||
|
|
|
||
|
|
def open(self):
|
||
|
|
self.is_open = True
|
||
|
|
|
||
|
|
def close(self):
|
||
|
|
self.is_open = False
|
||
|
|
|
||
|
|
|
||
|
|
class ErrorHandlingTest(unittest.TestCase):
|
||
|
|
def test_throw_exception(self):
|
||
|
|
with self.assertRaises(Exception):
|
||
|
|
er.handle_error_by_throwing_exception()
|
||
|
|
|
||
|
|
def test_return_none(self):
|
||
|
|
self.assertEqual(1, er.handle_error_by_returning_none('1'),
|
||
|
|
'Result of valid input should not be None')
|
||
|
|
self.assertNone(er.handle_error_by_returning_none('a'),
|
||
|
|
'Result of invalid input should be None')
|
||
|
|
|
||
|
|
def test_return_tuple(self):
|
||
|
|
successful_result, result = er.handle_error_by_returning_tuple('1')
|
||
|
|
self.assertTrue(successful_result, 'Valid input should be successful')
|
||
|
|
self.assertEqual(1, result, 'Result of valid input should not be None')
|
||
|
|
|
||
|
|
failure_result, result = er.handle_error_by_returning_tuple('a')
|
||
|
|
self.assertFalse(failure_result,
|
||
|
|
'Invalid input should not be successful')
|
||
|
|
|
||
|
|
def test_filelike_objects_are_closed_on_exception(self):
|
||
|
|
filelike_object = FileLike()
|
||
|
|
filelike_object.open()
|
||
|
|
with self.assertRaises(BaseException):
|
||
|
|
er.filelike_objects_are_closed_on_exception(filelike_object)
|
||
|
|
self.assertFalse(filelike_object.is_open)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|