3446026555
The license.txt file has been replaced with LICENSE for quite a while now. INAL but it didn't seem accurate to say "hey, checkout license.txt although there's no such file". Apart from this, there were inconsistencies in the headers altogether...this change brings consistency.
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
import frappe, unittest
|
|
import frappe.desk.form.assign_to
|
|
from frappe.desk.listview import get_group_by_count
|
|
from frappe.automation.doctype.assignment_rule.test_assignment_rule import make_note
|
|
|
|
class TestAssign(unittest.TestCase):
|
|
def test_assign(self):
|
|
todo = frappe.get_doc({"doctype":"ToDo", "description": "test"}).insert()
|
|
if not frappe.db.exists("User", "test@example.com"):
|
|
frappe.get_doc({"doctype":"User", "email":"test@example.com", "first_name":"Test"}).insert()
|
|
|
|
added = assign(todo, "test@example.com")
|
|
|
|
self.assertTrue("test@example.com" in [d.owner for d in added])
|
|
|
|
removed = frappe.desk.form.assign_to.remove(todo.doctype, todo.name, "test@example.com")
|
|
|
|
# assignment is cleared
|
|
assignments = frappe.desk.form.assign_to.get(dict(doctype = todo.doctype, name=todo.name))
|
|
self.assertEqual(len(assignments), 0)
|
|
|
|
def test_assignment_count(self):
|
|
frappe.db.delete("ToDo")
|
|
|
|
if not frappe.db.exists("User", "test_assign1@example.com"):
|
|
frappe.get_doc({"doctype":"User", "email":"test_assign1@example.com", "first_name":"Test", "roles": [{"role": "System Manager"}]}).insert()
|
|
|
|
if not frappe.db.exists("User", "test_assign2@example.com"):
|
|
frappe.get_doc({"doctype":"User", "email":"test_assign2@example.com", "first_name":"Test", "roles": [{"role": "System Manager"}]}).insert()
|
|
|
|
note = make_note()
|
|
assign(note, "test_assign1@example.com")
|
|
|
|
note = make_note(dict(public=1))
|
|
assign(note, "test_assign2@example.com")
|
|
|
|
note = make_note(dict(public=1))
|
|
assign(note, "test_assign2@example.com")
|
|
|
|
note = make_note()
|
|
assign(note, "test_assign2@example.com")
|
|
|
|
data = {d.name: d.count for d in get_group_by_count('Note', '[]', 'assigned_to')}
|
|
|
|
self.assertTrue('test_assign1@example.com' in data)
|
|
self.assertEqual(data['test_assign1@example.com'], 1)
|
|
self.assertEqual(data['test_assign2@example.com'], 3)
|
|
|
|
data = {d.name: d.count for d in get_group_by_count('Note', '[{"public": 1}]', 'assigned_to')}
|
|
|
|
self.assertFalse('test_assign1@example.com' in data)
|
|
self.assertEqual(data['test_assign2@example.com'], 2)
|
|
|
|
frappe.db.rollback()
|
|
|
|
|
|
def assign(doc, user):
|
|
return frappe.desk.form.assign_to.add({
|
|
"assign_to": [user],
|
|
"doctype": doc.doctype,
|
|
"name": doc.name,
|
|
"description": 'test',
|
|
})
|