def test_should_exit_when_invoked_as_script(self): args = [sys.executable, MRJob.mr_job_script(), '--quux', 'baz'] # add . to PYTHONPATH (in case mrjob isn't actually installed) env = combine_envs(os.environ, {'PYTHONPATH': os.path.abspath('.')}) proc = Popen(args, stderr=PIPE, stdout=PIPE, env=env) proc.communicate() self.assertEqual(proc.returncode, 2)
def test_should_exit_when_invoked_as_script(self): args = [sys.executable, MRJob.mr_job_script(), '--quux', 'baz'] # add . to PYTHONPATH (in case mrjob isn't actually installed) env = combine_envs(os.environ, {'PYTHONPATH': os.path.abspath('.')}) proc = Popen(args, stderr=PIPE, stdout=PIPE, env=env) proc.communicate() assert_equal(proc.returncode, 2)
def test_libjars_attr_relative_path(self): job_dir = os.path.dirname(MRJob.mr_job_script()) with patch.object(MRJob, 'LIBJARS', ['cookie.jar', '/left/dora.jar']): job = MRJob() self.assertEqual( job._runner_kwargs()['libjars'], [os.path.join(job_dir, 'cookie.jar'), '/left/dora.jar'])
def test_libjars_attr_relative_path(self): job_dir = os.path.dirname(MRJob.mr_job_script()) with patch.object(MRJob, "LIBJARS", ["cookie.jar", "/left/dora.jar"]): job = MRJob() self.assertEqual( job.job_runner_kwargs()["libjars"], [os.path.join(job_dir, "cookie.jar"), "/left/dora.jar"] )
def test_libjars_environment_variables(self): job_dir = os.path.dirname(MRJob.mr_job_script()) with patch.dict("os.environ", A="/path/to/a", B="b"): with patch.object(MRJob, "LIBJARS", ["$A/cookie.jar", "$B/honey.jar"]): job = MRJob() # libjars() peeks into envvars to figure out if the path # is relative or absolute self.assertEqual( job.job_runner_kwargs()["libjars"], ["$A/cookie.jar", os.path.join(job_dir, "$B/honey.jar")] )
def test_libjars_environment_variables(self): job_dir = os.path.dirname(MRJob.mr_job_script()) with patch.dict('os.environ', A='/path/to/a', B='b'): with patch.object(MRJob, 'LIBJARS', ['$A/cookie.jar', '$B/honey.jar']): job = MRJob() # libjars() peeks into envvars to figure out if the path # is relative or absolute self.assertEqual( job._runner_kwargs()['libjars'], ['$A/cookie.jar', os.path.join(job_dir, '$B/honey.jar')])
def test_should_exit_when_invoked_as_script(self): args = ['python', MRJob.mr_job_script(), '--quux', 'baz'] proc = Popen(args, stderr=PIPE, stdout=PIPE) stderr, stdout = proc.communicate() assert_equal(proc.returncode, 2)