mirror of
https://github.com/frappe/erpnext.git
synced 2026-09-08 23:09:33 +00:00
* fix: check write permission in whitelisted document methods * test: permission coverage for production plan status roll-ups * fix: add type hints to whitelisted arguments and submit MR in test
This commit is contained in:
@@ -237,7 +237,9 @@ class Lead(SellingController, CRMNote):
|
||||
return frappe.db.get_value("Quotation", {"party_name": self.name, "docstatus": 1, "status": "Lost"})
|
||||
|
||||
@frappe.whitelist()
|
||||
def create_prospect_and_contact(self, data):
|
||||
def create_prospect_and_contact(self, data: dict):
|
||||
self.check_permission("write")
|
||||
|
||||
data = frappe._dict(data)
|
||||
if data.create_contact:
|
||||
self.create_contact()
|
||||
|
||||
@@ -687,7 +687,9 @@ class ProductionPlan(Document):
|
||||
frappe.delete_doc("Work Order", d.name)
|
||||
|
||||
@frappe.whitelist()
|
||||
def set_status(self, close=None, update_bin=False):
|
||||
def set_status(self, close: bool | None = None, update_bin: bool = False):
|
||||
self.check_permission("write")
|
||||
|
||||
self.status = {0: "Draft", 1: "Submitted", 2: "Cancelled"}.get(self.docstatus)
|
||||
|
||||
if close:
|
||||
|
||||
@@ -3167,6 +3167,46 @@ class TestProductionPlan(ERPNextTestSuite):
|
||||
"The phantom BOM was not re-exploded for the second po_item.",
|
||||
)
|
||||
|
||||
def test_set_status_requires_write_permission(self):
|
||||
pln = create_production_plan(item_code="Test Production Item 1")
|
||||
|
||||
with self.set_user(create_user_without_production_plan_access()):
|
||||
doc = frappe.get_doc("Production Plan", pln.name)
|
||||
self.assertRaises(frappe.PermissionError, doc.set_status)
|
||||
|
||||
def test_work_order_status_rollup_without_production_plan_permission(self):
|
||||
pln = create_production_plan(item_code="Test Production Item 1")
|
||||
pln.make_work_order()
|
||||
|
||||
wo_name = frappe.db.get_value("Work Order", {"production_plan": pln.name}, "name")
|
||||
frappe.db.set_value("Production Plan Item", pln.po_items[0].name, "ordered_qty", 99)
|
||||
|
||||
with self.set_user(create_user_without_production_plan_access()):
|
||||
frappe.get_doc("Work Order", wo_name).update_ordered_qty()
|
||||
|
||||
pln.reload()
|
||||
self.assertEqual(pln.po_items[0].ordered_qty, 0.0)
|
||||
self.assertEqual(pln.status, "Submitted")
|
||||
|
||||
def test_material_request_status_rollup_without_production_plan_permission(self):
|
||||
pln = create_production_plan(item_code="Test Production Item 1")
|
||||
pln.make_material_request()
|
||||
|
||||
plan_item = pln.mr_items[0].name
|
||||
mr_name = frappe.db.get_value(
|
||||
"Material Request Item", {"material_request_plan_item": plan_item}, "parent"
|
||||
)
|
||||
frappe.get_doc("Material Request", mr_name).submit()
|
||||
frappe.db.set_value("Material Request Plan Item", plan_item, "requested_qty", 0)
|
||||
|
||||
with self.set_user(create_user_without_production_plan_access()):
|
||||
frappe.get_doc("Material Request", mr_name).update_requested_qty_in_production_plan()
|
||||
|
||||
pln.reload()
|
||||
requested_qty = frappe.db.get_value("Material Request Plan Item", plan_item, "requested_qty")
|
||||
self.assertGreater(requested_qty, 0)
|
||||
self.assertEqual(pln.status, "Material Requested")
|
||||
|
||||
|
||||
def create_production_plan(**args):
|
||||
"""
|
||||
@@ -3299,3 +3339,19 @@ def make_bom(**args):
|
||||
frappe.set_value("Item", args.item, "default_bom", bom.name)
|
||||
|
||||
return bom
|
||||
|
||||
|
||||
def create_user_without_production_plan_access():
|
||||
user = "test_production_plan_no_access@example.com"
|
||||
if not frappe.db.exists("User", user):
|
||||
frappe.get_doc(
|
||||
{
|
||||
"doctype": "User",
|
||||
"email": user,
|
||||
"first_name": "Production Plan No Access",
|
||||
"send_welcome_email": 0,
|
||||
"roles": [{"doctype": "Has Role", "role": "Stock User"}],
|
||||
}
|
||||
).insert(ignore_permissions=True)
|
||||
|
||||
return user
|
||||
|
||||
@@ -924,6 +924,7 @@ class WorkOrder(Document):
|
||||
|
||||
def update_production_plan_status(self):
|
||||
production_plan = frappe.get_doc("Production Plan", self.production_plan)
|
||||
production_plan.flags.ignore_permissions = True
|
||||
produced_qty = 0
|
||||
if self.production_plan_item:
|
||||
total_qty = frappe.get_all(
|
||||
@@ -1351,6 +1352,7 @@ class WorkOrder(Document):
|
||||
)
|
||||
|
||||
doc = frappe.get_doc("Production Plan", self.production_plan)
|
||||
doc.flags.ignore_permissions = True
|
||||
doc.set_status()
|
||||
doc.db_set("status", doc.status)
|
||||
|
||||
|
||||
@@ -157,6 +157,8 @@ class ImportSupplierInvoice(Document):
|
||||
|
||||
@frappe.whitelist()
|
||||
def process_file_data(self):
|
||||
self.check_permission("write")
|
||||
|
||||
self.db_set("status", "Processing File Data", notify=True, commit=True)
|
||||
frappe.enqueue_doc(self.doctype, self.name, "import_xml_data", queue="long", timeout=3600)
|
||||
|
||||
|
||||
@@ -157,6 +157,8 @@ class Batch(Document):
|
||||
|
||||
@frappe.whitelist()
|
||||
def recalculate_batch_qty(self):
|
||||
self.check_permission("write")
|
||||
|
||||
batches = get_batch_qty(
|
||||
batch_no=self.name,
|
||||
item_code=self.item,
|
||||
|
||||
@@ -483,6 +483,7 @@ class MaterialRequest(BuyingController):
|
||||
|
||||
for production_plan in production_plans:
|
||||
doc = frappe.get_doc("Production Plan", production_plan)
|
||||
doc.flags.ignore_permissions = True
|
||||
doc.set_status()
|
||||
doc.db_set("status", doc.status)
|
||||
|
||||
|
||||
@@ -277,6 +277,8 @@ class RepostItemValuation(Document):
|
||||
|
||||
@frappe.whitelist()
|
||||
def restart_reposting(self):
|
||||
self.check_permission("write")
|
||||
|
||||
self.set_status("Queued", write=False)
|
||||
self.current_index = 0
|
||||
self.distinct_item_and_warehouse = None
|
||||
|
||||
@@ -153,6 +153,8 @@ class StockClosingEntry(Document):
|
||||
|
||||
@frappe.whitelist(methods=["POST"])
|
||||
def enqueue_job(self):
|
||||
self.check_permission("write")
|
||||
|
||||
self.db_set("status", "In Progress")
|
||||
enqueue(prepare_closing_stock_balance, name=self.name, queue="long", timeout=1500)
|
||||
frappe.msgprint(
|
||||
@@ -163,6 +165,7 @@ class StockClosingEntry(Document):
|
||||
|
||||
@frappe.whitelist(methods=["POST"])
|
||||
def regenerate_closing_balance(self):
|
||||
self.check_permission("write")
|
||||
self.validate_closed_period_lock()
|
||||
self.remove_stock_closing()
|
||||
self.enqueue_job()
|
||||
|
||||
@@ -73,6 +73,8 @@ class StockRepostingSettings(Document):
|
||||
def convert_to_item_wh_reposting(self):
|
||||
"""Convert Transaction reposting to Item Warehouse based reposting if Item Based Reposting has enabled."""
|
||||
|
||||
self.check_permission("write")
|
||||
|
||||
reposting_data = get_reposting_entries()
|
||||
|
||||
vouchers = [d.voucher_no for d in reposting_data]
|
||||
|
||||
Reference in New Issue
Block a user