def test_fatal_error_catch(self, mock_err, mock_connect): code = a2b.main(['1001.1001']) self.assertEqual(code, 2) self.assertEqual(mock_err.getvalue().strip(), 'xxx')
def test_main_function(self, mock_out, mock_err): code = a2b.main(['1001.1001', 'x']) self.assertEqual(code, 1)