def test_cleanup(Popen):
"""Test that the spidermonkey processes are terminated during cleanup."""
process = mock.MagicMock()
Popen.side_effect = patch_init(process)
# Make sure we don't already have a cached shell.
JSShell.cleanup()
assert JSShell.instance is None
assert not Popen.called
process.stdout.readline.return_value = '{}\n'
get_tree('hello();')
assert Popen.call_count == 1
# Clear the reference to `self` in the call history so it can be reaped.
Popen.reset_mock()
assert JSShell.instance
assert not process.terminate.called
decorator.cleanup()
assert process.terminate.called
assert JSShell.instance is None
def test_cleanup(Popen):
"""Test that the spidermonkey processes are terminated during cleanup."""
process = mock.MagicMock()
Popen.side_effect = patch_init(process)
# Make sure we don't already have a cached shell.
JSShell.cleanup()
assert JSShell.instance is None
assert not Popen.called
process.stdout.readline.return_value = '{}\n'
get_tree('hello();')
eq_(Popen.call_count, 1)
# Clear the reference to `self` in the call history so it can be reaped.
Popen.reset_mock()
assert JSShell.instance
assert not process.terminate.called
decorator.cleanup()
assert process.terminate.called
assert JSShell.instance is None
def test_js_file(err,
filename,
data,
line=1,
column=0,
context=None,
pollutable=False):
'Test a JS file by parsing and analyzing its tokens.'
if err.detected_type == PACKAGE_THEME:
err.warning(err_id=('testcases_scripting', 'test_js_file', 'theme_js'),
warning='JS run from full theme',
description='Themes should not contain executable code.',
filename=filename,
line=line)
before_tier = None
# Set the tier to 4 (Security Tests)
if err is not None:
before_tier = err.tier
err.set_tier(3)
tree = get_tree(data, filename=filename, err=err)
if not tree:
if before_tier:
err.set_tier(before_tier)
return
# Generate a context if one is not available.
if context is None:
context = ContextGenerator(data)
t = traverser.Traverser(err,
filename,
start_line=line,
start_column=column,
context=context,
pollutable=pollutable,
is_jsm=(filename.endswith('.jsm')
or 'EXPORTED_SYMBOLS' in data))
t.run(tree)
# Reset the tier so we don't break the world
if err is not None:
err.set_tier(before_tier)
def test_cleanup_on_error(Popen):
"""Test that the cached shells are removed on IO error."""
process = mock.MagicMock()
process.stdout.readline.return_value = '{}'
Popen.side_effect = patch_init(process)
# Make sure we don't already have a cached shell.
JSShell.cleanup()
assert JSShell.instance is None
# Success, should have cached shell.
get_tree('hello();')
assert Popen.call_count == 1
assert JSShell.instance
assert not process.terminate.called
# Write error. No cached shell.
process.stdout.readline.reset_mock()
process.stdin.write.side_effect = IOError
with pytest.raises(IOError):
get_tree('hello();')
# No new processes created, since we had a cached shell from last time.
assert Popen.call_count == 1
assert process.stdin.write.called
assert not process.stdout.readline.called
# Read error. No cached shell.
process.stdin.write.reset_mock()
process.stdin.write.side_effect = None
process.stdout.readline.side_effect = IOError
with pytest.raises(IOError):
get_tree('hello();')
assert JSShell.instance is None
# Cached shell should have been removed in the last run. New
# process should be created.
assert Popen.call_count == 2
assert process.stdin.write.called
assert process.stdout.readline.called
assert JSShell.instance is None
def test_cleanup_on_error(Popen):
"""Test that the cached shells are removed on IO error."""
process = mock.MagicMock()
process.stdout.readline.return_value = '{}'
Popen.side_effect = patch_init(process)
# Make sure we don't already have a cached shell.
JSShell.cleanup()
assert JSShell.instance is None
# Success, should have cached shell.
get_tree('hello();')
eq_(Popen.call_count, 1)
assert JSShell.instance
assert not process.terminate.called
# Write error. No cached shell.
process.stdout.readline.reset_mock()
process.stdin.write.side_effect = IOError
with assert_raises(IOError):
get_tree('hello();')
# No new processes created, since we had a cached shell from last time.
eq_(Popen.call_count, 1)
assert process.stdin.write.called
assert not process.stdout.readline.called
# Read error. No cached shell.
process.stdin.write.reset_mock()
process.stdin.write.side_effect = None
process.stdout.readline.side_effect = IOError
with assert_raises(IOError):
get_tree('hello();')
assert JSShell.instance is None
# Cached shell should have been removed in the last run. New
# process should be created.
eq_(Popen.call_count, 2)
assert process.stdin.write.called
assert process.stdout.readline.called
assert JSShell.instance is None
def test_js_file(err, filename, data, line=1, column=0, context=None,
pollutable=False):
'Test a JS file by parsing and analyzing its tokens.'
if err.detected_type == PACKAGE_THEME:
err.warning(
err_id=('testcases_scripting',
'test_js_file',
'theme_js'),
warning='JS run from full theme',
description='Themes should not contain executable code.',
filename=filename,
line=line)
before_tier = None
# Set the tier to 4 (Security Tests)
if err is not None:
before_tier = err.tier
err.set_tier(3)
tree = get_tree(data, filename=filename, err=err)
if not tree:
if before_tier:
err.set_tier(before_tier)
return
# Generate a context if one is not available.
if context is None:
context = ContextGenerator(data)
t = traverser.Traverser(err, filename, start_line=line,
start_column=column, context=context,
pollutable=pollutable,
is_jsm=(filename.endswith('.jsm') or
'EXPORTED_SYMBOLS' in data))
t.run(tree)
# Reset the tier so we don't break the world
if err is not None:
err.set_tier(before_tier)
