Shiloh/brotherton-erpnext
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test: set country to India before running regional tests

Browse Source
This commit is contained in:
Rucha Mahabal 2022-05-23 14:15:36 +05:30 committed by Nabin Hait
parent 00adda7c8d
commit 2e98e9e0b9
2 changed files with 50 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
erpnext/payroll/doctype/employee_tax_exemption_declaration/test_employee_tax_exemption_declaration.py
View File

@ -116,6 +116,10 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
self.assertEqual(declaration.total_exemption_amount, 100000) self.assertEqual(declaration.total_exemption_amount, 100000)
def test_india_hra_exemption(self): def test_india_hra_exemption(self):
# set country
current_country = frappe.flags.country
frappe.flags.country = "India"
setup_hra_exemption_prerequisites("Monthly") setup_hra_exemption_prerequisites("Monthly")
employee = frappe.get_value("Employee", {"user_id": "employee@taxexemption.com"}, "name") employee = frappe.get_value("Employee", {"user_id": "employee@taxexemption.com"}, "name")
@ -124,7 +128,7 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
"doctype": "Employee Tax Exemption Declaration", "doctype": "Employee Tax Exemption Declaration",
"employee": employee, "employee": employee,
"company": "_Test Company", "company": "_Test Company",
"payroll_period": "_Test Payroll Period 1", "payroll_period": "_Test Payroll Period",
"currency": "INR", "currency": "INR",
"monthly_house_rent": 50000, "monthly_house_rent": 50000,
"rented_in_metro_city": 1, "rented_in_metro_city": 1,
@ -150,7 +154,14 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
# 100000 Standard Exemption + 36000 HRA exemption # 100000 Standard Exemption + 36000 HRA exemption
self.assertEqual(declaration.total_exemption_amount, 136000) self.assertEqual(declaration.total_exemption_amount, 136000)
# reset
frappe.flags.country = current_country
def test_india_hra_exemption_with_daily_payroll_frequency(self): def test_india_hra_exemption_with_daily_payroll_frequency(self):
# set country
current_country = frappe.flags.country
frappe.flags.country = "India"
setup_hra_exemption_prerequisites("Daily") setup_hra_exemption_prerequisites("Daily")
employee = frappe.get_value("Employee", {"user_id": "employee@taxexemption.com"}, "name") employee = frappe.get_value("Employee", {"user_id": "employee@taxexemption.com"}, "name")
@ -159,7 +170,7 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
"doctype": "Employee Tax Exemption Declaration", "doctype": "Employee Tax Exemption Declaration",
"employee": employee, "employee": employee,
"company": "_Test Company", "company": "_Test Company",
"payroll_period": "_Test Payroll Period 1", "payroll_period": "_Test Payroll Period",
"currency": "INR", "currency": "INR",
"monthly_house_rent": 170000, "monthly_house_rent": 170000,
"rented_in_metro_city": 1, "rented_in_metro_city": 1,
@ -180,7 +191,14 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
# 50000 Standard Exemption + 215000 HRA exemption # 50000 Standard Exemption + 215000 HRA exemption
self.assertEqual(declaration.total_exemption_amount, 265000) self.assertEqual(declaration.total_exemption_amount, 265000)
# reset
frappe.flags.country = current_country
def test_india_hra_exemption_with_weekly_payroll_frequency(self): def test_india_hra_exemption_with_weekly_payroll_frequency(self):
# set country
current_country = frappe.flags.country
frappe.flags.country = "India"
setup_hra_exemption_prerequisites("Weekly") setup_hra_exemption_prerequisites("Weekly")
employee = frappe.get_value("Employee", {"user_id": "employee@taxexemption.com"}, "name") employee = frappe.get_value("Employee", {"user_id": "employee@taxexemption.com"}, "name")
@ -189,7 +207,7 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
"doctype": "Employee Tax Exemption Declaration", "doctype": "Employee Tax Exemption Declaration",
"employee": employee, "employee": employee,
"company": "_Test Company", "company": "_Test Company",
"payroll_period": "_Test Payroll Period 1", "payroll_period": "_Test Payroll Period",
"currency": "INR", "currency": "INR",
"monthly_house_rent": 170000, "monthly_house_rent": 170000,
"rented_in_metro_city": 1, "rented_in_metro_city": 1,
@ -210,7 +228,14 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
# 50000 Standard Exemption + 156000 HRA exemption # 50000 Standard Exemption + 156000 HRA exemption
self.assertEqual(declaration.total_exemption_amount, 206000) self.assertEqual(declaration.total_exemption_amount, 206000)
# reset
frappe.flags.country = current_country
def test_india_hra_exemption_with_fortnightly_payroll_frequency(self): def test_india_hra_exemption_with_fortnightly_payroll_frequency(self):
# set country
current_country = frappe.flags.country
frappe.flags.country = "India"
setup_hra_exemption_prerequisites("Fortnightly") setup_hra_exemption_prerequisites("Fortnightly")
employee = frappe.get_value("Employee", {"user_id": "employee@taxexemption.com"}, "name") employee = frappe.get_value("Employee", {"user_id": "employee@taxexemption.com"}, "name")
@ -219,7 +244,7 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
"doctype": "Employee Tax Exemption Declaration", "doctype": "Employee Tax Exemption Declaration",
"employee": employee, "employee": employee,
"company": "_Test Company", "company": "_Test Company",
"payroll_period": "_Test Payroll Period 1", "payroll_period": "_Test Payroll Period",
"currency": "INR", "currency": "INR",
"monthly_house_rent": 170000, "monthly_house_rent": 170000,
"rented_in_metro_city": 1, "rented_in_metro_city": 1,
@ -240,7 +265,14 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
# 50000 Standard Exemption + 78000 HRA exemption # 50000 Standard Exemption + 78000 HRA exemption
self.assertEqual(declaration.total_exemption_amount, 128000) self.assertEqual(declaration.total_exemption_amount, 128000)
# reset
frappe.flags.country = current_country
def test_india_hra_exemption_with_bimonthly_payroll_frequency(self): def test_india_hra_exemption_with_bimonthly_payroll_frequency(self):
# set country
current_country = frappe.flags.country
frappe.flags.country = "India"
setup_hra_exemption_prerequisites("Bimonthly") setup_hra_exemption_prerequisites("Bimonthly")
employee = frappe.get_value("Employee", {"user_id": "employee@taxexemption.com"}, "name") employee = frappe.get_value("Employee", {"user_id": "employee@taxexemption.com"}, "name")
@ -249,7 +281,7 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
"doctype": "Employee Tax Exemption Declaration", "doctype": "Employee Tax Exemption Declaration",
"employee": employee, "employee": employee,
"company": "_Test Company", "company": "_Test Company",
"payroll_period": "_Test Payroll Period 1", "payroll_period": "_Test Payroll Period",
"currency": "INR", "currency": "INR",
"monthly_house_rent": 50000, "monthly_house_rent": 50000,
"rented_in_metro_city": 1, "rented_in_metro_city": 1,
@ -275,6 +307,9 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
# 100000 Standard Exemption + 18000 HRA exemption # 100000 Standard Exemption + 18000 HRA exemption
self.assertEqual(declaration.total_exemption_amount, 118000) self.assertEqual(declaration.total_exemption_amount, 118000)
# reset
frappe.flags.country = current_country
def test_india_hra_exemption_with_multiple_salary_structure_assignments(self): def test_india_hra_exemption_with_multiple_salary_structure_assignments(self):
from erpnext.payroll.doctype.salary_slip.test_salary_slip import create_tax_slab from erpnext.payroll.doctype.salary_slip.test_salary_slip import create_tax_slab
from erpnext.payroll.doctype.salary_structure.test_salary_structure import ( from erpnext.payroll.doctype.salary_structure.test_salary_structure import (
@ -282,8 +317,12 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
make_salary_structure, make_salary_structure,
) )
# set country
current_country = frappe.flags.country
frappe.flags.country = "India"
employee = make_employee("employee@taxexemption2.com", company="_Test Company") employee = make_employee("employee@taxexemption2.com", company="_Test Company")
payroll_period = create_payroll_period(name="_Test Payroll Period 1", company="_Test Company") payroll_period = create_payroll_period(name="_Test Payroll Period", company="_Test Company")
create_tax_slab( create_tax_slab(
payroll_period, payroll_period,
@ -365,6 +404,9 @@ class TestEmployeeTaxExemptionDeclaration(FrappeTestCase):
# 50000 Standard Exemption + 102000 HRA exemption # 50000 Standard Exemption + 102000 HRA exemption
self.assertEqual(declaration.total_exemption_amount, 152000) self.assertEqual(declaration.total_exemption_amount, 152000)
# reset
frappe.flags.country = current_country
def create_payroll_period(**args): def create_payroll_period(**args):
args = frappe._dict(args) args = frappe._dict(args)
@ -422,7 +464,7 @@ def setup_hra_exemption_prerequisites(frequency):
from erpnext.payroll.doctype.salary_slip.test_salary_slip import create_tax_slab from erpnext.payroll.doctype.salary_slip.test_salary_slip import create_tax_slab
from erpnext.payroll.doctype.salary_structure.test_salary_structure import make_salary_structure from erpnext.payroll.doctype.salary_structure.test_salary_structure import make_salary_structure
payroll_period = create_payroll_period(name="_Test Payroll Period 1", company="_Test Company") payroll_period = create_payroll_period(name="_Test Payroll Period", company="_Test Company")
create_tax_slab( create_tax_slab(
payroll_period, payroll_period,

1
erpnext/regional/india/utils.py
View File

@ -369,6 +369,7 @@ def calculate_annual_eligible_hra_exemption(doc):
basic_component, hra_component = frappe.db.get_value( basic_component, hra_component = frappe.db.get_value(
"Company", doc.company, ["basic_component", "hra_component"] "Company", doc.company, ["basic_component", "hra_component"]
) )
if not (basic_component and hra_component): if not (basic_component and hra_component):
frappe.throw( frappe.throw(
_("Please set Basic and HRA component in Company {0}").format( _("Please set Basic and HRA component in Company {0}").format(