import frappe
from frappe.model.document import Document, get_doc
from frappe.tests import IntegrationTestCase
from frappe.types import DocRef

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_value_representation(self):
		# Test the value representation of DocRef
		doc_ref = DocRef("User", "test@example.com")
		self.assertEqual(doc_ref.__value__(), "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")