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)