def test_run_invalid_input(self): program = Program() program.transport = mock.create_autospec(Transport) program.parser = mock.create_autospec(Parser) program.interpolator = mock.create_autospec(Interpolator) program.transport.read.return_value = 'input' program.parser.parse.side_effect = ValueError('Error') program.interpolator.interpolate.return_value = 'output' actual = program.run() program.transport.read.assert_called() program.parser.parse.assert_called() program.parser.parse.assert_called_with('input') program.interpolator.interpolate.assert_not_called() program.transport.write.assert_called_with('Invalid matrix!')
def test_run(self): program = Program() program.transport = mock.create_autospec(Transport) program.parser = mock.create_autospec(Parser) program.interpolator = mock.create_autospec(Interpolator) program.transport.read.return_value = 'input' program.parser.parse.return_value = 'parsed_input' program.interpolator.interpolate.return_value = 'output' actual = program.run() program.transport.read.assert_called() program.parser.parse.assert_called() program.parser.parse.assert_called_with('input') program.interpolator.interpolate.assert_called_with('parsed_input') program.transport.write.assert_called_with('output')