def test_main_multiple_files(self, process_results_mock, read_subunit_mock): tfile1 = tempfile.NamedTemporaryFile() tfile2 = tempfile.NamedTemporaryFile() tfile1.write(b'test me later 1') tfile2.write(b'test me later 2') tfile1.flush() tfile2.flush() self.fake_args.extend([tfile1.name, tfile2.name]) fake_read_subunit = mock.MagicMock('ReadSubunit') fake_get_results_1 = 'fake results 1' fake_get_results_2 = 'fake results 2' fake_read_subunit.get_results = mock.MagicMock('get_results') fake_read_subunit.get_results.side_effect = [ fake_get_results_1, fake_get_results_2 ] read_subunit_mock.return_value = fake_read_subunit shell.main() read_subunit_mock.assert_called_with(mock.ANY, attachments=False, attr_regex='\[(.*)\]', non_subunit_name=None, targets=[], use_wall_time=False) self.assertEqual(2, len(read_subunit_mock.call_args_list)) file_1 = read_subunit_mock.call_args_list[0][0][0] file_1.seek(0) self.assertEqual('test me later 1', file_1.read()) file_2 = read_subunit_mock.call_args_list[1][0][0] file_2.seek(0) self.assertEqual('test me later 2', file_2.read()) self.assertEqual(fake_get_results_1, process_results_mock.call_args_list[0][0][0]) self.assertEqual(fake_get_results_2, process_results_mock.call_args_list[1][0][0])
def test_main_multiple_files(self, process_results_mock, read_subunit_mock): tfile1 = tempfile.NamedTemporaryFile() tfile2 = tempfile.NamedTemporaryFile() tfile1.write(b'test me later 1') tfile2.write(b'test me later 2') tfile1.flush() tfile2.flush() self.fake_args.extend([tfile1.name, tfile2.name]) fake_read_subunit = mock.MagicMock('ReadSubunit') fake_get_results_1 = 'fake results 1' fake_get_results_2 = 'fake results 2' fake_read_subunit.get_results = mock.MagicMock('get_results') fake_read_subunit.get_results.side_effect = [fake_get_results_1, fake_get_results_2] read_subunit_mock.return_value = fake_read_subunit shell.main() read_subunit_mock.assert_called_with(mock.ANY, attachments=False, attr_regex='\[(.*)\]', non_subunit_name=None, targets=[], use_wall_time=False) self.assertEqual(2, len(read_subunit_mock.call_args_list)) file_1 = read_subunit_mock.call_args_list[0][0][0] file_1.seek(0) self.assertEqual('test me later 1', file_1.read()) file_2 = read_subunit_mock.call_args_list[1][0][0] file_2.seek(0) self.assertEqual('test me later 2', file_2.read()) self.assertEqual(fake_get_results_1, process_results_mock.call_args_list[0][0][0]) self.assertEqual(fake_get_results_2, process_results_mock.call_args_list[1][0][0])
def test_main(self, process_results_mock, read_subunit_mock): fake_read_subunit = mock.MagicMock('ReadSubunit') fake_get_results = 'fake results' fake_read_subunit.get_results = mock.MagicMock('get_results') fake_read_subunit.get_results.return_value = fake_get_results read_subunit_mock.return_value = fake_read_subunit shell.main() read_subunit_mock.assert_called_once_with(sys.stdin, attachments=False, attr_regex='\[(.*)\]', targets=[]) process_results_mock.assert_called_once_with(fake_get_results)
def test_main_with_targets(self, process_results_mock, read_subunit_mock, ext_mock): exts = mock.MagicMock('EnabledExtensionManager()') ext_mock.return_value = exts exts.map = mock.MagicMock('extensions.map') exts.map.return_value = [mock.sentinel.extension] fake_read_subunit = mock.MagicMock('ReadSubunit') fake_get_results = 'fake results' fake_read_subunit.get_results = mock.MagicMock('get_results') fake_read_subunit.get_results.return_value = fake_get_results read_subunit_mock.return_value = fake_read_subunit shell.main() read_subunit_mock.assert_called_once_with( sys.stdin, attachments=False, attr_regex='\[(.*)\]', targets=[mock.sentinel.extension]) process_results_mock.assert_called_once_with(fake_get_results)