7348572af8
* feat: add DocRef * feat: Add comprehensive test cases for DocRef functionality * chore(db): add field type hints * fix: ensure document stringer fulfills the DocRef contract
72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
import frappe
|
|
from frappe.model.document import DocRef, Document, get_doc
|
|
from frappe.tests import IntegrationTestCase
|
|
|
|
EXTRA_TEST_RECORD_DEPENDENCIES = ["User"]
|
|
|
|
|
|
class TestDocRef(IntegrationTestCase):
|
|
def test_doc_ref_get_doc(self):
|
|
# Test using DocRef with get_doc
|
|
doc_ref = DocRef("User", "test@example.com")
|
|
user = get_doc(doc_ref)
|
|
|
|
# Assert that user is an instance of both Document and DocRef
|
|
self.assertIsInstance(user, Document)
|
|
self.assertIsInstance(user, DocRef)
|
|
|
|
# Check more attributes
|
|
self.assertEqual(user.doctype, "User")
|
|
self.assertEqual(user.name, "test@example.com")
|
|
self.assertEqual(user.email, "test@example.com")
|
|
self.assertEqual(user.first_name, "_Test")
|
|
|
|
def test_doc_ref_in_query(self):
|
|
# Test using DocRef in a database query
|
|
user = frappe.get_doc("User", "test@example.com")
|
|
|
|
# Assert that user is an instance of both Document and DocRef
|
|
self.assertIsInstance(user, Document)
|
|
self.assertIsInstance(user, DocRef)
|
|
|
|
# Create a test document that references the user
|
|
test_doc = frappe.get_doc(
|
|
{
|
|
"doctype": "ToDo",
|
|
"description": "Test ToDo",
|
|
"reference_type": "User",
|
|
"reference_name": user, # This should work with DocRef
|
|
}
|
|
).insert()
|
|
|
|
# Getter using the DocRef
|
|
result = frappe.db.get_value("ToDo", {"reference_name": user}, ["name", "description"])
|
|
self.assertEqual(result[0], test_doc.name)
|
|
self.assertEqual(result[1], "Test ToDo")
|
|
# Setter using Document as DocRef
|
|
frappe.db.set_value("ToDo", test_doc, "description", "Revised Test ToDo")
|
|
test_doc.reload()
|
|
self.assertEqual(test_doc.description, "Revised Test ToDo")
|
|
|
|
def test_get_meta_with_doc_ref(self):
|
|
# Test get_meta with DocRef
|
|
doc_ref = DocRef("User", "test@example.com")
|
|
meta = frappe.get_meta(doc_ref)
|
|
|
|
# Check more attributes of the meta
|
|
self.assertEqual(meta.name, "User")
|
|
self.assertEqual(meta.module, "Core")
|
|
self.assertTrue("email" in [f.fieldname for f in meta.fields])
|
|
self.assertTrue("first_name" in [f.fieldname for f in meta.fields])
|
|
self.assertTrue("last_name" in [f.fieldname for f in meta.fields])
|
|
|
|
def test_doc_ref_str_representation(self):
|
|
# Test the string representation of DocRef
|
|
doc_ref = DocRef("User", "test@example.com")
|
|
self.assertEqual(str(doc_ref), "test@example.com")
|
|
|
|
def test_doc_ref_attributes(self):
|
|
# Test DocRef attributes
|
|
doc_ref = DocRef("User", "test@example.com")
|
|
self.assertEqual(doc_ref.doctype, "User")
|
|
self.assertEqual(doc_ref.name, "test@example.com")
|