From 67350e9ac3f2dc0fceb1899c8692adcd9cdd4213 Mon Sep 17 00:00:00 2001 From: Suraj Shetty Date: Wed, 20 Jul 2022 13:23:49 +0530 Subject: [PATCH] test: Add a test case to validate `get_unseen_notes` --- frappe/tests/test_boot.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 frappe/tests/test_boot.py diff --git a/frappe/tests/test_boot.py b/frappe/tests/test_boot.py new file mode 100644 index 0000000000..8f26a078b0 --- /dev/null +++ b/frappe/tests/test_boot.py @@ -0,0 +1,29 @@ +import unittest + +import frappe +from frappe.boot import get_unseen_notes +from frappe.desk.doctype.note.note import mark_as_seen + + +class TestBootData(unittest.TestCase): + def test_get_unseen_notes(self): + frappe.db.delete("Note") + frappe.db.delete("Note Seen By") + note = frappe.get_doc( + { + "doctype": "Note", + "title": "Test Note", + "notify_on_login": 1, + "content": "Test Note 1", + "public": 1, + } + ) + note.insert() + + frappe.set_user("test@example.com") + unseen_notes = [d.title for d in get_unseen_notes()] + self.assertListEqual(unseen_notes, ["Test Note"]) + + mark_as_seen(note.name) + unseen_notes = [d.title for d in get_unseen_notes()] + self.assertListEqual(unseen_notes, [])