def test_if_a_variable_not_declared_before_should_raise_error(self):
console = Console()
console.assignment_manager.clear()
console.codes.add(Block("b = 2"))
with self.assertRaises(NotDeclaredError):
console.prepare()
def test_call_export_when_input_export_command(self):
console = Console()
console.do_export = mock.MagicMock()
code = "export test"
console._run(code)
console.do_export.assert_called_once_with("test")
def test_handle_function_when_code_was_start_with_func(self):
console = Console()
console.custom_methods = mock.MagicMock()
block = Block("func test()")
console.process(block)
console.custom_methods.add.assert_called_once_with(block)
def test_console_import_code_with_single_import(self):
code = 'import "yyx"'
console = Console()
console._run(code)
self.assertEqual(
len(console.packages.get_declared()), 2
)
self.assertTrue(
"yyx" in console.packages.get_declared()
)
def test_could_not_import_same_package_multi_times(self):
code = 'import "fmt"'
console = Console()
console.packages.assignment_manager.get_all_declared().clear()
console._run(code)
console._run(code)
self.assertEqual(
len(console.packages.get_declared()), 1
)
self.assertTrue(
"fmt" in console.packages.get_declared()
)
def test_give_a_direct_command_would_invoke_direct_method(self):
console = Console()
console.direct_command = mock.MagicMock()
console._write_to_file = mock.MagicMock()
console.execute = mock.MagicMock()
code = "12 + 34"
console.try_run_direct_command(code)
console.direct_command.assert_called_once_with(code)
def test_invoke_sys_exit_when_given_code_exit(self, mock_exit):
console = Console()
console._run('exit')
mock_exit.assert_called_once_with(0)
