X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;ds=inline;f=tests%2Fdays.py;h=06e4302d71aaba58867e961506806a792c1fe807;hb=13bce001bd86de2c20dff2f93d9f08cfae95a16b;hp=f3ed08228aafa3550ed49caa1bf022f144285e6e;hpb=3558a14701955de18ae7adbda0e93aaee7710a92;p=plomtask diff --git a/tests/days.py b/tests/days.py index f3ed082..06e4302 100644 --- a/tests/days.py +++ b/tests/days.py @@ -3,7 +3,7 @@ from unittest import TestCase from datetime import datetime from tests.utils import TestCaseWithDB, TestCaseWithServer from plomtask.days import Day, todays_date -from plomtask.misc import HandledException +from plomtask.exceptions import BadFormatException, NotFoundException class TestsSansDB(TestCase): @@ -11,11 +11,11 @@ class TestsSansDB(TestCase): def test_Day_dates(self) -> None: """Test Day's date format.""" - with self.assertRaises(HandledException): + with self.assertRaises(BadFormatException): Day('foo') - with self.assertRaises(HandledException): + with self.assertRaises(BadFormatException): Day('2024-02-30') - with self.assertRaises(HandledException): + with self.assertRaises(BadFormatException): Day('2024-02-01 23:00:00') self.assertEqual(datetime(2024, 1, 1), Day('2024-01-01').datetime) @@ -37,12 +37,12 @@ class TestsWithDB(TestCaseWithDB): def test_Day_by_date(self) -> None: """Test Day.by_date().""" - with self.assertRaises(HandledException): + with self.assertRaises(NotFoundException): Day.by_date(self.db_conn, '2024-01-01') Day('2024-01-01').save(self.db_conn) self.assertEqual(Day('2024-01-01'), Day.by_date(self.db_conn, '2024-01-01')) - with self.assertRaises(HandledException): + with self.assertRaises(NotFoundException): Day.by_date(self.db_conn, '2024-01-02') self.assertEqual(Day('2024-01-02'), Day.by_date(self.db_conn, '2024-01-02', create=True))