import frappe
from frappe.tests.utils import FrappeTestCase

from erpnext.accounts.party import get_default_price_list


class PartyTestCase(FrappeTestCase):
	def test_get_default_price_list_should_return_none_for_invalid_group(self):
		customer = frappe.get_doc({
			'doctype': 'Customer',
			'customer_name': 'test customer',
		}).insert(ignore_permissions=True, ignore_mandatory=True)
		customer.customer_group = None
		customer.save()
		price_list = get_default_price_list(customer)
		assert price_list is None