def test_setup_command(self):
job = MROSWalkJob([
'-r', 'spark', '--spark-master', _LOCAL_CLUSTER_MASTER, '--setup',
'touch bar'
])
job.sandbox()
with job.make_runner() as r:
r.run()
path_to_size = dict(job.parse_output(r.cat_output()))
self.assertIn('./bar', path_to_size)
def test_setup_command(self):
job = MROSWalkJob(
['-r', 'spark',
'--spark-master', _LOCAL_CLUSTER_MASTER,
'--setup', 'touch bar'])
job.sandbox()
with job.make_runner() as r:
r.run()
path_to_size = dict(job.parse_output(r.cat_output()))
self.assertIn('./bar', path_to_size)
def test_file_uris(self):
f1_path = self.makefile('f1', b'contents')
f2_uri = 'file://' + self.makefile('f2', b'stuff')
job = MROSWalkJob(['--files', '%s,%s' % (f1_path, f2_uri)])
job.sandbox()
with job.make_runner() as runner:
runner.run()
path_to_size = dict(job.parse_output(runner.cat_output()))
self.assertEqual(path_to_size.get('./f1'), 8)
self.assertEqual(path_to_size.get('./f2'), 5)
def test_archive_uris(self):
qux_dir = self.makedirs('qux')
self.makefile(join(qux_dir, 'bar'), b'baz')
qux_tar_gz = make_archive(join(self.tmp_dir, 'qux'), 'gztar', qux_dir)
qux_tar_gz_uri = 'file://' + qux_tar_gz
job = MROSWalkJob(
['--archives',
'%s#qux,%s#qux2' % (qux_tar_gz, qux_tar_gz_uri)])
job.sandbox()
with job.make_runner() as runner:
runner.run()
path_to_size = dict(job.parse_output(runner.cat_output()))
self.assertEqual(path_to_size.get('./qux/bar'), 3)
self.assertEqual(path_to_size.get('./qux2/bar'), 3)
