def test_cmdenv(self):
import logging
logging.basicConfig()
# make sure previous environment is preserved
os.environ['SOMETHING'] = 'foofoofoo'
old_env = os.environ.copy()
mr_job = MRTestCmdenv(['--runner', 'inline', '--cmdenv=FOO=bar'])
mr_job.sandbox(stdin=BytesIO(b'foo\n'))
results = []
with mr_job.make_runner() as runner:
assert isinstance(runner, InlineMRJobRunner)
runner.run()
results.extend(mr_job.parse_output(runner.cat_output()))
self.assertEqual(sorted(results), [('FOO', 'bar'),
('SOMETHING', 'foofoofoo')])
# make sure we revert back
self.assertEqual(old_env, os.environ)
def test_cmdenv(self):
import logging
logging.basicConfig()
# make sure previous environment is preserved
os.environ['SOMETHING'] = 'foofoofoo'
old_env = os.environ.copy()
mr_job = MRTestCmdenv(['--runner', 'inline', '--cmdenv=FOO=bar'])
mr_job.sandbox(stdin=BytesIO(b'foo\n'))
results = []
with mr_job.make_runner() as runner:
assert isinstance(runner, InlineMRJobRunner)
runner.run()
results.extend(mr_job.parse_output(runner.cat_output()))
self.assertEqual(sorted(results),
[('FOO', 'bar'), ('SOMETHING', 'foofoofoo')])
# make sure we revert back
self.assertEqual(old_env, os.environ)
