diff --git a/frappe/automation/doctype/assignment_rule/test_assignment_rule.py b/frappe/automation/doctype/assignment_rule/test_assignment_rule.py index 1dcc654ee0..8f1773608f 100644 --- a/frappe/automation/doctype/assignment_rule/test_assignment_rule.py +++ b/frappe/automation/doctype/assignment_rule/test_assignment_rule.py @@ -1,16 +1,16 @@ # Copyright (c) 2021, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.test_runner import make_test_records +from frappe.tests.utils import FrappeTestCase from frappe.utils import random_string -class TestAutoAssign(unittest.TestCase): +class TestAutoAssign(FrappeTestCase): @classmethod def setUpClass(cls): + super().setUpClass() frappe.db.delete("Assignment Rule") @classmethod diff --git a/frappe/automation/doctype/auto_repeat/test_auto_repeat.py b/frappe/automation/doctype/auto_repeat/test_auto_repeat.py index ee0addf847..21ed7f2aa1 100644 --- a/frappe/automation/doctype/auto_repeat/test_auto_repeat.py +++ b/frappe/automation/doctype/auto_repeat/test_auto_repeat.py @@ -1,7 +1,5 @@ # Copyright (c) 2018, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.automation.doctype.auto_repeat.auto_repeat import ( create_repeated_entries, @@ -9,6 +7,7 @@ from frappe.automation.doctype.auto_repeat.auto_repeat import ( week_map, ) from frappe.custom.doctype.custom_field.custom_field import create_custom_field +from frappe.tests.utils import FrappeTestCase from frappe.utils import add_days, add_months, getdate, today @@ -26,7 +25,7 @@ def add_custom_fields(): create_custom_field("ToDo", df) -class TestAutoRepeat(unittest.TestCase): +class TestAutoRepeat(FrappeTestCase): def setUp(self): if not frappe.db.sql( "SELECT `fieldname` FROM `tabCustom Field` WHERE `fieldname`='auto_repeat' and `dt`=%s", "Todo" diff --git a/frappe/automation/doctype/milestone/test_milestone.py b/frappe/automation/doctype/milestone/test_milestone.py index 5ac0754e5a..5348479809 100644 --- a/frappe/automation/doctype/milestone/test_milestone.py +++ b/frappe/automation/doctype/milestone/test_milestone.py @@ -1,8 +1,8 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestMilestone(unittest.TestCase): +class TestMilestone(FrappeTestCase): pass diff --git a/frappe/automation/doctype/milestone_tracker/test_milestone_tracker.py b/frappe/automation/doctype/milestone_tracker/test_milestone_tracker.py index 2b48a76805..3242145bc4 100644 --- a/frappe/automation/doctype/milestone_tracker/test_milestone_tracker.py +++ b/frappe/automation/doctype/milestone_tracker/test_milestone_tracker.py @@ -1,12 +1,11 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe import frappe.cache_manager +from frappe.tests.utils import FrappeTestCase -class TestMilestoneTracker(unittest.TestCase): +class TestMilestoneTracker(FrappeTestCase): def test_milestone(self): frappe.db.delete("Milestone Tracker") diff --git a/frappe/contacts/doctype/address/test_address.py b/frappe/contacts/doctype/address/test_address.py index edcf87f5bc..1d11c5efef 100644 --- a/frappe/contacts/doctype/address/test_address.py +++ b/frappe/contacts/doctype/address/test_address.py @@ -1,12 +1,11 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.contacts.doctype.address.address import get_address_display +from frappe.tests.utils import FrappeTestCase -class TestAddress(unittest.TestCase): +class TestAddress(FrappeTestCase): def test_template_works(self): if not frappe.db.exists("Address Template", "India"): frappe.get_doc({"doctype": "Address Template", "country": "India", "is_default": 1}).insert() diff --git a/frappe/contacts/doctype/address_template/test_address_template.py b/frappe/contacts/doctype/address_template/test_address_template.py index 8045313c69..ee45ce98f8 100644 --- a/frappe/contacts/doctype/address_template/test_address_template.py +++ b/frappe/contacts/doctype/address_template/test_address_template.py @@ -1,11 +1,10 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestAddressTemplate(unittest.TestCase): +class TestAddressTemplate(FrappeTestCase): def setUp(self): self.make_default_address_template() diff --git a/frappe/contacts/doctype/contact/test_contact.py b/frappe/contacts/doctype/contact/test_contact.py index bf0d1037db..6f0fd4ae3f 100644 --- a/frappe/contacts/doctype/contact/test_contact.py +++ b/frappe/contacts/doctype/contact/test_contact.py @@ -1,13 +1,12 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase test_dependencies = ["Contact", "Salutation"] -class TestContact(unittest.TestCase): +class TestContact(FrappeTestCase): def test_check_default_email(self): emails = [ {"email": "test1@example.com", "is_primary": 0}, diff --git a/frappe/contacts/doctype/gender/test_gender.py b/frappe/contacts/doctype/gender/test_gender.py index c8df3b566d..1b428521b7 100644 --- a/frappe/contacts/doctype/gender/test_gender.py +++ b/frappe/contacts/doctype/gender/test_gender.py @@ -1,7 +1,7 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestGender(unittest.TestCase): +class TestGender(FrappeTestCase): pass diff --git a/frappe/contacts/doctype/salutation/test_salutation.py b/frappe/contacts/doctype/salutation/test_salutation.py index 2c35e5bd2b..a1d9e044a0 100644 --- a/frappe/contacts/doctype/salutation/test_salutation.py +++ b/frappe/contacts/doctype/salutation/test_salutation.py @@ -1,7 +1,7 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestSalutation(unittest.TestCase): +class TestSalutation(FrappeTestCase): pass diff --git a/frappe/contacts/report/addresses_and_contacts/test_addresses_and_contacts.py b/frappe/contacts/report/addresses_and_contacts/test_addresses_and_contacts.py index 2ad8bfaba3..d15d518f63 100644 --- a/frappe/contacts/report/addresses_and_contacts/test_addresses_and_contacts.py +++ b/frappe/contacts/report/addresses_and_contacts/test_addresses_and_contacts.py @@ -1,8 +1,7 @@ -import unittest - import frappe import frappe.defaults from frappe.contacts.report.addresses_and_contacts.addresses_and_contacts import get_data +from frappe.tests.utils import FrappeTestCase def get_custom_linked_doctype(): @@ -87,7 +86,7 @@ def create_linked_contact(link_list, address): frappe.flags.test_contact_created = True -class TestAddressesAndContacts(unittest.TestCase): +class TestAddressesAndContacts(FrappeTestCase): def test_get_data(self): linked_docs = [get_custom_doc_for_address_and_contacts()] links_list = [item.name for item in linked_docs] diff --git a/frappe/core/doctype/access_log/test_access_log.py b/frappe/core/doctype/access_log/test_access_log.py index ee0422e11a..b3432d60bf 100644 --- a/frappe/core/doctype/access_log/test_access_log.py +++ b/frappe/core/doctype/access_log/test_access_log.py @@ -4,9 +4,6 @@ import base64 import os -# imports - standard imports -import unittest - # imports - third party imports import requests @@ -15,10 +12,13 @@ import frappe from frappe.core.doctype.access_log.access_log import make_access_log from frappe.core.doctype.data_import.data_import import export_csv from frappe.core.doctype.user.user import generate_keys + +# imports - standard imports +from frappe.tests.utils import FrappeTestCase from frappe.utils import cstr, get_site_url -class TestAccessLog(unittest.TestCase): +class TestAccessLog(FrappeTestCase): def setUp(self): # generate keys for current user to send requests for the following tests generate_keys(frappe.session.user) diff --git a/frappe/core/doctype/activity_log/test_activity_log.py b/frappe/core/doctype/activity_log/test_activity_log.py index c362fca521..f610813944 100644 --- a/frappe/core/doctype/activity_log/test_activity_log.py +++ b/frappe/core/doctype/activity_log/test_activity_log.py @@ -1,13 +1,13 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE import time -import unittest import frappe from frappe.auth import CookieManager, LoginManager +from frappe.tests.utils import FrappeTestCase -class TestActivityLog(unittest.TestCase): +class TestActivityLog(FrappeTestCase): def test_activity_log(self): # test user login log diff --git a/frappe/core/doctype/comment/test_comment.py b/frappe/core/doctype/comment/test_comment.py index bedcea6e7e..f40f1659b3 100644 --- a/frappe/core/doctype/comment/test_comment.py +++ b/frappe/core/doctype/comment/test_comment.py @@ -1,12 +1,12 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE import json -import unittest import frappe +from frappe.tests.utils import FrappeTestCase -class TestComment(unittest.TestCase): +class TestComment(FrappeTestCase): def tearDown(self): frappe.form_dict.comment = None frappe.form_dict.comment_email = None diff --git a/frappe/core/doctype/communication/test_communication.py b/frappe/core/doctype/communication/test_communication.py index 085dd3fe60..7c032a926d 100644 --- a/frappe/core/doctype/communication/test_communication.py +++ b/frappe/core/doctype/communication/test_communication.py @@ -1,6 +1,5 @@ # Copyright (c) 2022, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest from typing import TYPE_CHECKING from urllib.parse import quote diff --git a/frappe/core/doctype/custom_docperm/test_custom_docperm.py b/frappe/core/doctype/custom_docperm/test_custom_docperm.py index 4aa04f0223..bc113f1f8f 100644 --- a/frappe/core/doctype/custom_docperm/test_custom_docperm.py +++ b/frappe/core/doctype/custom_docperm/test_custom_docperm.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Custom DocPerm') -class TestCustomDocPerm(unittest.TestCase): +class TestCustomDocPerm(FrappeTestCase): pass diff --git a/frappe/core/doctype/custom_role/test_custom_role.py b/frappe/core/doctype/custom_role/test_custom_role.py index 79c66255c0..c81d70d3b5 100644 --- a/frappe/core/doctype/custom_role/test_custom_role.py +++ b/frappe/core/doctype/custom_role/test_custom_role.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Custom Role') -class TestCustomRole(unittest.TestCase): +class TestCustomRole(FrappeTestCase): pass diff --git a/frappe/core/doctype/data_export/test_data_exporter.py b/frappe/core/doctype/data_export/test_data_exporter.py index 812f65aaad..eb3ebaa80d 100644 --- a/frappe/core/doctype/data_export/test_data_exporter.py +++ b/frappe/core/doctype/data_export/test_data_exporter.py @@ -1,12 +1,11 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.core.doctype.data_export.exporter import DataExporter +from frappe.tests.utils import FrappeTestCase -class TestDataExporter(unittest.TestCase): +class TestDataExporter(FrappeTestCase): def setUp(self): self.doctype_name = "Test DocType for Export Tool" self.doc_name = "Test Data for Export Tool" diff --git a/frappe/core/doctype/data_import/test_data_import.py b/frappe/core/doctype/data_import/test_data_import.py index 253882383c..10f8098b2b 100644 --- a/frappe/core/doctype/data_import/test_data_import.py +++ b/frappe/core/doctype/data_import/test_data_import.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestDataImport(unittest.TestCase): +class TestDataImport(FrappeTestCase): pass diff --git a/frappe/core/doctype/data_import/test_exporter.py b/frappe/core/doctype/data_import/test_exporter.py index ceeac90e36..cd7f91d079 100644 --- a/frappe/core/doctype/data_import/test_exporter.py +++ b/frappe/core/doctype/data_import/test_exporter.py @@ -1,15 +1,14 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.core.doctype.data_import.exporter import Exporter from frappe.core.doctype.data_import.test_importer import create_doctype_if_not_exists +from frappe.tests.utils import FrappeTestCase doctype_name = "DocType for Export" -class TestExporter(unittest.TestCase): +class TestExporter(FrappeTestCase): def setUp(self): create_doctype_if_not_exists(doctype_name) diff --git a/frappe/core/doctype/data_import/test_importer.py b/frappe/core/doctype/data_import/test_importer.py index 2c250a4e87..1edaf5c1ed 100644 --- a/frappe/core/doctype/data_import/test_importer.py +++ b/frappe/core/doctype/data_import/test_importer.py @@ -1,18 +1,18 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.core.doctype.data_import.importer import Importer from frappe.tests.test_query_builder import db_type_is, run_only_if +from frappe.tests.utils import FrappeTestCase from frappe.utils import format_duration, getdate doctype_name = "DocType for Import" -class TestImporter(unittest.TestCase): +class TestImporter(FrappeTestCase): @classmethod def setUpClass(cls): + super().setUpClass() create_doctype_if_not_exists( doctype_name, ) diff --git a/frappe/core/doctype/data_import_log/test_data_import_log.py b/frappe/core/doctype/data_import_log/test_data_import_log.py index 6ae7c532bf..6db1b87b9b 100644 --- a/frappe/core/doctype/data_import_log/test_data_import_log.py +++ b/frappe/core/doctype/data_import_log/test_data_import_log.py @@ -2,8 +2,8 @@ # See license.txt # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestDataImportLog(unittest.TestCase): +class TestDataImportLog(FrappeTestCase): pass diff --git a/frappe/core/doctype/deleted_document/test_deleted_document.py b/frappe/core/doctype/deleted_document/test_deleted_document.py index f696689664..28d494556f 100644 --- a/frappe/core/doctype/deleted_document/test_deleted_document.py +++ b/frappe/core/doctype/deleted_document/test_deleted_document.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Deleted Document') -class TestDeletedDocument(unittest.TestCase): +class TestDeletedDocument(FrappeTestCase): pass diff --git a/frappe/core/doctype/docshare/test_docshare.py b/frappe/core/doctype/docshare/test_docshare.py index e374e4069d..f2ed8a32af 100644 --- a/frappe/core/doctype/docshare/test_docshare.py +++ b/frappe/core/doctype/docshare/test_docshare.py @@ -1,16 +1,15 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe import frappe.share from frappe.automation.doctype.auto_repeat.test_auto_repeat import create_submittable_doctype +from frappe.tests.utils import FrappeTestCase test_dependencies = ["User"] -class TestDocShare(unittest.TestCase): +class TestDocShare(FrappeTestCase): def setUp(self): self.user = "test@example.com" self.event = frappe.get_doc( diff --git a/frappe/core/doctype/doctype/test_doctype.py b/frappe/core/doctype/doctype/test_doctype.py index a083939c94..44f2877e2b 100644 --- a/frappe/core/doctype/doctype/test_doctype.py +++ b/frappe/core/doctype/doctype/test_doctype.py @@ -2,7 +2,6 @@ # License: MIT. See LICENSE import random import string -import unittest from unittest.mock import patch import frappe @@ -19,9 +18,10 @@ from frappe.core.doctype.doctype.doctype import ( ) from frappe.custom.doctype.custom_field.custom_field import create_custom_fields from frappe.desk.form.load import getdoc +from frappe.tests.utils import FrappeTestCase -class TestDocType(unittest.TestCase): +class TestDocType(FrappeTestCase): def tearDown(self): frappe.db.rollback() diff --git a/frappe/core/doctype/document_naming_rule/test_document_naming_rule.py b/frappe/core/doctype/document_naming_rule/test_document_naming_rule.py index cc406ed5cd..60e6803a1a 100644 --- a/frappe/core/doctype/document_naming_rule/test_document_naming_rule.py +++ b/frappe/core/doctype/document_naming_rule/test_document_naming_rule.py @@ -1,11 +1,10 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestDocumentNamingRule(unittest.TestCase): +class TestDocumentNamingRule(FrappeTestCase): def test_naming_rule_by_series(self): naming_rule = frappe.get_doc( dict(doctype="Document Naming Rule", document_type="ToDo", prefix="test-todo-", prefix_digits=5) diff --git a/frappe/core/doctype/document_naming_rule_condition/test_document_naming_rule_condition.py b/frappe/core/doctype/document_naming_rule_condition/test_document_naming_rule_condition.py index 68f0677f2c..edf52fc0c4 100644 --- a/frappe/core/doctype/document_naming_rule_condition/test_document_naming_rule_condition.py +++ b/frappe/core/doctype/document_naming_rule_condition/test_document_naming_rule_condition.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestDocumentNamingRuleCondition(unittest.TestCase): +class TestDocumentNamingRuleCondition(FrappeTestCase): pass diff --git a/frappe/core/doctype/document_share_key/test_document_share_key.py b/frappe/core/doctype/document_share_key/test_document_share_key.py index 10499fcc5d..23f91157e6 100644 --- a/frappe/core/doctype/document_share_key/test_document_share_key.py +++ b/frappe/core/doctype/document_share_key/test_document_share_key.py @@ -2,8 +2,8 @@ # See license.txt # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestDocumentShareKey(unittest.TestCase): +class TestDocumentShareKey(FrappeTestCase): pass diff --git a/frappe/core/doctype/domain/test_domain.py b/frappe/core/doctype/domain/test_domain.py index 32592705b4..92540cda6b 100644 --- a/frappe/core/doctype/domain/test_domain.py +++ b/frappe/core/doctype/domain/test_domain.py @@ -1,7 +1,7 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestDomain(unittest.TestCase): +class TestDomain(FrappeTestCase): pass diff --git a/frappe/core/doctype/error_log/test_error_log.py b/frappe/core/doctype/error_log/test_error_log.py index c7cf26a0cf..59670de8d2 100644 --- a/frappe/core/doctype/error_log/test_error_log.py +++ b/frappe/core/doctype/error_log/test_error_log.py @@ -1,13 +1,12 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Error Log') -class TestErrorLog(unittest.TestCase): +class TestErrorLog(FrappeTestCase): def test_error_log(self): """let's do an error log on error log?""" doc = frappe.new_doc("Error Log") diff --git a/frappe/core/doctype/error_snapshot/test_error_snapshot.py b/frappe/core/doctype/error_snapshot/test_error_snapshot.py index 0c1f861b43..8ff48bc5c6 100644 --- a/frappe/core/doctype/error_snapshot/test_error_snapshot.py +++ b/frappe/core/doctype/error_snapshot/test_error_snapshot.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Error Snapshot') -class TestErrorSnapshot(unittest.TestCase): +class TestErrorSnapshot(FrappeTestCase): pass diff --git a/frappe/core/doctype/file/test_file.py b/frappe/core/doctype/file/test_file.py index d849a2de4d..d496eec219 100644 --- a/frappe/core/doctype/file/test_file.py +++ b/frappe/core/doctype/file/test_file.py @@ -3,7 +3,6 @@ import base64 import json import os -import unittest from contextlib import contextmanager from typing import TYPE_CHECKING @@ -521,11 +520,12 @@ class TestFile(FrappeTestCase): assert test_file is not None -class TestAttachment(unittest.TestCase): +class TestAttachment(FrappeTestCase): test_doctype = "Test For Attachment" @classmethod def setUpClass(cls): + super().setUpClass() frappe.get_doc( doctype="DocType", name=cls.test_doctype, diff --git a/frappe/core/doctype/installed_applications/test_installed_applications.py b/frappe/core/doctype/installed_applications/test_installed_applications.py index b67cc4c8c7..854433ce40 100644 --- a/frappe/core/doctype/installed_applications/test_installed_applications.py +++ b/frappe/core/doctype/installed_applications/test_installed_applications.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestInstalledApplications(unittest.TestCase): +class TestInstalledApplications(FrappeTestCase): pass diff --git a/frappe/core/doctype/language/test_language.py b/frappe/core/doctype/language/test_language.py index 1f9c96a5d8..1f1e9cb913 100644 --- a/frappe/core/doctype/language/test_language.py +++ b/frappe/core/doctype/language/test_language.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Language') -class TestLanguage(unittest.TestCase): +class TestLanguage(FrappeTestCase): pass diff --git a/frappe/core/doctype/log_setting_user/test_log_setting_user.py b/frappe/core/doctype/log_setting_user/test_log_setting_user.py index 9ea56e8ec4..556dc36dc9 100644 --- a/frappe/core/doctype/log_setting_user/test_log_setting_user.py +++ b/frappe/core/doctype/log_setting_user/test_log_setting_user.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestLogSettingUser(unittest.TestCase): +class TestLogSettingUser(FrappeTestCase): pass diff --git a/frappe/core/doctype/module_def/test_module_def.py b/frappe/core/doctype/module_def/test_module_def.py index 914ba07949..e44741c894 100644 --- a/frappe/core/doctype/module_def/test_module_def.py +++ b/frappe/core/doctype/module_def/test_module_def.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Module Def') -class TestModuleDef(unittest.TestCase): +class TestModuleDef(FrappeTestCase): pass diff --git a/frappe/core/doctype/module_profile/test_module_profile.py b/frappe/core/doctype/module_profile/test_module_profile.py index 099d1371fb..a92adb0e74 100644 --- a/frappe/core/doctype/module_profile/test_module_profile.py +++ b/frappe/core/doctype/module_profile/test_module_profile.py @@ -1,11 +1,10 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestModuleProfile(unittest.TestCase): +class TestModuleProfile(FrappeTestCase): def test_make_new_module_profile(self): if not frappe.db.get_value("Module Profile", "_Test Module Profile"): frappe.get_doc( diff --git a/frappe/core/doctype/navbar_item/test_navbar_item.py b/frappe/core/doctype/navbar_item/test_navbar_item.py index 7eeb4f642b..7ad92a3ae8 100644 --- a/frappe/core/doctype/navbar_item/test_navbar_item.py +++ b/frappe/core/doctype/navbar_item/test_navbar_item.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestNavbarItem(unittest.TestCase): +class TestNavbarItem(FrappeTestCase): pass diff --git a/frappe/core/doctype/navbar_settings/test_navbar_settings.py b/frappe/core/doctype/navbar_settings/test_navbar_settings.py index 76fb3d298a..f6c2cf69cd 100644 --- a/frappe/core/doctype/navbar_settings/test_navbar_settings.py +++ b/frappe/core/doctype/navbar_settings/test_navbar_settings.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestNavbarSettings(unittest.TestCase): +class TestNavbarSettings(FrappeTestCase): pass diff --git a/frappe/core/doctype/package/test_package.py b/frappe/core/doctype/package/test_package.py index 0bd52a9e3c..8af076f1be 100644 --- a/frappe/core/doctype/package/test_package.py +++ b/frappe/core/doctype/package/test_package.py @@ -3,12 +3,12 @@ import json import os -import unittest import frappe +from frappe.tests.utils import FrappeTestCase -class TestPackage(unittest.TestCase): +class TestPackage(FrappeTestCase): def test_package_release(self): make_test_package() make_test_module() diff --git a/frappe/core/doctype/package_import/test_package_import.py b/frappe/core/doctype/package_import/test_package_import.py index 7e8008cc44..e4bb3d6715 100644 --- a/frappe/core/doctype/package_import/test_package_import.py +++ b/frappe/core/doctype/package_import/test_package_import.py @@ -2,8 +2,8 @@ # See license.txt # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestPackageImport(unittest.TestCase): +class TestPackageImport(FrappeTestCase): pass diff --git a/frappe/core/doctype/package_release/test_package_release.py b/frappe/core/doctype/package_release/test_package_release.py index e7b680463d..8cea4d0aff 100644 --- a/frappe/core/doctype/package_release/test_package_release.py +++ b/frappe/core/doctype/package_release/test_package_release.py @@ -2,8 +2,8 @@ # See license.txt # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestPackageRelease(unittest.TestCase): +class TestPackageRelease(FrappeTestCase): pass diff --git a/frappe/core/doctype/page/test_page.py b/frappe/core/doctype/page/test_page.py index 1eff54cad7..edf7f7c9b8 100644 --- a/frappe/core/doctype/page/test_page.py +++ b/frappe/core/doctype/page/test_page.py @@ -1,13 +1,12 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase test_records = frappe.get_test_records("Page") -class TestPage(unittest.TestCase): +class TestPage(FrappeTestCase): def test_naming(self): self.assertRaises( frappe.NameError, diff --git a/frappe/core/doctype/patch_log/test_patch_log.py b/frappe/core/doctype/patch_log/test_patch_log.py index 0c8a2ae4d4..849c3b30a2 100644 --- a/frappe/core/doctype/patch_log/test_patch_log.py +++ b/frappe/core/doctype/patch_log/test_patch_log.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Patch Log') -class TestPatchLog(unittest.TestCase): +class TestPatchLog(FrappeTestCase): pass diff --git a/frappe/core/doctype/prepared_report/test_prepared_report.py b/frappe/core/doctype/prepared_report/test_prepared_report.py index 6d0c809a01..cd96e8a18f 100644 --- a/frappe/core/doctype/prepared_report/test_prepared_report.py +++ b/frappe/core/doctype/prepared_report/test_prepared_report.py @@ -1,12 +1,12 @@ # Copyright (c) 2018, Frappe Technologies and Contributors # License: MIT. See LICENSE import json -import unittest import frappe +from frappe.tests.utils import FrappeTestCase -class TestPreparedReport(unittest.TestCase): +class TestPreparedReport(FrappeTestCase): def setUp(self): self.report = frappe.get_doc({"doctype": "Report", "name": "Permitted Documents For User"}) self.filters = {"user": "Administrator", "doctype": "Role"} diff --git a/frappe/core/doctype/role/test_role.py b/frappe/core/doctype/role/test_role.py index 44b9b1cdee..58aadfcbca 100644 --- a/frappe/core/doctype/role/test_role.py +++ b/frappe/core/doctype/role/test_role.py @@ -1,14 +1,14 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest import frappe from frappe.core.doctype.role.role import get_info_based_on_role +from frappe.tests.utils import FrappeTestCase test_records = frappe.get_test_records("Role") -class TestUser(unittest.TestCase): +class TestUser(FrappeTestCase): def test_disable_role(self): frappe.get_doc("User", "test@example.com").add_roles("_Test Role 3") diff --git a/frappe/core/doctype/role_profile/test_role_profile.py b/frappe/core/doctype/role_profile/test_role_profile.py index 726a5fc83e..34fb9741f9 100644 --- a/frappe/core/doctype/role_profile/test_role_profile.py +++ b/frappe/core/doctype/role_profile/test_role_profile.py @@ -1,13 +1,12 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase test_dependencies = ["Role"] -class TestRoleProfile(unittest.TestCase): +class TestRoleProfile(FrappeTestCase): def test_make_new_role_profile(self): frappe.delete_doc_if_exists("Role Profile", "Test 1", force=1) new_role_profile = frappe.get_doc(dict(doctype="Role Profile", role_profile="Test 1")).insert() diff --git a/frappe/core/doctype/scheduled_job_log/test_scheduled_job_log.py b/frappe/core/doctype/scheduled_job_log/test_scheduled_job_log.py index 11d60e35d8..6fd187b4e4 100644 --- a/frappe/core/doctype/scheduled_job_log/test_scheduled_job_log.py +++ b/frappe/core/doctype/scheduled_job_log/test_scheduled_job_log.py @@ -1,8 +1,8 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestScheduledJobLog(unittest.TestCase): +class TestScheduledJobLog(FrappeTestCase): pass diff --git a/frappe/core/doctype/scheduled_job_type/test_scheduled_job_type.py b/frappe/core/doctype/scheduled_job_type/test_scheduled_job_type.py index 5448bda91f..bbc92dfbc9 100644 --- a/frappe/core/doctype/scheduled_job_type/test_scheduled_job_type.py +++ b/frappe/core/doctype/scheduled_job_type/test_scheduled_job_type.py @@ -1,13 +1,12 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.core.doctype.scheduled_job_type.scheduled_job_type import sync_jobs +from frappe.tests.utils import FrappeTestCase from frappe.utils import get_datetime -class TestScheduledJobType(unittest.TestCase): +class TestScheduledJobType(FrappeTestCase): def setUp(self): frappe.db.rollback() frappe.db.truncate("Scheduled Job Type") diff --git a/frappe/core/doctype/server_script/test_server_script.py b/frappe/core/doctype/server_script/test_server_script.py index 4c1c12b7f2..7002de9691 100644 --- a/frappe/core/doctype/server_script/test_server_script.py +++ b/frappe/core/doctype/server_script/test_server_script.py @@ -1,10 +1,9 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import requests import frappe +from frappe.tests.utils import FrappeTestCase from frappe.utils import get_site_url scripts = [ @@ -86,10 +85,10 @@ frappe.db.add_index("Todo", ["color", "date"]) ] -class TestServerScript(unittest.TestCase): +class TestServerScript(FrappeTestCase): @classmethod def setUpClass(cls): - frappe.db.commit() + super().setUpClass() frappe.db.truncate("Server Script") frappe.get_doc("User", "Administrator").add_roles("Script Manager") for script in scripts: diff --git a/frappe/core/doctype/session_default_settings/test_session_default_settings.py b/frappe/core/doctype/session_default_settings/test_session_default_settings.py index aa60085ce9..532e805141 100644 --- a/frappe/core/doctype/session_default_settings/test_session_default_settings.py +++ b/frappe/core/doctype/session_default_settings/test_session_default_settings.py @@ -1,15 +1,14 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.core.doctype.session_default_settings.session_default_settings import ( clear_session_defaults, set_session_default_values, ) +from frappe.tests.utils import FrappeTestCase -class TestSessionDefaultSettings(unittest.TestCase): +class TestSessionDefaultSettings(FrappeTestCase): def test_set_session_default_settings(self): frappe.set_user("Administrator") settings = frappe.get_single("Session Default Settings") diff --git a/frappe/core/doctype/sms_settings/test_sms_settings.py b/frappe/core/doctype/sms_settings/test_sms_settings.py index 61be20ff66..56cbe9b163 100644 --- a/frappe/core/doctype/sms_settings/test_sms_settings.py +++ b/frappe/core/doctype/sms_settings/test_sms_settings.py @@ -1,7 +1,7 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestSMSSettings(unittest.TestCase): +class TestSMSSettings(FrappeTestCase): pass diff --git a/frappe/core/doctype/system_settings/test_system_settings.py b/frappe/core/doctype/system_settings/test_system_settings.py index b126976eeb..a876e8301d 100644 --- a/frappe/core/doctype/system_settings/test_system_settings.py +++ b/frappe/core/doctype/system_settings/test_system_settings.py @@ -1,7 +1,7 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestSystemSettings(unittest.TestCase): +class TestSystemSettings(FrappeTestCase): pass diff --git a/frappe/core/doctype/transaction_log/test_transaction_log.py b/frappe/core/doctype/transaction_log/test_transaction_log.py index 8b179f8d85..0a76e5ac65 100644 --- a/frappe/core/doctype/transaction_log/test_transaction_log.py +++ b/frappe/core/doctype/transaction_log/test_transaction_log.py @@ -1,14 +1,14 @@ # Copyright (c) 2018, Frappe Technologies and Contributors # License: MIT. See LICENSE import hashlib -import unittest import frappe +from frappe.tests.utils import FrappeTestCase test_records = [] -class TestTransactionLog(unittest.TestCase): +class TestTransactionLog(FrappeTestCase): def test_validate_chaining(self): frappe.get_doc( { diff --git a/frappe/core/doctype/translation/test_translation.py b/frappe/core/doctype/translation/test_translation.py index c9f4e85086..581cda83f0 100644 --- a/frappe/core/doctype/translation/test_translation.py +++ b/frappe/core/doctype/translation/test_translation.py @@ -1,12 +1,11 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe import _ +from frappe.tests.utils import FrappeTestCase -class TestTranslation(unittest.TestCase): +class TestTranslation(FrappeTestCase): def setUp(self): frappe.db.delete("Translation") diff --git a/frappe/core/doctype/user/test_user.py b/frappe/core/doctype/user/test_user.py index d4536c6355..d6f51f4c67 100644 --- a/frappe/core/doctype/user/test_user.py +++ b/frappe/core/doctype/user/test_user.py @@ -2,7 +2,6 @@ # License: MIT. See LICENSE import json import time -import unittest from unittest.mock import patch import frappe @@ -18,13 +17,14 @@ from frappe.core.doctype.user.user import ( from frappe.desk.notifications import extract_mentions from frappe.frappeclient import FrappeClient from frappe.model.delete_doc import delete_doc +from frappe.tests.utils import FrappeTestCase from frappe.utils import get_url user_module = frappe.core.doctype.user.user test_records = frappe.get_test_records("User") -class TestUser(unittest.TestCase): +class TestUser(FrappeTestCase): def tearDown(self): # disable password strength test frappe.db.set_value("System Settings", "System Settings", "enable_password_policy", 0) diff --git a/frappe/core/doctype/user_group/test_user_group.py b/frappe/core/doctype/user_group/test_user_group.py index 368f4eaef2..79e013672d 100644 --- a/frappe/core/doctype/user_group/test_user_group.py +++ b/frappe/core/doctype/user_group/test_user_group.py @@ -1,8 +1,8 @@ # Copyright (c) 2021, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestUserGroup(unittest.TestCase): +class TestUserGroup(FrappeTestCase): pass diff --git a/frappe/core/doctype/user_group_member/test_user_group_member.py b/frappe/core/doctype/user_group_member/test_user_group_member.py index 5d709d0bec..a2eb5c7bfc 100644 --- a/frappe/core/doctype/user_group_member/test_user_group_member.py +++ b/frappe/core/doctype/user_group_member/test_user_group_member.py @@ -1,8 +1,8 @@ # Copyright (c) 2021, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestUserGroupMember(unittest.TestCase): +class TestUserGroupMember(FrappeTestCase): pass diff --git a/frappe/core/doctype/user_permission/test_user_permission.py b/frappe/core/doctype/user_permission/test_user_permission.py index b963da2f49..10dc75ba39 100644 --- a/frappe/core/doctype/user_permission/test_user_permission.py +++ b/frappe/core/doctype/user_permission/test_user_permission.py @@ -1,7 +1,5 @@ # Copyright (c) 2021, Frappe Technologies and Contributors # See LICENSE -import unittest - import frappe from frappe.core.doctype.doctype.test_doctype import new_doctype from frappe.core.doctype.user_permission.user_permission import ( @@ -9,10 +7,11 @@ from frappe.core.doctype.user_permission.user_permission import ( remove_applicable, ) from frappe.permissions import has_user_permission +from frappe.tests.utils import FrappeTestCase from frappe.website.doctype.blog_post.test_blog_post import make_test_blog -class TestUserPermission(unittest.TestCase): +class TestUserPermission(FrappeTestCase): def setUp(self): test_users = ( "test_bulk_creation_update@example.com", diff --git a/frappe/core/doctype/user_type/test_user_type.py b/frappe/core/doctype/user_type/test_user_type.py index 235881517a..ec1a5e3bfc 100644 --- a/frappe/core/doctype/user_type/test_user_type.py +++ b/frappe/core/doctype/user_type/test_user_type.py @@ -1,12 +1,11 @@ # Copyright (c) 2021, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.installer import update_site_config +from frappe.tests.utils import FrappeTestCase -class TestUserType(unittest.TestCase): +class TestUserType(FrappeTestCase): def setUp(self): create_role() diff --git a/frappe/core/doctype/version/test_version.py b/frappe/core/doctype/version/test_version.py index 3e82f30f06..ce8e0e8b89 100644 --- a/frappe/core/doctype/version/test_version.py +++ b/frappe/core/doctype/version/test_version.py @@ -1,14 +1,14 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE import copy -import unittest import frappe from frappe.core.doctype.version.version import get_diff from frappe.test_runner import make_test_objects +from frappe.tests.utils import FrappeTestCase -class TestVersion(unittest.TestCase): +class TestVersion(FrappeTestCase): def test_get_diff(self): frappe.set_user("Administrator") test_records = make_test_objects("Event", reset=True) diff --git a/frappe/core/doctype/view_log/test_view_log.py b/frappe/core/doctype/view_log/test_view_log.py index 5a88269028..d1596d84a4 100644 --- a/frappe/core/doctype/view_log/test_view_log.py +++ b/frappe/core/doctype/view_log/test_view_log.py @@ -1,11 +1,10 @@ # Copyright (c) 2018, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestViewLog(unittest.TestCase): +class TestViewLog(FrappeTestCase): def tearDown(self): frappe.set_user("Administrator") diff --git a/frappe/custom/doctype/client_script/test_client_script.py b/frappe/custom/doctype/client_script/test_client_script.py index c93df04c98..9b50a3d0b0 100644 --- a/frappe/custom/doctype/client_script/test_client_script.py +++ b/frappe/custom/doctype/client_script/test_client_script.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Client Script') -class TestClientScript(unittest.TestCase): +class TestClientScript(FrappeTestCase): pass diff --git a/frappe/custom/doctype/custom_field/test_custom_field.py b/frappe/custom/doctype/custom_field/test_custom_field.py index 34223315c5..dfdaf7ea9a 100644 --- a/frappe/custom/doctype/custom_field/test_custom_field.py +++ b/frappe/custom/doctype/custom_field/test_custom_field.py @@ -1,14 +1,13 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase test_records = frappe.get_test_records("Custom Field") -class TestCustomField(unittest.TestCase): +class TestCustomField(FrappeTestCase): def test_create_custom_fields(self): from .custom_field import create_custom_fields diff --git a/frappe/custom/doctype/customize_form/test_customize_form.py b/frappe/custom/doctype/customize_form/test_customize_form.py index b00f45f5d2..6c4564c958 100644 --- a/frappe/custom/doctype/customize_form/test_customize_form.py +++ b/frappe/custom/doctype/customize_form/test_customize_form.py @@ -2,17 +2,17 @@ # License: MIT. See LICENSE import json -import unittest import frappe from frappe.core.doctype.doctype.doctype import InvalidFieldNameError from frappe.core.doctype.doctype.test_doctype import new_doctype from frappe.test_runner import make_test_records_for_doctype +from frappe.tests.utils import FrappeTestCase test_dependencies = ["Custom Field", "Property Setter"] -class TestCustomizeForm(unittest.TestCase): +class TestCustomizeForm(FrappeTestCase): def insert_custom_field(self): frappe.delete_doc_if_exists("Custom Field", "Event-test_custom_field") frappe.get_doc( diff --git a/frappe/custom/doctype/doctype_layout/test_doctype_layout.py b/frappe/custom/doctype/doctype_layout/test_doctype_layout.py index 0e64a9e727..c568cd4df7 100644 --- a/frappe/custom/doctype/doctype_layout/test_doctype_layout.py +++ b/frappe/custom/doctype/doctype_layout/test_doctype_layout.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestDocTypeLayout(unittest.TestCase): +class TestDocTypeLayout(FrappeTestCase): pass diff --git a/frappe/custom/doctype/property_setter/test_property_setter.py b/frappe/custom/doctype/property_setter/test_property_setter.py index 1fa2d2cefb..dc74a69161 100644 --- a/frappe/custom/doctype/property_setter/test_property_setter.py +++ b/frappe/custom/doctype/property_setter/test_property_setter.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Property Setter') -class TestPropertySetter(unittest.TestCase): +class TestPropertySetter(FrappeTestCase): pass diff --git a/frappe/desk/doctype/console_log/test_console_log.py b/frappe/desk/doctype/console_log/test_console_log.py index 0579098382..9ae8d1f1e8 100644 --- a/frappe/desk/doctype/console_log/test_console_log.py +++ b/frappe/desk/doctype/console_log/test_console_log.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestConsoleLog(unittest.TestCase): +class TestConsoleLog(FrappeTestCase): pass diff --git a/frappe/desk/doctype/dashboard/test_dashboard.py b/frappe/desk/doctype/dashboard/test_dashboard.py index d2ba871509..99aeecaee6 100644 --- a/frappe/desk/doctype/dashboard/test_dashboard.py +++ b/frappe/desk/doctype/dashboard/test_dashboard.py @@ -1,7 +1,7 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestDashboard(unittest.TestCase): +class TestDashboard(FrappeTestCase): pass diff --git a/frappe/desk/doctype/dashboard_chart_source/test_dashboard_chart_source.py b/frappe/desk/doctype/dashboard_chart_source/test_dashboard_chart_source.py index 457487bb6d..5b4c114287 100644 --- a/frappe/desk/doctype/dashboard_chart_source/test_dashboard_chart_source.py +++ b/frappe/desk/doctype/dashboard_chart_source/test_dashboard_chart_source.py @@ -1,7 +1,7 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestDashboardChartSource(unittest.TestCase): +class TestDashboardChartSource(FrappeTestCase): pass diff --git a/frappe/desk/doctype/event/test_event.py b/frappe/desk/doctype/event/test_event.py index efbd54fb09..72eab8f416 100644 --- a/frappe/desk/doctype/event/test_event.py +++ b/frappe/desk/doctype/event/test_event.py @@ -3,17 +3,17 @@ """Use blog post test to test user permissions logic""" import json -import unittest import frappe import frappe.defaults from frappe.desk.doctype.event.event import get_events from frappe.test_runner import make_test_objects +from frappe.tests.utils import FrappeTestCase test_records = frappe.get_test_records("Event") -class TestEvent(unittest.TestCase): +class TestEvent(FrappeTestCase): def setUp(self): frappe.db.delete("Event") make_test_objects("Event", reset=True) diff --git a/frappe/desk/doctype/form_tour/test_form_tour.py b/frappe/desk/doctype/form_tour/test_form_tour.py index cb0c4ef33a..3fdcdf95a6 100644 --- a/frappe/desk/doctype/form_tour/test_form_tour.py +++ b/frappe/desk/doctype/form_tour/test_form_tour.py @@ -2,8 +2,8 @@ # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestFormTour(unittest.TestCase): +class TestFormTour(FrappeTestCase): pass diff --git a/frappe/desk/doctype/kanban_board/test_kanban_board.py b/frappe/desk/doctype/kanban_board/test_kanban_board.py index 73f566b906..05cd5723c4 100644 --- a/frappe/desk/doctype/kanban_board/test_kanban_board.py +++ b/frappe/desk/doctype/kanban_board/test_kanban_board.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Kanban Board') -class TestKanbanBoard(unittest.TestCase): +class TestKanbanBoard(FrappeTestCase): pass diff --git a/frappe/desk/doctype/list_view_settings/test_list_view_settings.py b/frappe/desk/doctype/list_view_settings/test_list_view_settings.py index 0eab9cd7a6..8e27385230 100644 --- a/frappe/desk/doctype/list_view_settings/test_list_view_settings.py +++ b/frappe/desk/doctype/list_view_settings/test_list_view_settings.py @@ -1,8 +1,8 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestListViewSettings(unittest.TestCase): +class TestListViewSettings(FrappeTestCase): pass diff --git a/frappe/desk/doctype/module_onboarding/test_module_onboarding.py b/frappe/desk/doctype/module_onboarding/test_module_onboarding.py index fa19794c1e..e84a5c4d2b 100644 --- a/frappe/desk/doctype/module_onboarding/test_module_onboarding.py +++ b/frappe/desk/doctype/module_onboarding/test_module_onboarding.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestModuleOnboarding(unittest.TestCase): +class TestModuleOnboarding(FrappeTestCase): pass diff --git a/frappe/desk/doctype/note/test_note.py b/frappe/desk/doctype/note/test_note.py index d8bdb9efc4..426fb5a16e 100644 --- a/frappe/desk/doctype/note/test_note.py +++ b/frappe/desk/doctype/note/test_note.py @@ -1,14 +1,13 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase test_records = frappe.get_test_records("Note") -class TestNote(unittest.TestCase): +class TestNote(FrappeTestCase): def insert_note(self): frappe.db.delete("Version") frappe.db.delete("Note") diff --git a/frappe/desk/doctype/notification_log/test_notification_log.py b/frappe/desk/doctype/notification_log/test_notification_log.py index 532f05ab57..6606edcb25 100644 --- a/frappe/desk/doctype/notification_log/test_notification_log.py +++ b/frappe/desk/doctype/notification_log/test_notification_log.py @@ -1,13 +1,12 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.core.doctype.user.user import get_system_users from frappe.desk.form.assign_to import add as assign_task +from frappe.tests.utils import FrappeTestCase -class TestNotificationLog(unittest.TestCase): +class TestNotificationLog(FrappeTestCase): def test_assignment(self): todo = get_todo() user = get_user() diff --git a/frappe/desk/doctype/notification_settings/test_notification_settings.py b/frappe/desk/doctype/notification_settings/test_notification_settings.py index 966b923567..e88f98443b 100644 --- a/frappe/desk/doctype/notification_settings/test_notification_settings.py +++ b/frappe/desk/doctype/notification_settings/test_notification_settings.py @@ -2,8 +2,8 @@ # See license.txt # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestNotificationSettings(unittest.TestCase): +class TestNotificationSettings(FrappeTestCase): pass diff --git a/frappe/desk/doctype/number_card/test_number_card.py b/frappe/desk/doctype/number_card/test_number_card.py index c0dda40104..a002b8aad3 100644 --- a/frappe/desk/doctype/number_card/test_number_card.py +++ b/frappe/desk/doctype/number_card/test_number_card.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestNumberCard(unittest.TestCase): +class TestNumberCard(FrappeTestCase): pass diff --git a/frappe/desk/doctype/onboarding_permission/test_onboarding_permission.py b/frappe/desk/doctype/onboarding_permission/test_onboarding_permission.py index cdfe0d7890..d82cb3c346 100644 --- a/frappe/desk/doctype/onboarding_permission/test_onboarding_permission.py +++ b/frappe/desk/doctype/onboarding_permission/test_onboarding_permission.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestOnboardingPermission(unittest.TestCase): +class TestOnboardingPermission(FrappeTestCase): pass diff --git a/frappe/desk/doctype/onboarding_step/test_onboarding_step.py b/frappe/desk/doctype/onboarding_step/test_onboarding_step.py index d8bf55584c..73b9ab4ac3 100644 --- a/frappe/desk/doctype/onboarding_step/test_onboarding_step.py +++ b/frappe/desk/doctype/onboarding_step/test_onboarding_step.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestOnboardingStep(unittest.TestCase): +class TestOnboardingStep(FrappeTestCase): pass diff --git a/frappe/desk/doctype/system_console/test_system_console.py b/frappe/desk/doctype/system_console/test_system_console.py index 96bf555f59..2664f7c925 100644 --- a/frappe/desk/doctype/system_console/test_system_console.py +++ b/frappe/desk/doctype/system_console/test_system_console.py @@ -1,11 +1,10 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestSystemConsole(unittest.TestCase): +class TestSystemConsole(FrappeTestCase): def test_system_console(self): system_console = frappe.get_doc("System Console") system_console.console = 'log("hello")' diff --git a/frappe/desk/doctype/tag/test_tag.py b/frappe/desk/doctype/tag/test_tag.py index 8719da8c21..0c746e67ac 100644 --- a/frappe/desk/doctype/tag/test_tag.py +++ b/frappe/desk/doctype/tag/test_tag.py @@ -1,11 +1,10 @@ -import unittest - import frappe from frappe.desk.doctype.tag.tag import add_tag from frappe.desk.reportview import get_stats +from frappe.tests.utils import FrappeTestCase -class TestTag(unittest.TestCase): +class TestTag(FrappeTestCase): def setUp(self) -> None: frappe.db.delete("Tag") frappe.db.sql("UPDATE `tabDocType` set _user_tags=''") diff --git a/frappe/desk/doctype/tag_link/test_tag_link.py b/frappe/desk/doctype/tag_link/test_tag_link.py index 59d7bcd2bc..10edb2859e 100644 --- a/frappe/desk/doctype/tag_link/test_tag_link.py +++ b/frappe/desk/doctype/tag_link/test_tag_link.py @@ -1,8 +1,8 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestTagLink(unittest.TestCase): +class TestTagLink(FrappeTestCase): pass diff --git a/frappe/desk/doctype/todo/test_todo.py b/frappe/desk/doctype/todo/test_todo.py index 56ca1f30e7..4880fad9d5 100644 --- a/frappe/desk/doctype/todo/test_todo.py +++ b/frappe/desk/doctype/todo/test_todo.py @@ -1,16 +1,15 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.core.doctype.doctype.doctype import clear_permissions_cache from frappe.model.db_query import DatabaseQuery from frappe.permissions import add_permission, reset_perms +from frappe.tests.utils import FrappeTestCase test_dependencies = ["User"] -class TestToDo(unittest.TestCase): +class TestToDo(FrappeTestCase): def test_delete(self): todo = frappe.get_doc( dict(doctype="ToDo", description="test todo", assigned_by="Administrator") diff --git a/frappe/desk/doctype/workspace/test_workspace.py b/frappe/desk/doctype/workspace/test_workspace.py index d0b0eba9e2..22665e1dd2 100644 --- a/frappe/desk/doctype/workspace/test_workspace.py +++ b/frappe/desk/doctype/workspace/test_workspace.py @@ -1,11 +1,10 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestWorkspace(unittest.TestCase): +class TestWorkspace(FrappeTestCase): def setUp(self): create_module("Test Module") diff --git a/frappe/desk/form/test_form.py b/frappe/desk/form/test_form.py index c05a932241..f256b03d27 100644 --- a/frappe/desk/form/test_form.py +++ b/frappe/desk/form/test_form.py @@ -1,13 +1,12 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.desk.form.linked_with import get_linked_docs, get_linked_doctypes +from frappe.tests.utils import FrappeTestCase -class TestForm(unittest.TestCase): +class TestForm(FrappeTestCase): def test_linked_with(self): results = get_linked_docs("Role", "System Manager", linkinfo=get_linked_doctypes("Role")) self.assertTrue("User" in results) @@ -15,5 +14,7 @@ class TestForm(unittest.TestCase): if __name__ == "__main__": + import unittest + frappe.connect() unittest.main() diff --git a/frappe/email/doctype/auto_email_report/test_auto_email_report.py b/frappe/email/doctype/auto_email_report/test_auto_email_report.py index ee0a363bd9..1f806f22b5 100644 --- a/frappe/email/doctype/auto_email_report/test_auto_email_report.py +++ b/frappe/email/doctype/auto_email_report/test_auto_email_report.py @@ -1,16 +1,16 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE import json -import unittest import frappe +from frappe.tests.utils import FrappeTestCase from frappe.utils import add_to_date, get_link_to_form, today from frappe.utils.data import is_html # test_records = frappe.get_test_records('Auto Email Report') -class TestAutoEmailReport(unittest.TestCase): +class TestAutoEmailReport(FrappeTestCase): def test_auto_email(self): frappe.delete_doc("Auto Email Report", "Permitted Documents For User") diff --git a/frappe/email/doctype/document_follow/test_document_follow.py b/frappe/email/doctype/document_follow/test_document_follow.py index 159a399ee8..845f6e93bb 100644 --- a/frappe/email/doctype/document_follow/test_document_follow.py +++ b/frappe/email/doctype/document_follow/test_document_follow.py @@ -1,6 +1,5 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest from dataclasses import dataclass import frappe @@ -12,9 +11,10 @@ from frappe.desk.like import toggle_like from frappe.query_builder import DocType from frappe.query_builder.functions import Cast_ from frappe.share import add as share +from frappe.tests.utils import FrappeTestCase -class TestDocumentFollow(unittest.TestCase): +class TestDocumentFollow(FrappeTestCase): def test_document_follow_version(self): user = get_user() event_doc = get_event() diff --git a/frappe/email/doctype/email_account/test_email_account.py b/frappe/email/doctype/email_account/test_email_account.py index 9ab004cdd0..783249071a 100644 --- a/frappe/email/doctype/email_account/test_email_account.py +++ b/frappe/email/doctype/email_account/test_email_account.py @@ -3,7 +3,6 @@ import email import os -import unittest from datetime import datetime, timedelta from unittest.mock import patch @@ -14,14 +13,16 @@ from frappe.email.doctype.email_account.email_account import notify_unreplied from frappe.email.email_body import get_message_id from frappe.email.receive import Email, InboundMail, SentEmailInInboxError from frappe.test_runner import make_test_records +from frappe.tests.utils import FrappeTestCase make_test_records("User") make_test_records("Email Account") -class TestEmailAccount(unittest.TestCase): +class TestEmailAccount(FrappeTestCase): @classmethod def setUpClass(cls): + super().setUpClass() email_account = frappe.get_doc("Email Account", "_Test Email Account 1") email_account.db_set("enable_incoming", 1) email_account.db_set("enable_auto_reply", 1) @@ -434,9 +435,10 @@ class TestEmailAccount(unittest.TestCase): email_account.receive() -class TestInboundMail(unittest.TestCase): +class TestInboundMail(FrappeTestCase): @classmethod def setUpClass(cls): + super().setUpClass() email_account = frappe.get_doc("Email Account", "_Test Email Account 1") email_account.db_set("enable_incoming", 1) diff --git a/frappe/email/doctype/email_domain/test_email_domain.py b/frappe/email/doctype/email_domain/test_email_domain.py index 55a8d620a8..4cb39c52db 100644 --- a/frappe/email/doctype/email_domain/test_email_domain.py +++ b/frappe/email/doctype/email_domain/test_email_domain.py @@ -1,14 +1,13 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.test_runner import make_test_objects +from frappe.tests.utils import FrappeTestCase test_records = frappe.get_test_records("Email Domain") -class TestDomain(unittest.TestCase): +class TestDomain(FrappeTestCase): def setUp(self): make_test_objects("Email Domain", reset=True) diff --git a/frappe/email/doctype/email_flag_queue/test_email_flag_queue.py b/frappe/email/doctype/email_flag_queue/test_email_flag_queue.py index 8bfc9230a4..0d731c37da 100644 --- a/frappe/email/doctype/email_flag_queue/test_email_flag_queue.py +++ b/frappe/email/doctype/email_flag_queue/test_email_flag_queue.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Email Flag Queue') -class TestEmailFlagQueue(unittest.TestCase): +class TestEmailFlagQueue(FrappeTestCase): pass diff --git a/frappe/email/doctype/email_group/test_email_group.py b/frappe/email/doctype/email_group/test_email_group.py index aa41285f90..ffc325a6bd 100644 --- a/frappe/email/doctype/email_group/test_email_group.py +++ b/frappe/email/doctype/email_group/test_email_group.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Email Group') -class TestEmailGroup(unittest.TestCase): +class TestEmailGroup(FrappeTestCase): pass diff --git a/frappe/email/doctype/email_group_member/test_email_group_member.py b/frappe/email/doctype/email_group_member/test_email_group_member.py index 749792fe52..6432cadf54 100644 --- a/frappe/email/doctype/email_group_member/test_email_group_member.py +++ b/frappe/email/doctype/email_group_member/test_email_group_member.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Email Group Member') -class TestEmailGroupMember(unittest.TestCase): +class TestEmailGroupMember(FrappeTestCase): pass diff --git a/frappe/email/doctype/email_rule/test_email_rule.py b/frappe/email/doctype/email_rule/test_email_rule.py index 1da5d34d6b..489379451d 100644 --- a/frappe/email/doctype/email_rule/test_email_rule.py +++ b/frappe/email/doctype/email_rule/test_email_rule.py @@ -1,7 +1,7 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestEmailRule(unittest.TestCase): +class TestEmailRule(FrappeTestCase): pass diff --git a/frappe/email/doctype/email_template/test_email_template.py b/frappe/email/doctype/email_template/test_email_template.py index 291f7e1df0..f2c53647bc 100644 --- a/frappe/email/doctype/email_template/test_email_template.py +++ b/frappe/email/doctype/email_template/test_email_template.py @@ -1,7 +1,7 @@ # Copyright (c) 2018, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestEmailTemplate(unittest.TestCase): +class TestEmailTemplate(FrappeTestCase): pass diff --git a/frappe/email/doctype/email_unsubscribe/test_email_unsubscribe.py b/frappe/email/doctype/email_unsubscribe/test_email_unsubscribe.py index 9ba99a6690..d0b1731ad9 100644 --- a/frappe/email/doctype/email_unsubscribe/test_email_unsubscribe.py +++ b/frappe/email/doctype/email_unsubscribe/test_email_unsubscribe.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Email Unsubscribe') -class TestEmailUnsubscribe(unittest.TestCase): +class TestEmailUnsubscribe(FrappeTestCase): pass diff --git a/frappe/email/doctype/notification/test_notification.py b/frappe/email/doctype/notification/test_notification.py index 0f570b1fd3..576d9e9a66 100644 --- a/frappe/email/doctype/notification/test_notification.py +++ b/frappe/email/doctype/notification/test_notification.py @@ -1,12 +1,13 @@ # Copyright (c) 2018, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest + from contextlib import contextmanager import frappe import frappe.utils import frappe.utils.scheduler from frappe.desk.form import assign_to +from frappe.tests.utils import FrappeTestCase test_dependencies = ["User", "Notification"] @@ -20,7 +21,7 @@ def get_test_notification(config): notification.delete() -class TestNotification(unittest.TestCase): +class TestNotification(FrappeTestCase): def setUp(self): frappe.db.delete("Email Queue") frappe.set_user("test@example.com") diff --git a/frappe/email/doctype/unhandled_email/test_unhandled_email.py b/frappe/email/doctype/unhandled_email/test_unhandled_email.py index debc52d685..0d36b6b091 100644 --- a/frappe/email/doctype/unhandled_email/test_unhandled_email.py +++ b/frappe/email/doctype/unhandled_email/test_unhandled_email.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Unhandled Emails') -class TestUnhandledEmail(unittest.TestCase): +class TestUnhandledEmail(FrappeTestCase): pass diff --git a/frappe/email/test_email_body.py b/frappe/email/test_email_body.py index d5b1013a73..b9128bc979 100644 --- a/frappe/email/test_email_body.py +++ b/frappe/email/test_email_body.py @@ -3,7 +3,6 @@ import base64 import os -import unittest import frappe from frappe import safe_decode @@ -15,9 +14,10 @@ from frappe.email.email_body import ( replace_filename_with_cid, ) from frappe.email.receive import Email +from frappe.tests.utils import FrappeTestCase -class TestEmailBody(unittest.TestCase): +class TestEmailBody(FrappeTestCase): def setUp(self): email_html = """
diff --git a/frappe/email/test_smtp.py b/frappe/email/test_smtp.py index 448f61925c..b101f610a1 100644 --- a/frappe/email/test_smtp.py +++ b/frappe/email/test_smtp.py @@ -1,14 +1,13 @@ # Copyright (c) 2020, Frappe Technologies Pvt. Ltd. and Contributors # License: The MIT License -import unittest - import frappe from frappe.email.doctype.email_account.email_account import EmailAccount from frappe.email.smtp import SMTPServer +from frappe.tests.utils import FrappeTestCase -class TestSMTP(unittest.TestCase): +class TestSMTP(FrappeTestCase): def test_smtp_ssl_session(self): for port in [None, 0, 465, "465"]: make_server(port, 1, 0) diff --git a/frappe/event_streaming/doctype/document_type_mapping/test_document_type_mapping.py b/frappe/event_streaming/doctype/document_type_mapping/test_document_type_mapping.py index 676d5040ff..defaa8b9c6 100644 --- a/frappe/event_streaming/doctype/document_type_mapping/test_document_type_mapping.py +++ b/frappe/event_streaming/doctype/document_type_mapping/test_document_type_mapping.py @@ -1,8 +1,8 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestDocumentTypeMapping(unittest.TestCase): +class TestDocumentTypeMapping(FrappeTestCase): pass diff --git a/frappe/event_streaming/doctype/event_consumer/test_event_consumer.py b/frappe/event_streaming/doctype/event_consumer/test_event_consumer.py index 6f04af643e..54bf718f17 100644 --- a/frappe/event_streaming/doctype/event_consumer/test_event_consumer.py +++ b/frappe/event_streaming/doctype/event_consumer/test_event_consumer.py @@ -1,8 +1,8 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestEventConsumer(unittest.TestCase): +class TestEventConsumer(FrappeTestCase): pass diff --git a/frappe/event_streaming/doctype/event_producer/test_event_producer.py b/frappe/event_streaming/doctype/event_producer/test_event_producer.py index 168c9a61cf..70e7483f92 100644 --- a/frappe/event_streaming/doctype/event_producer/test_event_producer.py +++ b/frappe/event_streaming/doctype/event_producer/test_event_producer.py @@ -1,7 +1,6 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE import json -import unittest import frappe from frappe.core.doctype.user.user import generate_keys @@ -9,19 +8,12 @@ from frappe.event_streaming.doctype.event_producer.event_producer import pull_fr from frappe.frappeclient import FrappeClient from frappe.query_builder.utils import db_type_is from frappe.tests.test_query_builder import run_only_if +from frappe.tests.utils import FrappeTestCase producer_url = "http://test_site_producer:8000" -class TestEventProducer(unittest.TestCase): - # @classmethod - # def setUpClass(cls): - # frappe.print_sql(True) - - # @classmethod - # def tearDownClass(cls): - # frappe.print_sql(False) - +class TestEventProducer(FrappeTestCase): def setUp(self): create_event_producer(producer_url) diff --git a/frappe/event_streaming/doctype/event_producer_last_update/test_event_producer_last_update.py b/frappe/event_streaming/doctype/event_producer_last_update/test_event_producer_last_update.py index ccdea6c694..3e68159790 100644 --- a/frappe/event_streaming/doctype/event_producer_last_update/test_event_producer_last_update.py +++ b/frappe/event_streaming/doctype/event_producer_last_update/test_event_producer_last_update.py @@ -1,8 +1,8 @@ # Copyright (c) 2020, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestEventProducerLastUpdate(unittest.TestCase): +class TestEventProducerLastUpdate(FrappeTestCase): pass diff --git a/frappe/event_streaming/doctype/event_sync_log/test_event_sync_log.py b/frappe/event_streaming/doctype/event_sync_log/test_event_sync_log.py index 13028cbac7..5efc030026 100644 --- a/frappe/event_streaming/doctype/event_sync_log/test_event_sync_log.py +++ b/frappe/event_streaming/doctype/event_sync_log/test_event_sync_log.py @@ -1,8 +1,8 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestEventSyncLog(unittest.TestCase): +class TestEventSyncLog(FrappeTestCase): pass diff --git a/frappe/event_streaming/doctype/event_update_log/test_event_update_log.py b/frappe/event_streaming/doctype/event_update_log/test_event_update_log.py index 0cbff47912..a9065ab4ed 100644 --- a/frappe/event_streaming/doctype/event_update_log/test_event_update_log.py +++ b/frappe/event_streaming/doctype/event_update_log/test_event_update_log.py @@ -1,8 +1,8 @@ # Copyright (c) 2019, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestEventUpdateLog(unittest.TestCase): +class TestEventUpdateLog(FrappeTestCase): pass diff --git a/frappe/integrations/doctype/connected_app/test_connected_app.py b/frappe/integrations/doctype/connected_app/test_connected_app.py index 1acedff160..88441db6b2 100644 --- a/frappe/integrations/doctype/connected_app/test_connected_app.py +++ b/frappe/integrations/doctype/connected_app/test_connected_app.py @@ -1,6 +1,5 @@ # Copyright (c) 2019, Frappe Technologies and contributors # License: MIT. See LICENSE -import unittest from urllib.parse import urljoin import requests @@ -9,6 +8,7 @@ import frappe from frappe.integrations.doctype.social_login_key.test_social_login_key import ( create_or_update_social_login_key, ) +from frappe.tests.utils import FrappeTestCase def get_user(usr, pwd): @@ -48,7 +48,7 @@ def get_oauth_client(): return oauth_client -class TestConnectedApp(unittest.TestCase): +class TestConnectedApp(FrappeTestCase): def setUp(self): """Set up a Connected App that connects to our own oAuth provider. diff --git a/frappe/integrations/doctype/dropbox_settings/test_dropbox_settings.py b/frappe/integrations/doctype/dropbox_settings/test_dropbox_settings.py index b165e03780..1c95b130e8 100644 --- a/frappe/integrations/doctype/dropbox_settings/test_dropbox_settings.py +++ b/frappe/integrations/doctype/dropbox_settings/test_dropbox_settings.py @@ -1,8 +1,8 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestDropboxSettings(unittest.TestCase): +class TestDropboxSettings(FrappeTestCase): pass diff --git a/frappe/integrations/doctype/google_drive/test_google_drive.py b/frappe/integrations/doctype/google_drive/test_google_drive.py index 4dcc79afd6..1dd048048c 100644 --- a/frappe/integrations/doctype/google_drive/test_google_drive.py +++ b/frappe/integrations/doctype/google_drive/test_google_drive.py @@ -1,8 +1,8 @@ # Copyright (c) 2019, Frappe Technologies and Contributors # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestGoogleDrive(unittest.TestCase): +class TestGoogleDrive(FrappeTestCase): pass diff --git a/frappe/integrations/doctype/google_settings/test_google_settings.py b/frappe/integrations/doctype/google_settings/test_google_settings.py index 8d07ffa54f..d67d7cf40b 100644 --- a/frappe/integrations/doctype/google_settings/test_google_settings.py +++ b/frappe/integrations/doctype/google_settings/test_google_settings.py @@ -1,14 +1,13 @@ # Copyright (c) 2021, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase from .google_settings import get_file_picker_settings -class TestGoogleSettings(unittest.TestCase): +class TestGoogleSettings(FrappeTestCase): def setUp(self): settings = frappe.get_single("Google Settings") settings.client_id = "test_client_id" diff --git a/frappe/integrations/doctype/integration_request/test_integration_request.py b/frappe/integrations/doctype/integration_request/test_integration_request.py index 45963d5096..d80d72fd7a 100644 --- a/frappe/integrations/doctype/integration_request/test_integration_request.py +++ b/frappe/integrations/doctype/integration_request/test_integration_request.py @@ -1,11 +1,10 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('Integration Request') -class TestIntegrationRequest(unittest.TestCase): +class TestIntegrationRequest(FrappeTestCase): pass diff --git a/frappe/integrations/doctype/oauth_authorization_code/test_oauth_authorization_code.py b/frappe/integrations/doctype/oauth_authorization_code/test_oauth_authorization_code.py index 2036a42f15..775bc0f968 100644 --- a/frappe/integrations/doctype/oauth_authorization_code/test_oauth_authorization_code.py +++ b/frappe/integrations/doctype/oauth_authorization_code/test_oauth_authorization_code.py @@ -1,11 +1,10 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('OAuth Authorization Code') -class TestOAuthAuthorizationCode(unittest.TestCase): +class TestOAuthAuthorizationCode(FrappeTestCase): pass diff --git a/frappe/integrations/doctype/oauth_bearer_token/test_oauth_bearer_token.py b/frappe/integrations/doctype/oauth_bearer_token/test_oauth_bearer_token.py index 3439096809..84b65893f6 100644 --- a/frappe/integrations/doctype/oauth_bearer_token/test_oauth_bearer_token.py +++ b/frappe/integrations/doctype/oauth_bearer_token/test_oauth_bearer_token.py @@ -1,11 +1,10 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('OAuth Bearer Token') -class TestOAuthBearerToken(unittest.TestCase): +class TestOAuthBearerToken(FrappeTestCase): pass diff --git a/frappe/integrations/doctype/oauth_client/test_oauth_client.py b/frappe/integrations/doctype/oauth_client/test_oauth_client.py index 8fd732673e..53e7bff40e 100644 --- a/frappe/integrations/doctype/oauth_client/test_oauth_client.py +++ b/frappe/integrations/doctype/oauth_client/test_oauth_client.py @@ -1,11 +1,10 @@ # Copyright (c) 2015, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase # test_records = frappe.get_test_records('OAuth Client') -class TestOAuthClient(unittest.TestCase): +class TestOAuthClient(FrappeTestCase): pass diff --git a/frappe/integrations/doctype/s3_backup_settings/test_s3_backup_settings.py b/frappe/integrations/doctype/s3_backup_settings/test_s3_backup_settings.py index 48b1ccd113..add4e7a1b2 100755 --- a/frappe/integrations/doctype/s3_backup_settings/test_s3_backup_settings.py +++ b/frappe/integrations/doctype/s3_backup_settings/test_s3_backup_settings.py @@ -1,7 +1,7 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestS3BackupSettings(unittest.TestCase): +class TestS3BackupSettings(FrappeTestCase): pass diff --git a/frappe/integrations/doctype/slack_webhook_url/test_slack_webhook_url.py b/frappe/integrations/doctype/slack_webhook_url/test_slack_webhook_url.py index 16b1bcd3c2..6697c2a320 100644 --- a/frappe/integrations/doctype/slack_webhook_url/test_slack_webhook_url.py +++ b/frappe/integrations/doctype/slack_webhook_url/test_slack_webhook_url.py @@ -1,7 +1,7 @@ # Copyright (c) 2018, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestSlackWebhookURL(unittest.TestCase): +class TestSlackWebhookURL(FrappeTestCase): pass diff --git a/frappe/integrations/doctype/social_login_key/test_social_login_key.py b/frappe/integrations/doctype/social_login_key/test_social_login_key.py index c51ccb2c0f..c8eedf23da 100644 --- a/frappe/integrations/doctype/social_login_key/test_social_login_key.py +++ b/frappe/integrations/doctype/social_login_key/test_social_login_key.py @@ -1,6 +1,5 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest from unittest.mock import MagicMock, patch from rauth import OAuth2Service @@ -8,11 +7,12 @@ from rauth import OAuth2Service import frappe from frappe.auth import CookieManager, LoginManager from frappe.integrations.doctype.social_login_key.social_login_key import BaseUrlNotSetError +from frappe.tests.utils import FrappeTestCase from frappe.utils import set_request from frappe.utils.oauth import login_via_oauth2 -class TestSocialLoginKey(unittest.TestCase): +class TestSocialLoginKey(FrappeTestCase): def test_adding_frappe_social_login_provider(self): provider_name = "Frappe" social_login_key = make_social_login_key(social_login_provider=provider_name) diff --git a/frappe/integrations/doctype/token_cache/test_token_cache.py b/frappe/integrations/doctype/token_cache/test_token_cache.py index a9366d84d3..44b58fb238 100644 --- a/frappe/integrations/doctype/token_cache/test_token_cache.py +++ b/frappe/integrations/doctype/token_cache/test_token_cache.py @@ -1,13 +1,12 @@ # Copyright (c) 2019, Frappe Technologies and contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase test_dependencies = ["User", "Connected App", "Token Cache"] -class TestTokenCache(unittest.TestCase): +class TestTokenCache(FrappeTestCase): def setUp(self): self.token_cache = frappe.get_last_doc("Token Cache") self.token_cache.update({"connected_app": frappe.get_last_doc("Connected App").name}) diff --git a/frappe/integrations/doctype/webhook/test_webhook.py b/frappe/integrations/doctype/webhook/test_webhook.py index 7d9d05cd9e..9c008b2e47 100644 --- a/frappe/integrations/doctype/webhook/test_webhook.py +++ b/frappe/integrations/doctype/webhook/test_webhook.py @@ -1,7 +1,6 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE import json -import unittest from contextlib import contextmanager import frappe @@ -10,6 +9,7 @@ from frappe.integrations.doctype.webhook.webhook import ( get_webhook_data, get_webhook_headers, ) +from frappe.tests.utils import FrappeTestCase @contextmanager @@ -22,13 +22,14 @@ def get_test_webhook(config): wh.delete() -class TestWebhook(unittest.TestCase): +class TestWebhook(FrappeTestCase): @classmethod def setUpClass(cls): # delete any existing webhooks frappe.db.delete("Webhook") # Delete existing logs if any frappe.db.delete("Webhook Request Log") + super().setUpClass() # create test webhooks cls.create_sample_webhooks() diff --git a/frappe/integrations/doctype/webhook_request_log/test_webhook_request_log.py b/frappe/integrations/doctype/webhook_request_log/test_webhook_request_log.py index e3ad5a88a4..fe15bb8b58 100644 --- a/frappe/integrations/doctype/webhook_request_log/test_webhook_request_log.py +++ b/frappe/integrations/doctype/webhook_request_log/test_webhook_request_log.py @@ -2,8 +2,8 @@ # License: MIT. See LICENSE # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestWebhookRequestLog(unittest.TestCase): +class TestWebhookRequestLog(FrappeTestCase): pass diff --git a/frappe/printing/doctype/letter_head/test_letter_head.py b/frappe/printing/doctype/letter_head/test_letter_head.py index 75019ce275..b36daeae03 100644 --- a/frappe/printing/doctype/letter_head/test_letter_head.py +++ b/frappe/printing/doctype/letter_head/test_letter_head.py @@ -1,11 +1,10 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestLetterHead(unittest.TestCase): +class TestLetterHead(FrappeTestCase): def test_auto_image(self): letter_head = frappe.get_doc( dict(doctype="Letter Head", letter_head_name="Test", source="Image", image="/public/test.png") diff --git a/frappe/printing/doctype/network_printer_settings/test_network_printer_settings.py b/frappe/printing/doctype/network_printer_settings/test_network_printer_settings.py index 17ce16cb82..de04b5277b 100644 --- a/frappe/printing/doctype/network_printer_settings/test_network_printer_settings.py +++ b/frappe/printing/doctype/network_printer_settings/test_network_printer_settings.py @@ -2,8 +2,8 @@ # See license.txt # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestNetworkPrinterSettings(unittest.TestCase): +class TestNetworkPrinterSettings(FrappeTestCase): pass diff --git a/frappe/printing/doctype/print_format/test_print_format.py b/frappe/printing/doctype/print_format/test_print_format.py index b4d0c87da5..7caa5c6102 100644 --- a/frappe/printing/doctype/print_format/test_print_format.py +++ b/frappe/printing/doctype/print_format/test_print_format.py @@ -2,10 +2,10 @@ # License: MIT. See LICENSE import os import re -import unittest from typing import TYPE_CHECKING import frappe +from frappe.tests.utils import FrappeTestCase if TYPE_CHECKING: from frappe.printing.doctype.print_format.print_format import PrintFormat @@ -13,7 +13,7 @@ if TYPE_CHECKING: test_records = frappe.get_test_records("Print Format") -class TestPrintFormat(unittest.TestCase): +class TestPrintFormat(FrappeTestCase): def test_print_user(self, style=None): print_html = frappe.get_print("User", "Administrator", style=style) self.assertTrue("" in print_html) diff --git a/frappe/printing/doctype/print_format_field_template/test_print_format_field_template.py b/frappe/printing/doctype/print_format_field_template/test_print_format_field_template.py index b55a492635..74f495bd3a 100644 --- a/frappe/printing/doctype/print_format_field_template/test_print_format_field_template.py +++ b/frappe/printing/doctype/print_format_field_template/test_print_format_field_template.py @@ -2,8 +2,8 @@ # See license.txt # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestPrintFormatFieldTemplate(unittest.TestCase): +class TestPrintFormatFieldTemplate(FrappeTestCase): pass diff --git a/frappe/printing/doctype/print_heading/test_print_heading.py b/frappe/printing/doctype/print_heading/test_print_heading.py index 74ff7ce74f..8fab30e79f 100644 --- a/frappe/printing/doctype/print_heading/test_print_heading.py +++ b/frappe/printing/doctype/print_heading/test_print_heading.py @@ -1,9 +1,8 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestPrintHeading(unittest.TestCase): +class TestPrintHeading(FrappeTestCase): pass diff --git a/frappe/printing/doctype/print_settings/test_print_settings.py b/frappe/printing/doctype/print_settings/test_print_settings.py index 6a6437bf97..80069824ab 100644 --- a/frappe/printing/doctype/print_settings/test_print_settings.py +++ b/frappe/printing/doctype/print_settings/test_print_settings.py @@ -1,7 +1,7 @@ # Copyright (c) 2018, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestPrintSettings(unittest.TestCase): +class TestPrintSettings(FrappeTestCase): pass diff --git a/frappe/printing/doctype/print_style/test_print_style.py b/frappe/printing/doctype/print_style/test_print_style.py index f8ce54b9bb..744181ad3e 100644 --- a/frappe/printing/doctype/print_style/test_print_style.py +++ b/frappe/printing/doctype/print_style/test_print_style.py @@ -1,9 +1,8 @@ # Copyright (c) 2017, Frappe Technologies and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestPrintStyle(unittest.TestCase): +class TestPrintStyle(FrappeTestCase): pass diff --git a/frappe/search/test_full_text_search.py b/frappe/search/test_full_text_search.py index 4284b64d9c..0a11dd6953 100644 --- a/frappe/search/test_full_text_search.py +++ b/frappe/search/test_full_text_search.py @@ -1,11 +1,10 @@ # Copyright (c) 2020, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - from frappe.search.full_text_search import FullTextSearch +from frappe.tests.utils import FrappeTestCase -class TestFullTextSearch(unittest.TestCase): +class TestFullTextSearch(FrappeTestCase): def setUp(self): index = get_index() index.build() diff --git a/frappe/social/doctype/energy_point_log/test_energy_point_log.py b/frappe/social/doctype/energy_point_log/test_energy_point_log.py index 3830f0505b..5043019c19 100644 --- a/frappe/social/doctype/energy_point_log/test_energy_point_log.py +++ b/frappe/social/doctype/energy_point_log/test_energy_point_log.py @@ -13,6 +13,7 @@ from .energy_point_log import review class TestEnergyPointLog(FrappeTestCase): @classmethod def setUpClass(cls): + super().setUpClass() settings = frappe.get_single("Energy Point Settings") settings.enabled = 1 settings.save() diff --git a/frappe/social/doctype/energy_point_settings/test_energy_point_settings.py b/frappe/social/doctype/energy_point_settings/test_energy_point_settings.py index 10416a127b..7907eecfe9 100644 --- a/frappe/social/doctype/energy_point_settings/test_energy_point_settings.py +++ b/frappe/social/doctype/energy_point_settings/test_energy_point_settings.py @@ -2,8 +2,8 @@ # See license.txt # import frappe -import unittest +from frappe.tests.utils import FrappeTestCase -class TestEnergyPointSettings(unittest.TestCase): +class TestEnergyPointSettings(FrappeTestCase): pass diff --git a/frappe/tests/test_api.py b/frappe/tests/test_api.py index 9ee17e91f3..1f7377ba01 100644 --- a/frappe/tests/test_api.py +++ b/frappe/tests/test_api.py @@ -1,5 +1,4 @@ import sys -import unittest from contextlib import contextmanager from random import choice from threading import Thread @@ -10,6 +9,7 @@ from semantic_version import Version from werkzeug.test import TestResponse import frappe +from frappe.tests.utils import FrappeTestCase from frappe.utils import get_site_url, get_test_client try: @@ -69,7 +69,7 @@ class ThreadWithReturnValue(Thread): return self._return -class FrappeAPITestCase(unittest.TestCase): +class FrappeAPITestCase(FrappeTestCase): SITE = frappe.local.site SITE_URL = get_site_url(SITE) RESOURCE_URL = f"{SITE_URL}/api/resource" @@ -108,6 +108,7 @@ class TestResourceAPI(FrappeAPITestCase): @classmethod def setUpClass(cls): + super().setUpClass() for _ in range(10): doc = frappe.get_doc({"doctype": "ToDo", "description": frappe.mock("paragraph")}).insert() cls.GENERATED_DOCUMENTS.append(doc.name) diff --git a/frappe/tests/test_assign.py b/frappe/tests/test_assign.py index aa646d2827..f3b579c028 100644 --- a/frappe/tests/test_assign.py +++ b/frappe/tests/test_assign.py @@ -1,15 +1,14 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe import frappe.desk.form.assign_to from frappe.automation.doctype.assignment_rule.test_assignment_rule import make_note from frappe.desk.form.load import get_assignments from frappe.desk.listview import get_group_by_count +from frappe.tests.utils import FrappeTestCase -class TestAssign(unittest.TestCase): +class TestAssign(FrappeTestCase): def test_assign(self): todo = frappe.get_doc({"doctype": "ToDo", "description": "test"}).insert() if not frappe.db.exists("User", "test@example.com"): diff --git a/frappe/tests/test_auth.py b/frappe/tests/test_auth.py index b9c492715c..4378d75484 100644 --- a/frappe/tests/test_auth.py +++ b/frappe/tests/test_auth.py @@ -1,12 +1,12 @@ # Copyright (c) 2021, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE import time -import unittest import frappe import frappe.utils from frappe.auth import LoginAttemptTracker from frappe.frappeclient import AuthError, FrappeClient +from frappe.tests.utils import FrappeTestCase def add_user(email, password, username=None, mobile_no=None): @@ -19,9 +19,10 @@ def add_user(email, password, username=None, mobile_no=None): frappe.db.commit() -class TestAuth(unittest.TestCase): +class TestAuth(FrappeTestCase): @classmethod def setUpClass(cls): + super().setUpClass() cls.HOST_NAME = frappe.get_site_config().host_name or frappe.utils.get_site_url( frappe.local.site ) @@ -114,7 +115,7 @@ class TestAuth(unittest.TestCase): third_login.get_list("ToDo") -class TestLoginAttemptTracker(unittest.TestCase): +class TestLoginAttemptTracker(FrappeTestCase): def test_account_lock(self): """Make sure that account locks after `n consecutive failures""" tracker = LoginAttemptTracker( diff --git a/frappe/tests/test_background_jobs.py b/frappe/tests/test_background_jobs.py index 6c7dda51f1..d05d01bf6b 100644 --- a/frappe/tests/test_background_jobs.py +++ b/frappe/tests/test_background_jobs.py @@ -1,14 +1,14 @@ import time -import unittest from rq import Queue import frappe from frappe.core.page.background_jobs.background_jobs import remove_failed_jobs +from frappe.tests.utils import FrappeTestCase from frappe.utils.background_jobs import generate_qname, get_redis_conn -class TestBackgroundJobs(unittest.TestCase): +class TestBackgroundJobs(FrappeTestCase): def test_remove_failed_jobs(self): frappe.enqueue(method="frappe.tests.test_background_jobs.fail_function", queue="short") # wait for enqueued job to execute diff --git a/frappe/tests/test_base_document.py b/frappe/tests/test_base_document.py index a861b3454b..0f49e7d7f5 100644 --- a/frappe/tests/test_base_document.py +++ b/frappe/tests/test_base_document.py @@ -1,9 +1,8 @@ -import unittest - from frappe.model.base_document import BaseDocument +from frappe.tests.utils import FrappeTestCase -class TestBaseDocument(unittest.TestCase): +class TestBaseDocument(FrappeTestCase): def test_docstatus(self): doc = BaseDocument({"docstatus": 0, "doctype": "ToDo"}) self.assertTrue(doc.docstatus.is_draft()) diff --git a/frappe/tests/test_boilerplate.py b/frappe/tests/test_boilerplate.py index 092a1ff126..8dd29b24af 100644 --- a/frappe/tests/test_boilerplate.py +++ b/frappe/tests/test_boilerplate.py @@ -3,13 +3,13 @@ import copy import glob import os import shutil -import unittest from io import StringIO from unittest.mock import patch import yaml import frappe +from frappe.tests.utils import FrappeTestCase from frappe.utils.boilerplate import ( _create_app_boilerplate, _get_user_inputs, @@ -17,9 +17,10 @@ from frappe.utils.boilerplate import ( ) -class TestBoilerPlate(unittest.TestCase): +class TestBoilerPlate(FrappeTestCase): @classmethod def setUpClass(cls): + super().setUpClass() cls.default_hooks = frappe._dict( { "app_name": "test_app", diff --git a/frappe/tests/test_boot.py b/frappe/tests/test_boot.py index 11cbeb7621..0b688d6aee 100644 --- a/frappe/tests/test_boot.py +++ b/frappe/tests/test_boot.py @@ -1,5 +1,3 @@ -import unittest - import frappe from frappe.boot import get_unseen_notes from frappe.desk.doctype.note.note import mark_as_seen diff --git a/frappe/tests/test_bot.py b/frappe/tests/test_bot.py index ffc3513619..aead54dd63 100644 --- a/frappe/tests/test_bot.py +++ b/frappe/tests/test_bot.py @@ -1,8 +1,8 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestBot(unittest.TestCase): +class TestBot(FrappeTestCase): pass diff --git a/frappe/tests/test_caching.py b/frappe/tests/test_caching.py index 66b47b6e31..d1de587d0d 100644 --- a/frappe/tests/test_caching.py +++ b/frappe/tests/test_caching.py @@ -1,9 +1,9 @@ import time -import unittest from unittest.mock import MagicMock import frappe from frappe.tests.test_api import FrappeAPITestCase +from frappe.tests.utils import FrappeTestCase from frappe.utils.caching import request_cache, site_cache CACHE_TTL = 4 @@ -34,7 +34,7 @@ def ping_with_ttl() -> str: return frappe.local.site -class TestCachingUtils(unittest.TestCase): +class TestCachingUtils(FrappeTestCase): def test_request_cache(self): retval = [] acceptable_args = [ diff --git a/frappe/tests/test_child_table.py b/frappe/tests/test_child_table.py index eb632eac90..5dce5f54c8 100644 --- a/frappe/tests/test_child_table.py +++ b/frappe/tests/test_child_table.py @@ -1,11 +1,11 @@ -import unittest from typing import Callable import frappe from frappe.model import child_table_fields +from frappe.tests.utils import FrappeTestCase -class TestChildTable(unittest.TestCase): +class TestChildTable(FrappeTestCase): def tearDown(self) -> None: try: frappe.delete_doc("DocType", self.doctype_name, force=1) diff --git a/frappe/tests/test_client.py b/frappe/tests/test_client.py index 3587d5eb63..93a2ce93a1 100644 --- a/frappe/tests/test_client.py +++ b/frappe/tests/test_client.py @@ -1,11 +1,10 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestClient(unittest.TestCase): +class TestClient(FrappeTestCase): def test_set_value(self): todo = frappe.get_doc(dict(doctype="ToDo", description="test")).insert() frappe.set_value("ToDo", todo.name, "description", "test 1") diff --git a/frappe/tests/test_commands.py b/frappe/tests/test_commands.py index 86d9075561..f97b2ee754 100644 --- a/frappe/tests/test_commands.py +++ b/frappe/tests/test_commands.py @@ -29,6 +29,7 @@ import frappe.recorder from frappe.installer import add_to_installed_apps, remove_app from frappe.query_builder.utils import db_type_is from frappe.tests.test_query_builder import run_only_if +from frappe.tests.utils import FrappeTestCase from frappe.utils import add_to_date, get_bench_path, get_bench_relative_path, now from frappe.utils.backups import fetch_latest_backups from frappe.utils.jinja_globals import bundled_asset @@ -134,11 +135,11 @@ def cli(cmd: Command, args: list | None = None): importlib.invalidate_caches() -class BaseTestCommands(unittest.TestCase): +class BaseTestCommands(FrappeTestCase): @classmethod def setUpClass(cls) -> None: + super().setUpClass() cls.setup_test_site() - return super().setUpClass() @classmethod def execute(self, command, kwargs=None): @@ -636,7 +637,7 @@ class TestBackups(BaseTestCommands): self.assertEqual([], missing_in_backup(self.backup_map["excludes"]["excludes"], database)) -class TestRemoveApp(unittest.TestCase): +class TestRemoveApp(FrappeTestCase): def test_delete_modules(self): from frappe.installer import ( _delete_doctypes, @@ -715,7 +716,7 @@ class TestBenchBuild(BaseTestCommands): ) -class TestCommandUtils(unittest.TestCase): +class TestCommandUtils(FrappeTestCase): def test_bench_helper(self): from frappe.utils.bench_helper import get_app_groups diff --git a/frappe/tests/test_config.py b/frappe/tests/test_config.py index 21ceb47bb3..d2b1196f52 100644 --- a/frappe/tests/test_config.py +++ b/frappe/tests/test_config.py @@ -1,12 +1,11 @@ # Copyright (c) 2022, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.config import get_modules_from_all_apps_for_user +from frappe.tests.utils import FrappeTestCase -class TestConfig(unittest.TestCase): +class TestConfig(FrappeTestCase): def test_get_modules(self): frappe_modules = frappe.get_all("Module Def", filters={"app_name": "frappe"}, pluck="name") all_modules_data = get_modules_from_all_apps_for_user() diff --git a/frappe/tests/test_cors.py b/frappe/tests/test_cors.py index 723cb7e804..0712c8d24a 100644 --- a/frappe/tests/test_cors.py +++ b/frappe/tests/test_cors.py @@ -1,11 +1,10 @@ # Copyright (c) 2020, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - from werkzeug.wrappers import Response import frappe from frappe.app import process_response +from frappe.tests.utils import FrappeTestCase HEADERS = ( "Access-Control-Allow-Origin", @@ -15,7 +14,7 @@ HEADERS = ( ) -class TestCORS(unittest.TestCase): +class TestCORS(FrappeTestCase): def make_request_and_test(self, origin="http://example.com", absent=False): self.origin = origin diff --git a/frappe/tests/test_db.py b/frappe/tests/test_db.py index bb6db76bf5..d2aa7df043 100644 --- a/frappe/tests/test_db.py +++ b/frappe/tests/test_db.py @@ -3,7 +3,6 @@ import datetime import inspect -import unittest from math import ceil from random import choice from unittest.mock import patch @@ -17,11 +16,12 @@ from frappe.database.utils import FallBackDateTimeStr from frappe.query_builder import Field from frappe.query_builder.functions import Concat_ws from frappe.tests.test_query_builder import db_type_is, run_only_if +from frappe.tests.utils import FrappeTestCase from frappe.utils import add_days, cint, now, random_string from frappe.utils.testutils import clear_custom_fields -class TestDB(unittest.TestCase): +class TestDB(FrappeTestCase): def test_datetime_format(self): now_str = now() self.assertEqual(frappe.db.format_datetime(None), FallBackDateTimeStr) @@ -559,7 +559,7 @@ class TestDB(unittest.TestCase): @run_only_if(db_type_is.MARIADB) -class TestDDLCommandsMaria(unittest.TestCase): +class TestDDLCommandsMaria(FrappeTestCase): test_table_name = "TestNotes" def setUp(self) -> None: @@ -621,9 +621,10 @@ class TestDDLCommandsMaria(unittest.TestCase): self.assertEqual(len(indexs_in_table), 2) -class TestDBSetValue(unittest.TestCase): +class TestDBSetValue(FrappeTestCase): @classmethod def setUpClass(cls): + super().setUpClass() cls.todo1 = frappe.get_doc(doctype="ToDo", description="test_set_value 1").insert() cls.todo2 = frappe.get_doc(doctype="ToDo", description="test_set_value 2").insert() @@ -782,7 +783,7 @@ class TestDBSetValue(unittest.TestCase): @run_only_if(db_type_is.POSTGRES) -class TestDDLCommandsPost(unittest.TestCase): +class TestDDLCommandsPost(FrappeTestCase): test_table_name = "TestNotes" def setUp(self) -> None: @@ -891,7 +892,7 @@ class TestDDLCommandsPost(unittest.TestCase): @run_only_if(db_type_is.POSTGRES) -class TestTransactionManagement(unittest.TestCase): +class TestTransactionManagement(FrappeTestCase): def test_create_proper_transactions(self): def _get_transaction_id(): return frappe.db.sql("select txid_current()", pluck=True) diff --git a/frappe/tests/test_db_query.py b/frappe/tests/test_db_query.py index 0eb54c7ab0..32f1492774 100644 --- a/frappe/tests/test_db_query.py +++ b/frappe/tests/test_db_query.py @@ -1,7 +1,6 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE import datetime -import unittest import frappe from frappe.core.page.permission_manager.permission_manager import add, reset, update @@ -11,12 +10,13 @@ from frappe.handler import execute_cmd from frappe.model.db_query import DatabaseQuery from frappe.permissions import add_user_permission, clear_user_permissions_for_doctype from frappe.query_builder import Column +from frappe.tests.utils import FrappeTestCase from frappe.utils.testutils import add_custom_field, clear_custom_fields test_dependencies = ["User", "Blog Post", "Blog Category", "Blogger"] -class TestReportview(unittest.TestCase): +class TestReportview(FrappeTestCase): def setUp(self): frappe.set_user("Administrator") diff --git a/frappe/tests/test_db_update.py b/frappe/tests/test_db_update.py index 38cc8af2f3..dde6639526 100644 --- a/frappe/tests/test_db_update.py +++ b/frappe/tests/test_db_update.py @@ -1,12 +1,11 @@ -import unittest - import frappe from frappe.core.utils import find from frappe.custom.doctype.property_setter.property_setter import make_property_setter +from frappe.tests.utils import FrappeTestCase from frappe.utils import cstr -class TestDBUpdate(unittest.TestCase): +class TestDBUpdate(FrappeTestCase): def test_db_update(self): doctype = "User" frappe.reload_doctype("User", force=True) diff --git a/frappe/tests/test_defaults.py b/frappe/tests/test_defaults.py index 3841bde34e..3c04f16ec8 100644 --- a/frappe/tests/test_defaults.py +++ b/frappe/tests/test_defaults.py @@ -1,12 +1,11 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.defaults import * +from frappe.tests.utils import FrappeTestCase -class TestDefaults(unittest.TestCase): +class TestDefaults(FrappeTestCase): def test_global(self): clear_user_default("key1") set_global_default("key1", "value1") diff --git a/frappe/tests/test_docstatus.py b/frappe/tests/test_docstatus.py index 7692bca48b..ae29681206 100644 --- a/frappe/tests/test_docstatus.py +++ b/frappe/tests/test_docstatus.py @@ -1,9 +1,8 @@ -import unittest - from frappe.model.docstatus import DocStatus +from frappe.tests.utils import FrappeTestCase -class TestDocStatus(unittest.TestCase): +class TestDocStatus(FrappeTestCase): def test_draft(self): self.assertEqual(DocStatus(0), DocStatus.draft()) diff --git a/frappe/tests/test_document.py b/frappe/tests/test_document.py index dcd3e87cf8..e05c4dd78c 100644 --- a/frappe/tests/test_document.py +++ b/frappe/tests/test_document.py @@ -1,6 +1,5 @@ # Copyright (c) 2022, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest from contextlib import contextmanager from datetime import timedelta from unittest.mock import Mock, patch @@ -9,6 +8,7 @@ import frappe from frappe.app import make_form_dict from frappe.desk.doctype.note.note import Note from frappe.model.naming import make_autoname, parse_naming_series, revert_series_if_last +from frappe.tests.utils import FrappeTestCase from frappe.utils import cint, now_datetime, set_request from frappe.website.serve import get_response @@ -21,7 +21,7 @@ class CustomTestNote(Note): return now_datetime() - self.creation -class TestDocument(unittest.TestCase): +class TestDocument(FrappeTestCase): def test_get_return_empty_list_for_table_field_if_none(self): d = frappe.get_doc({"doctype": "User"}) self.assertEqual(d.get("roles"), []) @@ -392,7 +392,7 @@ class TestDocument(unittest.TestCase): self.assertEqual(todo.notify_update.call_count, 1) -class TestDocumentWebView(unittest.TestCase): +class TestDocumentWebView(FrappeTestCase): def get(self, path, user="Guest"): frappe.set_user(user) set_request(method="GET", path=path) diff --git a/frappe/tests/test_document_locks.py b/frappe/tests/test_document_locks.py index 36e07ca2ca..a92c9ffc54 100644 --- a/frappe/tests/test_document_locks.py +++ b/frappe/tests/test_document_locks.py @@ -1,11 +1,10 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestDocumentLocks(unittest.TestCase): +class TestDocumentLocks(FrappeTestCase): def test_locking(self): todo = frappe.get_doc(dict(doctype="ToDo", description="test")).insert() todo_1 = frappe.get_doc("ToDo", todo.name) diff --git a/frappe/tests/test_domainification.py b/frappe/tests/test_domainification.py index 575445e34c..577110e938 100644 --- a/frappe/tests/test_domainification.py +++ b/frappe/tests/test_domainification.py @@ -1,7 +1,5 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.core.doctype.domain_settings.domain_settings import get_active_modules from frappe.core.page.permission_manager.permission_manager import get_roles_and_doctypes @@ -10,9 +8,10 @@ from frappe.desk.doctype.desktop_icon.desktop_icon import ( clear_desktop_icons_cache, get_desktop_icons, ) +from frappe.tests.utils import FrappeTestCase -class TestDomainification(unittest.TestCase): +class TestDomainification(FrappeTestCase): def setUp(self): # create test domain self.new_domain("_Test Domain 1") diff --git a/frappe/tests/test_dynamic_links.py b/frappe/tests/test_dynamic_links.py index d5a2a52288..95f613fc70 100644 --- a/frappe/tests/test_dynamic_links.py +++ b/frappe/tests/test_dynamic_links.py @@ -1,11 +1,10 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase -class TestDynamicLinks(unittest.TestCase): +class TestDynamicLinks(FrappeTestCase): def setUp(self): frappe.db.delete("Email Unsubscribe") diff --git a/frappe/tests/test_email.py b/frappe/tests/test_email.py index 71886bb625..13e95c38e6 100644 --- a/frappe/tests/test_email.py +++ b/frappe/tests/test_email.py @@ -3,15 +3,15 @@ import email import re -import unittest import frappe from frappe.email.doctype.email_account.test_email_account import TestEmailAccount +from frappe.tests.utils import FrappeTestCase test_dependencies = ["Email Account"] -class TestEmail(unittest.TestCase): +class TestEmail(FrappeTestCase): def setUp(self): frappe.db.delete("Email Unsubscribe") frappe.db.delete("Email Queue") @@ -311,5 +311,7 @@ class TestEmail(unittest.TestCase): if __name__ == "__main__": + import unittest + frappe.connect() unittest.main() diff --git a/frappe/tests/test_exporter_fixtures.py b/frappe/tests/test_exporter_fixtures.py index 0f82954dc6..8b8ca1b8d8 100644 --- a/frappe/tests/test_exporter_fixtures.py +++ b/frappe/tests/test_exporter_fixtures.py @@ -1,14 +1,14 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE import os -import unittest import frappe import frappe.defaults from frappe.core.doctype.data_import.data_import import export_csv +from frappe.tests.utils import FrappeTestCase -class TestDataImportFixtures(unittest.TestCase): +class TestDataImportFixtures(FrappeTestCase): def setUp(self): pass diff --git a/frappe/tests/test_fixture_import.py b/frappe/tests/test_fixture_import.py index 3113812fa6..b9bd4550b2 100644 --- a/frappe/tests/test_fixture_import.py +++ b/frappe/tests/test_fixture_import.py @@ -1,13 +1,13 @@ import os -import unittest import frappe from frappe.core.doctype.data_import.data_import import export_json, import_doc from frappe.desk.form.save import savedocs from frappe.model.delete_doc import delete_doc +from frappe.tests.utils import FrappeTestCase -class TestFixtureImport(unittest.TestCase): +class TestFixtureImport(FrappeTestCase): def create_new_doctype(self, DocType: str) -> None: file = frappe.get_app_path("frappe", "custom", "fixtures", f"{DocType}.json") diff --git a/frappe/tests/test_fmt_datetime.py b/frappe/tests/test_fmt_datetime.py index 706aca6d7c..43cf701c33 100644 --- a/frappe/tests/test_fmt_datetime.py +++ b/frappe/tests/test_fmt_datetime.py @@ -1,9 +1,9 @@ # Copyright (c) 2019, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE import datetime -import unittest import frappe +from frappe.tests.utils import FrappeTestCase from frappe.utils import ( format_datetime, format_time, @@ -33,7 +33,7 @@ test_time_formats = { } -class TestFmtDatetime(unittest.TestCase): +class TestFmtDatetime(FrappeTestCase): """Tests date, time and datetime formatters and some associated utility functions. These rely on the system-wide date and time formats. diff --git a/frappe/tests/test_fmt_money.py b/frappe/tests/test_fmt_money.py index ee58cd82c3..fff5011189 100644 --- a/frappe/tests/test_fmt_money.py +++ b/frappe/tests/test_fmt_money.py @@ -1,12 +1,11 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe +from frappe.tests.utils import FrappeTestCase from frappe.utils import fmt_money -class TestFmtMoney(unittest.TestCase): +class TestFmtMoney(FrappeTestCase): def test_standard(self): frappe.db.set_default("number_format", "#,###.##") self.assertEqual(fmt_money(100), "100.00") @@ -104,5 +103,7 @@ class TestFmtMoney(unittest.TestCase): if __name__ == "__main__": + import unittest + frappe.connect() unittest.main() diff --git a/frappe/tests/test_form_load.py b/frappe/tests/test_form_load.py index 83baf94b1c..283f23080e 100644 --- a/frappe/tests/test_form_load.py +++ b/frappe/tests/test_form_load.py @@ -1,17 +1,16 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.core.page.permission_manager.permission_manager import add, reset, update from frappe.custom.doctype.property_setter.property_setter import make_property_setter from frappe.desk.form.load import get_docinfo, getdoc, getdoctype +from frappe.tests.utils import FrappeTestCase from frappe.utils.file_manager import save_file test_dependencies = ["Blog Category", "Blogger"] -class TestFormLoad(unittest.TestCase): +class TestFormLoad(FrappeTestCase): def test_load(self): getdoctype("DocType") meta = list(filter(lambda d: d.name == "DocType", frappe.response.docs))[0] diff --git a/frappe/tests/test_formatter.py b/frappe/tests/test_formatter.py index b34aa40e77..c7dca303ec 100644 --- a/frappe/tests/test_formatter.py +++ b/frappe/tests/test_formatter.py @@ -1,10 +1,9 @@ -import unittest - import frappe from frappe import format +from frappe.tests.utils import FrappeTestCase -class TestFormatter(unittest.TestCase): +class TestFormatter(FrappeTestCase): def test_currency_formatting(self): df = frappe._dict({"fieldname": "amount", "fieldtype": "Currency", "options": "currency"}) diff --git a/frappe/tests/test_geo_ip.py b/frappe/tests/test_geo_ip.py index 5e98339ffa..101512a5f1 100644 --- a/frappe/tests/test_geo_ip.py +++ b/frappe/tests/test_geo_ip.py @@ -1,9 +1,9 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest +from frappe.tests.utils import FrappeTestCase -class TestGeoIP(unittest.TestCase): +class TestGeoIP(FrappeTestCase): def test_geo_ip(self): return from frappe.sessions import get_geo_ip_country diff --git a/frappe/tests/test_global_search.py b/frappe/tests/test_global_search.py index 1b9662219b..cadbb2b35d 100644 --- a/frappe/tests/test_global_search.py +++ b/frappe/tests/test_global_search.py @@ -1,16 +1,15 @@ # Copyright (c) 2022, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.custom.doctype.property_setter.property_setter import make_property_setter from frappe.desk.page.setup_wizard.install_fixtures import update_global_search_doctypes from frappe.test_runner import make_test_objects +from frappe.tests.utils import FrappeTestCase from frappe.utils import global_search, now_datetime -class TestGlobalSearch(unittest.TestCase): +class TestGlobalSearch(FrappeTestCase): def setUp(self): update_global_search_doctypes() global_search.setup_global_search_table() diff --git a/frappe/tests/test_hooks.py b/frappe/tests/test_hooks.py index 1311d59341..bfe299ab73 100644 --- a/frappe/tests/test_hooks.py +++ b/frappe/tests/test_hooks.py @@ -1,14 +1,13 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe from frappe.cache_manager import clear_controller_cache from frappe.desk.doctype.todo.todo import ToDo +from frappe.tests.utils import FrappeTestCase -class TestHooks(unittest.TestCase): +class TestHooks(FrappeTestCase): def test_hooks(self): hooks = frappe.get_hooks() self.assertTrue(isinstance(hooks.get("app_name"), list)) diff --git a/frappe/tests/test_linked_with.py b/frappe/tests/test_linked_with.py index d1be3cb775..065d6f4b04 100644 --- a/frappe/tests/test_linked_with.py +++ b/frappe/tests/test_linked_with.py @@ -1,11 +1,10 @@ -import unittest - import frappe from frappe.core.doctype.doctype.test_doctype import new_doctype from frappe.desk.form import linked_with +from frappe.tests.utils import FrappeTestCase -class TestLinkedWith(unittest.TestCase): +class TestLinkedWith(FrappeTestCase): def setUp(self): parent_doc = new_doctype("Parent Doc") parent_doc.is_submittable = 1 diff --git a/frappe/tests/test_listview.py b/frappe/tests/test_listview.py index 7d75e80f95..b6d6fb5223 100644 --- a/frappe/tests/test_listview.py +++ b/frappe/tests/test_listview.py @@ -1,13 +1,13 @@ # Copyright (c) 2019, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE import json -import unittest import frappe from frappe.desk.listview import get_group_by_count, get_list_settings, set_list_settings +from frappe.tests.utils import FrappeTestCase -class TestListView(unittest.TestCase): +class TestListView(FrappeTestCase): def setUp(self): if frappe.db.exists("List View Settings", "DocType"): frappe.delete_doc("List View Settings", "DocType") diff --git a/frappe/tests/test_model_utils.py b/frappe/tests/test_model_utils.py index 95e3e714a7..0baef8bb85 100644 --- a/frappe/tests/test_model_utils.py +++ b/frappe/tests/test_model_utils.py @@ -1,10 +1,9 @@ -import unittest - import frappe from frappe.model.utils import get_fetch_values +from frappe.tests.utils import FrappeTestCase -class TestModelUtils(unittest.TestCase): +class TestModelUtils(FrappeTestCase): def test_get_fetch_values(self): doctype = "ToDo" diff --git a/frappe/tests/test_monitor.py b/frappe/tests/test_monitor.py index 48de913241..7536c6a75a 100644 --- a/frappe/tests/test_monitor.py +++ b/frappe/tests/test_monitor.py @@ -1,16 +1,15 @@ # Copyright (c) 2020, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - import frappe import frappe.monitor from frappe.monitor import MONITOR_REDIS_KEY +from frappe.tests.utils import FrappeTestCase from frappe.utils import set_request from frappe.utils.response import build_response -class TestMonitor(unittest.TestCase): +class TestMonitor(FrappeTestCase): def setUp(self): frappe.conf.monitor = 1 frappe.cache().delete_value(MONITOR_REDIS_KEY) diff --git a/frappe/tests/test_oauth20.py b/frappe/tests/test_oauth20.py index c615138810..9790637004 100644 --- a/frappe/tests/test_oauth20.py +++ b/frappe/tests/test_oauth20.py @@ -1,7 +1,6 @@ # Copyright (c) 2022, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest from typing import TYPE_CHECKING from urllib.parse import parse_qs, urljoin, urlparse @@ -13,12 +12,13 @@ import frappe from frappe.integrations.oauth2 import encode_params from frappe.test_runner import make_test_records from frappe.tests.test_api import get_test_client, make_request, suppress_stdout +from frappe.tests.utils import FrappeTestCase if TYPE_CHECKING: from frappe.integrations.doctype.social_login_key.social_login_key import SocialLoginKey -class FrappeRequestTestCase(unittest.TestCase): +class FrappeRequestTestCase(FrappeTestCase): @property def sid(self) -> str: if not getattr(self, "_sid", None): @@ -57,6 +57,7 @@ class TestOAuth20(FrappeRequestTestCase): @classmethod def setUpClass(cls): + super().setUpClass() make_test_records("User") cls.form_header = {"content-type": "application/x-www-form-urlencoded"} diff --git a/frappe/tests/test_password.py b/frappe/tests/test_password.py index a6864f360c..d58118d6f7 100644 --- a/frappe/tests/test_password.py +++ b/frappe/tests/test_password.py @@ -1,14 +1,13 @@ # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE -import unittest - from cryptography.fernet import Fernet import frappe +from frappe.tests.utils import FrappeTestCase from frappe.utils.password import check_password, decrypt, encrypt, passlibctx, update_password -class TestPassword(unittest.TestCase): +class TestPassword(FrappeTestCase): def setUp(self): frappe.delete_doc("Email Account", "Test Email Account Password") frappe.delete_doc("Email Account", "Test Email Account Password-new") diff --git a/frappe/tests/test_patches.py b/frappe/tests/test_patches.py index aefd6da4e5..f12f3a182c 100644 --- a/frappe/tests/test_patches.py +++ b/frappe/tests/test_patches.py @@ -1,9 +1,9 @@ -import unittest from pathlib import Path from unittest.mock import mock_open, patch import frappe from frappe.modules import patch_handler +from frappe.tests.utils import FrappeTestCase from frappe.utils import get_bench_path EMTPY_FILE = "" @@ -49,7 +49,7 @@ app.module.patch4 """ -class TestPatches(unittest.TestCase): +class TestPatches(FrappeTestCase): def test_patch_module_names(self): frappe.flags.final_patches = [] frappe.flags.in_install = True @@ -79,7 +79,7 @@ class TestPatches(unittest.TestCase): self.assertGreaterEqual(finished_patches, len(all_patches)) -class TestPatchReader(unittest.TestCase): +class TestPatchReader(FrappeTestCase): def get_patches(self): return ( patch_handler.get_patches_from_app("frappe"), diff --git a/frappe/tests/test_pdf.py b/frappe/tests/test_pdf.py index 8f2a2c1cfa..4a8fef253c 100644 --- a/frappe/tests/test_pdf.py +++ b/frappe/tests/test_pdf.py @@ -1,15 +1,15 @@ # Copyright (c) 2018, Frappe Technologies Pvt. Ltd. and Contributors # License: MIT. See LICENSE import io -import unittest from PyPDF2 import PdfReader import frappe import frappe.utils.pdf as pdfgen +from frappe.tests.utils import FrappeTestCase -class TestPdf(unittest.TestCase): +class TestPdf(FrappeTestCase): @property def html(self): return """