[feature] override a function regionally by adding a decorator

This commit is contained in:
Rushabh Mehta 2017-07-13 15:07:51 +05:30
parent 7231f29e78
commit 8f2e21def2

View File

@ -0,0 +1,13 @@
import unittest, frappe, erpnext
@erpnext.regional
def test_method():
return 'original'
class TestInit(unittest.TestCase):
def test_regional_overrides(self):
frappe.flags.country = 'India'
self.assertEqual(test_method(), 'overridden')
frappe.flags.country = 'Nepal'
self.assertEqual(test_method(), 'original')