40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
import unittest
|
|
|
|
import frappe
|
|
from erpnext import encode_company_abbr
|
|
from six.moves import range
|
|
|
|
test_records = frappe.get_test_records('Company')
|
|
|
|
class TestInit(unittest.TestCase):
|
|
def test_encode_company_abbr(self):
|
|
company = frappe.new_doc("Company")
|
|
company.company_name = "New from Existing Company For Test"
|
|
company.abbr = "NFECT"
|
|
company.default_currency = "INR"
|
|
company.save()
|
|
|
|
abbr = company.abbr
|
|
|
|
names = [
|
|
"Warehouse Name", "ERPNext Foundation India", "Gold - Member - {a}".format(a=abbr),
|
|
" - {a}".format(a=abbr), "ERPNext - Foundation - India",
|
|
"ERPNext Foundation India - {a}".format(a=abbr),
|
|
"No-Space-{a}".format(a=abbr), "- Warehouse"
|
|
]
|
|
|
|
expected_names = [
|
|
"Warehouse Name - {a}".format(a=abbr), "ERPNext Foundation India - {a}".format(a=abbr),
|
|
"Gold - Member - {a}".format(a=abbr), " - {a}".format(a=abbr),
|
|
"ERPNext - Foundation - India - {a}".format(a=abbr),
|
|
"ERPNext Foundation India - {a}".format(a=abbr), "No-Space-{a} - {a}".format(a=abbr),
|
|
"- Warehouse - {a}".format(a=abbr)
|
|
]
|
|
|
|
for i in range(len(names)):
|
|
enc_name = encode_company_abbr(names[i], company.name)
|
|
self.assertTrue(
|
|
enc_name == expected_names[i],
|
|
"{enc} is not same as {exp}".format(enc=enc_name, exp=expected_names[i])
|
|
)
|