68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
import json
|
|
import unittest
|
|
|
|
import frappe
|
|
import frappe.utils
|
|
from frappe.model import mapper
|
|
from frappe.test_runner import make_test_records
|
|
from frappe.utils import add_months, nowdate
|
|
|
|
|
|
class TestMapper(unittest.TestCase):
|
|
def test_map_docs(self):
|
|
'''Test mapping of multiple source docs on a single target doc'''
|
|
|
|
make_test_records("Item")
|
|
items = ['_Test Item', '_Test Item 2', '_Test FG Item']
|
|
|
|
# Make source docs (quotations) and a target doc (sales order)
|
|
qtn1, item_list_1 = self.make_quotation(items, '_Test Customer')
|
|
qtn2, item_list_2 = self.make_quotation(items, '_Test Customer')
|
|
so, item_list_3 = self.make_sales_order()
|
|
|
|
# Map source docs to target with corresponding mapper method
|
|
method = "erpnext.selling.doctype.quotation.quotation.make_sales_order"
|
|
updated_so = mapper.map_docs(method, json.dumps([qtn1.name, qtn2.name]), so)
|
|
|
|
# Assert that all inserted items are present in updated sales order
|
|
src_items = item_list_1 + item_list_2 + item_list_3
|
|
self.assertEqual(set(d for d in src_items),
|
|
set(d.item_code for d in updated_so.items))
|
|
|
|
|
|
def make_quotation(self, item_list, customer):
|
|
|
|
qtn = frappe.get_doc({
|
|
"doctype": "Quotation",
|
|
"quotation_to": "Customer",
|
|
"party_name": customer,
|
|
"order_type": "Sales",
|
|
"transaction_date" : nowdate(),
|
|
"valid_till" : add_months(nowdate(), 1)
|
|
})
|
|
for item in item_list:
|
|
qtn.append("items", {"qty": "2", "item_code": item})
|
|
|
|
qtn.submit()
|
|
return qtn, item_list
|
|
|
|
def make_sales_order(self):
|
|
item = frappe.get_doc({
|
|
"base_amount": 1000.0,
|
|
"base_rate": 100.0,
|
|
"description": "CPU",
|
|
"doctype": "Sales Order Item",
|
|
"item_code": "_Test Item",
|
|
"item_name": "CPU",
|
|
"parentfield": "items",
|
|
"qty": 10.0,
|
|
"rate": 100.0,
|
|
"warehouse": "_Test Warehouse - _TC",
|
|
"stock_uom": "_Test UOM",
|
|
"conversion_factor": 1.0,
|
|
"uom": "_Test UOM"
|
|
})
|
|
so = frappe.get_doc(frappe.get_test_records('Sales Order')[0])
|
|
so.insert(ignore_permissions=True)
|
|
return so, [item.item_code]
|