039be73af4
Found all usage using this semgrep rule:
```yaml
- pattern: frappe.db.set_value($DOCTYPE, ...)
- pattern-not: frappe.db.set_value("$STR", ...)
```
114 lines
3.3 KiB
Python
114 lines
3.3 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
import frappe
|
|
import frappe.desk.form.assign_to
|
|
from frappe.automation.doctype.assignment_rule.test_assignment_rule import (
|
|
TEST_DOCTYPE,
|
|
_make_test_record,
|
|
create_test_doctype,
|
|
)
|
|
from frappe.desk.form.load import get_assignments
|
|
from frappe.desk.listview import get_group_by_count
|
|
from frappe.tests.utils import FrappeTestCase
|
|
|
|
|
|
class TestAssign(FrappeTestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().setUpClass()
|
|
create_test_doctype(TEST_DOCTYPE)
|
|
|
|
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()
|
|
|
|
self._test_basic_assign_on_document(todo)
|
|
|
|
def _test_basic_assign_on_document(self, doc):
|
|
added = assign(doc, "test@example.com")
|
|
|
|
self.assertTrue("test@example.com" in [d.owner for d in added])
|
|
|
|
frappe.desk.form.assign_to.remove(doc.doctype, doc.name, "test@example.com")
|
|
|
|
# assignment is cleared
|
|
assignments = frappe.desk.form.assign_to.get(dict(doctype=doc.doctype, name=doc.name))
|
|
self.assertEqual(len(assignments), 0)
|
|
|
|
def test_assign_single(self):
|
|
c = frappe.get_doc("Contact Us Settings")
|
|
self._test_basic_assign_on_document(c)
|
|
|
|
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_test_record()
|
|
assign(note, "test_assign1@example.com")
|
|
|
|
note = _make_test_record(public=1)
|
|
assign(note, "test_assign2@example.com")
|
|
|
|
note = _make_test_record(public=1)
|
|
assign(note, "test_assign2@example.com")
|
|
|
|
note = _make_test_record()
|
|
assign(note, "test_assign2@example.com")
|
|
|
|
data = {d.name: d.count for d in get_group_by_count(TEST_DOCTYPE, "[]", "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(TEST_DOCTYPE, '[{"public": 1}]', "assigned_to")
|
|
}
|
|
|
|
self.assertFalse("test_assign1@example.com" in data)
|
|
self.assertEqual(data["test_assign2@example.com"], 2)
|
|
|
|
frappe.db.rollback()
|
|
|
|
def test_assignment_removal(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()
|
|
|
|
new_todo = assign(todo, "test@example.com")
|
|
|
|
# remove assignment
|
|
frappe.db.set_value("ToDo", new_todo[0].name, "allocated_to", "")
|
|
|
|
self.assertFalse(get_assignments("ToDo", todo.name))
|
|
|
|
|
|
def assign(doc, user):
|
|
return frappe.desk.form.assign_to.add(
|
|
{
|
|
"assign_to": [user],
|
|
"doctype": doc.doctype,
|
|
"name": doc.name,
|
|
"description": "test",
|
|
}
|
|
)
|