e407b78506
* Remove six for PY2 compatability since our dependencies are not, PY2 is legacy. * Removed usages of utils from future/past libraries since they are deprecated. This includes 'from __future__ ...' and 'from past...' statements. * Removed compatibility imports for PY2, switched from six imports to standard library imports. * Removed utils code blocks that handle operations depending on PY2/3 versions. * Removed 'from __future__ ...' lines from templates/code generators * Used PY3 syntaxes in place of PY2 compatible blocks. eg: metaclass
26 lines
No EOL
965 B
Python
26 lines
No EOL
965 B
Python
|
|
import unittest, frappe
|
|
from frappe.utils.safe_exec import safe_exec, get_safe_globals
|
|
|
|
class TestSafeExec(unittest.TestCase):
|
|
def test_import_fails(self):
|
|
self.assertRaises(ImportError, safe_exec, 'import os')
|
|
|
|
def test_internal_attributes(self):
|
|
self.assertRaises(SyntaxError, safe_exec, '().__class__.__call__')
|
|
|
|
def test_utils(self):
|
|
_locals = dict(out=None)
|
|
safe_exec('''out = frappe.utils.cint("1")''', None, _locals)
|
|
self.assertEqual(_locals['out'], 1)
|
|
|
|
def test_safe_eval(self):
|
|
self.assertEqual(frappe.safe_eval('1+1'), 2)
|
|
self.assertRaises(AttributeError, frappe.safe_eval, 'frappe.utils.os.path', get_safe_globals())
|
|
|
|
def test_sql(self):
|
|
_locals = dict(out=None)
|
|
safe_exec('''out = frappe.db.sql("select name from tabDocType where name='DocType'")''', None, _locals)
|
|
self.assertEqual(_locals['out'][0][0], 'DocType')
|
|
|
|
self.assertRaises(frappe.PermissionError, safe_exec, 'frappe.db.sql("update tabToDo set description=NULL")') |