From ea3bfc914c1e3e448ca976bbbb96c01ebbaedf99 Mon Sep 17 00:00:00 2001 From: Sagar Vora Date: Mon, 11 Jul 2022 13:43:38 +0530 Subject: [PATCH] test: use newly created `get_default` helper --- frappe/tests/test_twofactor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/frappe/tests/test_twofactor.py b/frappe/tests/test_twofactor.py index d9bf982cd2..b04cb6a91d 100644 --- a/frappe/tests/test_twofactor.py +++ b/frappe/tests/test_twofactor.py @@ -12,6 +12,7 @@ from frappe.twofactor import ( authenticate_for_2factor, confirm_otp_token, get_cached_user_pass, + get_default, get_otpsecret_for_, get_verification_obj, should_run_2fa, @@ -111,7 +112,7 @@ class TestTwoFactor(unittest.TestCase): def test_get_otpsecret_for_user(self): """OTP secret should be set for user.""" self.assertTrue(get_otpsecret_for_(self.user)) - self.assertTrue(frappe.db.get_default(self.user + "_otpsecret")) + self.assertTrue(get_default(self.user + "_otpsecret")) def test_confirm_otp_token(self): """Ensure otp is confirmed"""