diff --git a/frappe/tests/test_sequence.py b/frappe/tests/test_sequence.py index a60e4b1ac9..82bb8ab257 100644 --- a/frappe/tests/test_sequence.py +++ b/frappe/tests/test_sequence.py @@ -1,5 +1,4 @@ import psycopg2 -import pymysql import frappe from frappe.tests.utils import FrappeTestCase @@ -35,10 +34,10 @@ class TestSequence(FrappeTestCase): try: frappe.db.get_next_sequence_val(seq_name) - except pymysql.err.OperationalError as e: - self.assertEqual(e.args[0], 4084) except psycopg2.errors.SequenceGeneratorLimitExceeded: pass + except frappe.db.ProgrammingError as e: + self.assertEqual(getattr(e, "errno", None), 4084) else: self.fail("NEXTVAL didn't raise any error upon sequence's end")