Skip to content

Commit 5a68db4

Browse files
committed
Implement __str__ method for OperationalError: #83
Signed-off-by: Tsuyoshi Hombashi <tsuyoshi.hombashi@gmail.com>
1 parent 5cc1641 commit 5a68db4

File tree

2 files changed

+18
-0
lines changed

2 files changed

+18
-0
lines changed

simplesqlite/error.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,3 +56,9 @@ def __init__(self, *args: Any, **kwargs: Any) -> None:
5656
self.__message = kwargs.pop("message", None)
5757

5858
super().__init__(*args)
59+
60+
def __str__(self) -> str:
61+
if not self.message:
62+
return ""
63+
64+
return str(self.message)

test/test_error.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
from simplesqlite.error import OperationalError
2+
3+
4+
def test_operational_error_str_with_message():
5+
error_message = "Test error message"
6+
error = OperationalError(message=error_message)
7+
assert str(error) == error_message
8+
9+
10+
def test_operational_error_str_without_message():
11+
error = OperationalError()
12+
assert str(error) == ""

0 commit comments

Comments
 (0)