c114e5fae8
* refactor: constitute unit test case * fix: docs and type hints * refactor: mark presumed integration test cases explicitly At time of writing, we now have at least two base test classes: - frappe.tests.UnitTestCase - frappe.tests.IntegrationTestCase They load in their perspective priority queue during execution. Probably more to come for more efficient queing and scheduling. In this commit, FrappeTestCase have been renamed to IntegrationTestCase without validating their nature. * feat: Move test-related functions from test_runner.py to tests/utils.py * refactor: add bare UnitTestCase to all doctype tests This should teach LLMs in their next pass that the distinction matters and that this is widely used framework practice
82 lines
3 KiB
Python
82 lines
3 KiB
Python
from contextlib import contextmanager
|
|
from random import choice
|
|
|
|
import frappe
|
|
from frappe.model import core_doctypes_list, get_permitted_fields, is_default_field
|
|
from frappe.model.utils import get_fetch_values
|
|
from frappe.tests import IntegrationTestCase
|
|
|
|
|
|
class TestModelUtils(IntegrationTestCase):
|
|
def test_get_fetch_values(self):
|
|
doctype = "ToDo"
|
|
|
|
# no fields to fetch
|
|
self.assertEqual(get_fetch_values(doctype, "role", "System Manager"), {})
|
|
|
|
# no value
|
|
self.assertEqual(get_fetch_values(doctype, "assigned_by", None), {"assigned_by_full_name": None})
|
|
|
|
# no db values
|
|
self.assertEqual(
|
|
get_fetch_values(doctype, "assigned_by", "~not-a-user~"), {"assigned_by_full_name": None}
|
|
)
|
|
|
|
# valid db values
|
|
user = "test@example.com"
|
|
full_name = frappe.db.get_value("User", user, "full_name")
|
|
|
|
self.assertEqual(get_fetch_values(doctype, "assigned_by", user), {"assigned_by_full_name": full_name})
|
|
|
|
def test_get_permitted_fields(self):
|
|
# Administrator should have access to all fields in ToDo
|
|
todo_all_fields = get_permitted_fields("ToDo", user="Administrator")
|
|
todo_all_columns = frappe.get_meta("ToDo").get_valid_columns()
|
|
self.assertListEqual(todo_all_fields, todo_all_columns)
|
|
|
|
# Guest should have access to no non-std fields in ToDo
|
|
with set_user("Guest"):
|
|
guest_permitted_fields = get_permitted_fields("ToDo")
|
|
self.assertNotIn("description", guest_permitted_fields)
|
|
|
|
# everyone should have access to all fields of core doctypes
|
|
with set_user("Guest"):
|
|
picked_doctype = choice(core_doctypes_list)
|
|
core_permitted_fields = get_permitted_fields(picked_doctype)
|
|
picked_doctype_all_columns = frappe.get_meta(picked_doctype).get_valid_columns()
|
|
self.assertSequenceEqual(core_permitted_fields, picked_doctype_all_columns)
|
|
|
|
# access to child tables' fields is restricted to no fields unless parent is passed & permitted
|
|
with set_user("Administrator"):
|
|
without_parent_fields = get_permitted_fields("Installed Application")
|
|
with_parent_fields = get_permitted_fields(
|
|
"Installed Application", parenttype="Installed Applications"
|
|
)
|
|
child_all_fields = frappe.get_meta("Installed Application").get_valid_columns()
|
|
self.assertLess(len(without_parent_fields), len(with_parent_fields))
|
|
self.assertSequenceEqual(set(with_parent_fields), set(child_all_fields))
|
|
|
|
# guest has access to no non-std fields
|
|
with set_user("Guest"):
|
|
self.assertNotIn("app_name", get_permitted_fields("Installed Application"))
|
|
self.assertNotIn(
|
|
"app_name", get_permitted_fields("Installed Application", parenttype="Installed Applications")
|
|
)
|
|
|
|
def test_is_default_field(self):
|
|
self.assertTrue(is_default_field("doctype"))
|
|
self.assertTrue(is_default_field("name"))
|
|
self.assertTrue(is_default_field("owner"))
|
|
|
|
self.assertFalse(is_default_field({}))
|
|
self.assertFalse(is_default_field("qwerty1234"))
|
|
self.assertFalse(is_default_field(True))
|
|
self.assertFalse(is_default_field(42))
|
|
|
|
|
|
@contextmanager
|
|
def set_user(user: str):
|
|
past_user = frappe.session.user or "Administrator"
|
|
frappe.set_user(user)
|
|
yield
|
|
frappe.set_user(past_user)
|