235 lines
7.1 KiB
Python
235 lines
7.1 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: GNU General Public License v3. See license.txt
|
|
|
|
from __future__ import unicode_literals
|
|
import unittest
|
|
import frappe
|
|
from frappe.utils import nowdate, add_months
|
|
from erpnext.shopping_cart.cart import _get_cart_quotation, update_cart, get_party
|
|
from erpnext.tests.utils import create_test_contact_and_address
|
|
|
|
|
|
# test_dependencies = ['Payment Terms Template']
|
|
|
|
class TestShoppingCart(unittest.TestCase):
|
|
"""
|
|
Note:
|
|
Shopping Cart == Quotation
|
|
"""
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
frappe.db.sql("delete from `tabTax Rule`")
|
|
|
|
def setUp(self):
|
|
frappe.set_user("Administrator")
|
|
create_test_contact_and_address()
|
|
self.enable_shopping_cart()
|
|
|
|
def tearDown(self):
|
|
frappe.set_user("Administrator")
|
|
self.disable_shopping_cart()
|
|
|
|
def test_get_cart_new_user(self):
|
|
self.login_as_new_user()
|
|
|
|
# test if lead is created and quotation with new lead is fetched
|
|
quotation = _get_cart_quotation()
|
|
self.assertEqual(quotation.quotation_to, "Customer")
|
|
self.assertEqual(quotation.contact_person,
|
|
frappe.db.get_value("Contact", dict(email_id="test_cart_user@example.com")))
|
|
self.assertEqual(quotation.contact_email, frappe.session.user)
|
|
|
|
return quotation
|
|
|
|
def test_get_cart_customer(self):
|
|
self.login_as_customer()
|
|
|
|
# test if quotation with customer is fetched
|
|
quotation = _get_cart_quotation()
|
|
self.assertEqual(quotation.quotation_to, "Customer")
|
|
self.assertEqual(quotation.party_name, "_Test Customer")
|
|
self.assertEqual(quotation.contact_email, frappe.session.user)
|
|
|
|
return quotation
|
|
|
|
def test_add_to_cart(self):
|
|
self.login_as_customer()
|
|
|
|
# clear existing quotations
|
|
self.clear_existing_quotations()
|
|
|
|
# add first item
|
|
update_cart("_Test Item", 1)
|
|
|
|
quotation = self.test_get_cart_customer()
|
|
|
|
self.assertEqual(quotation.get("items")[0].item_code, "_Test Item")
|
|
self.assertEqual(quotation.get("items")[0].qty, 1)
|
|
self.assertEqual(quotation.get("items")[0].amount, 10)
|
|
|
|
# add second item
|
|
update_cart("_Test Item 2", 1)
|
|
quotation = self.test_get_cart_customer()
|
|
self.assertEqual(quotation.get("items")[1].item_code, "_Test Item 2")
|
|
self.assertEqual(quotation.get("items")[1].qty, 1)
|
|
self.assertEqual(quotation.get("items")[1].amount, 20)
|
|
|
|
self.assertEqual(len(quotation.get("items")), 2)
|
|
|
|
def test_update_cart(self):
|
|
# first, add to cart
|
|
self.test_add_to_cart()
|
|
|
|
# update first item
|
|
update_cart("_Test Item", 5)
|
|
quotation = self.test_get_cart_customer()
|
|
self.assertEqual(quotation.get("items")[0].item_code, "_Test Item")
|
|
self.assertEqual(quotation.get("items")[0].qty, 5)
|
|
self.assertEqual(quotation.get("items")[0].amount, 50)
|
|
self.assertEqual(quotation.net_total, 70)
|
|
self.assertEqual(len(quotation.get("items")), 2)
|
|
|
|
def test_remove_from_cart(self):
|
|
# first, add to cart
|
|
self.test_add_to_cart()
|
|
|
|
# remove first item
|
|
update_cart("_Test Item", 0)
|
|
quotation = self.test_get_cart_customer()
|
|
|
|
self.assertEqual(quotation.get("items")[0].item_code, "_Test Item 2")
|
|
self.assertEqual(quotation.get("items")[0].qty, 1)
|
|
self.assertEqual(quotation.get("items")[0].amount, 20)
|
|
self.assertEqual(quotation.net_total, 20)
|
|
self.assertEqual(len(quotation.get("items")), 1)
|
|
|
|
def test_tax_rule(self):
|
|
self.create_tax_rule()
|
|
self.login_as_customer()
|
|
quotation = self.create_quotation()
|
|
|
|
from erpnext.accounts.party import set_taxes
|
|
|
|
tax_rule_master = set_taxes(quotation.party_name, "Customer",
|
|
quotation.transaction_date, quotation.company, customer_group=None, supplier_group=None,
|
|
tax_category=quotation.tax_category, billing_address=quotation.customer_address,
|
|
shipping_address=quotation.shipping_address_name, use_for_shopping_cart=1)
|
|
|
|
self.assertEqual(quotation.taxes_and_charges, tax_rule_master)
|
|
self.assertEqual(quotation.total_taxes_and_charges, 1000.0)
|
|
|
|
self.remove_test_quotation(quotation)
|
|
|
|
def create_tax_rule(self):
|
|
tax_rule = frappe.get_test_records("Tax Rule")[0]
|
|
try:
|
|
frappe.get_doc(tax_rule).insert()
|
|
except frappe.DuplicateEntryError:
|
|
pass
|
|
|
|
def create_quotation(self):
|
|
quotation = frappe.new_doc("Quotation")
|
|
|
|
values = {
|
|
"doctype": "Quotation",
|
|
"quotation_to": "Customer",
|
|
"order_type": "Shopping Cart",
|
|
"party_name": get_party(frappe.session.user).name,
|
|
"docstatus": 0,
|
|
"contact_email": frappe.session.user,
|
|
"selling_price_list": "_Test Price List Rest of the World",
|
|
"currency": "USD",
|
|
"taxes_and_charges" : "_Test Tax 1 - _TC",
|
|
"conversion_rate":1,
|
|
"transaction_date" : nowdate(),
|
|
"valid_till" : add_months(nowdate(), 1),
|
|
"items": [{
|
|
"item_code": "_Test Item",
|
|
"qty": 1
|
|
}],
|
|
"taxes": frappe.get_doc("Sales Taxes and Charges Template", "_Test Tax 1 - _TC").taxes,
|
|
"company": "_Test Company"
|
|
}
|
|
|
|
quotation.update(values)
|
|
|
|
quotation.insert(ignore_permissions=True)
|
|
|
|
return quotation
|
|
|
|
def remove_test_quotation(self, quotation):
|
|
frappe.set_user("Administrator")
|
|
quotation.delete()
|
|
|
|
# helper functions
|
|
def enable_shopping_cart(self):
|
|
settings = frappe.get_doc("Shopping Cart Settings", "Shopping Cart Settings")
|
|
|
|
settings.update({
|
|
"enabled": 1,
|
|
"company": "_Test Company",
|
|
"default_customer_group": "_Test Customer Group",
|
|
"quotation_series": "_T-Quotation-",
|
|
"price_list": "_Test Price List India"
|
|
})
|
|
|
|
# insert item price
|
|
if not frappe.db.get_value("Item Price", {"price_list": "_Test Price List India",
|
|
"item_code": "_Test Item"}):
|
|
frappe.get_doc({
|
|
"doctype": "Item Price",
|
|
"price_list": "_Test Price List India",
|
|
"item_code": "_Test Item",
|
|
"price_list_rate": 10
|
|
}).insert()
|
|
frappe.get_doc({
|
|
"doctype": "Item Price",
|
|
"price_list": "_Test Price List India",
|
|
"item_code": "_Test Item 2",
|
|
"price_list_rate": 20
|
|
}).insert()
|
|
|
|
settings.save()
|
|
frappe.local.shopping_cart_settings = None
|
|
|
|
def disable_shopping_cart(self):
|
|
settings = frappe.get_doc("Shopping Cart Settings", "Shopping Cart Settings")
|
|
settings.enabled = 0
|
|
settings.save()
|
|
frappe.local.shopping_cart_settings = None
|
|
|
|
def login_as_new_user(self):
|
|
self.create_user_if_not_exists("test_cart_user@example.com")
|
|
frappe.set_user("test_cart_user@example.com")
|
|
|
|
def login_as_customer(self):
|
|
self.create_user_if_not_exists("test_contact_customer@example.com",
|
|
"_Test Contact For _Test Customer")
|
|
frappe.set_user("test_contact_customer@example.com")
|
|
|
|
def clear_existing_quotations(self):
|
|
quotations = frappe.get_all("Quotation", filters={
|
|
"party_name": get_party().name,
|
|
"order_type": "Shopping Cart",
|
|
"docstatus": 0
|
|
}, order_by="modified desc", pluck="name")
|
|
|
|
for quotation in quotations:
|
|
frappe.delete_doc("Quotation", quotation, ignore_permissions=True, force=True)
|
|
|
|
def create_user_if_not_exists(self, email, first_name = None):
|
|
if frappe.db.exists("User", email):
|
|
return
|
|
|
|
frappe.get_doc({
|
|
"doctype": "User",
|
|
"user_type": "Website User",
|
|
"email": email,
|
|
"send_welcome_email": 0,
|
|
"first_name": first_name or email.split("@")[0]
|
|
}).insert(ignore_permissions=True)
|
|
|
|
test_dependencies = ["Sales Taxes and Charges Template", "Price List", "Item Price", "Shipping Rule", "Currency Exchange",
|
|
"Customer Group", "Lead", "Customer", "Contact", "Address", "Item", "Tax Rule"]
|