def test_no_output(self): launcher = MRJobLauncher(args=['--no-conf', '--no-output', '']) launcher.sandbox() with patch.object(launcher, 'make_runner') as m_make_runner: runner = RunnerStub() _mock_context_mgr(m_make_runner, runner) runner.stream_output.return_value = ['a line'] launcher.run_job() self.assertEqual(launcher.stdout.getvalue(), '') self.assertEqual(launcher.stderr.getvalue(), '')
def test_no_output(self): launcher = MRJobLauncher(args=['--no-conf', '--no-output', '']) launcher.sandbox() with patch.object(launcher, 'make_runner') as m_make_runner: runner = Mock() _mock_context_mgr(m_make_runner, runner) runner.stream_output.return_value = ['a line'] launcher.run_job() self.assertEqual(launcher.stdout.getvalue(), b'') self.assertEqual(launcher.stderr.getvalue(), b'')
def test_no_output(self): launcher = MRJobLauncher(args=["--no-conf", "--no-output", ""]) launcher.sandbox() with patch.object(launcher, "make_runner") as m_make_runner: runner = Mock() _mock_context_mgr(m_make_runner, runner) runner.stream_output.return_value = ["a line"] launcher.run_job() self.assertEqual(launcher.stdout.getvalue(), "") self.assertEqual(launcher.stderr.getvalue(), "")
def test_no_file_args_required(self): words1 = self.makefile('words1', b'kit and caboodle\n') words2 = self.makefile('words2', b'baubles\nbangles and beads\n') job = MRJobLauncher( args=['-r', 'local', tests.sr_wc.__file__, words1, words2]) job.sandbox() with job.make_runner() as runner: runner.run() lines = list(to_lines(runner.cat_output())) self.assertEqual(len(lines), 1) self.assertEqual(int(lines[0]), 7)
def _make_launcher(self, *args): """Make a launcher, add a mock runner (``launcher.mock_runner``), and set it up so that ``launcher.make_runner().__enter__()`` returns ``launcher.mock_runner()``. """ launcher = MRJobLauncher(args=['--no-conf', ''] + list(args)) launcher.sandbox() launcher.mock_runner = Mock() launcher.mock_runner.stream_output.return_value = [b'a line\n'] launcher.make_runner = MagicMock() # include __enter__ launcher.make_runner.return_value.__enter__.return_value = ( launcher.mock_runner) return launcher