From ca0016a996b7ed4dfb83b6f7f324037780fe3bee Mon Sep 17 00:00:00 2001 From: Gavin D'souza Date: Fri, 17 Jun 2022 11:32:40 +0530 Subject: [PATCH] test(sequence): Use mariadb client's exception handling --- frappe/tests/test_sequence.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) 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")