def test_two_nodes(self):
self.add_worker()
self.ssh_worker_hosts.return_value = ['testworker0']
self.runner._opts['num_core_instances'] = 1
mock_ssh_file('testmaster', 'some_file', b'file contents 1')
mock_ssh_file('testmaster!testworker0', 'some_file',
b'file contents 2')
self.runner.fs # force initialization of _ssh_fs
_run_on_all_nodes(self.runner, self.output_dir, ['cat', 'some_file'],
print_stderr=False)
with open(os.path.join(self.output_dir, 'master', 'stdout'), 'r') as f:
self.assertEqual(f.read().rstrip(), 'file contents 1')
with open(
os.path.join(self.output_dir, 'worker testworker0', 'stdout'),
'r') as f:
self.assertEqual(f.read().strip(), 'file contents 2')
self.assertEqual(sorted(os.listdir(self.output_dir)),
['master', 'worker testworker0'])
def test_two_nodes(self):
self.add_worker()
self.ssh_worker_hosts.return_value = ['testworker0']
self.runner._opts['num_core_instances'] = 1
mock_ssh_file('testmaster', 'some_file', b'file contents 1')
mock_ssh_file('testmaster!testworker0', 'some_file',
b'file contents 2')
self.runner.fs # force initialization of _ssh_fs
_run_on_all_nodes(self.runner,
self.output_dir, ['cat', 'some_file'],
print_stderr=False)
with open(os.path.join(self.output_dir, 'master', 'stdout'), 'r') as f:
self.assertEqual(f.read().rstrip(), 'file contents 1')
with open(
os.path.join(self.output_dir, 'worker testworker0', 'stdout'),
'r') as f:
self.assertEqual(f.read().strip(), 'file contents 2')
self.assertEqual(sorted(os.listdir(self.output_dir)),
['master', 'worker testworker0'])
def test_one_node(self):
mock_ssh_file('testmaster', 'some_file', b'file contents')
_run_on_all_nodes(self.runner, self.output_dir, ['cat', 'some_file'],
print_stderr=False)
with open(os.path.join(self.output_dir, 'master', 'stdout'), 'r') as f:
self.assertEqual(f.read().rstrip(), 'file contents')
self.assertEqual(os.listdir(self.output_dir), ['master'])
def test_one_node(self):
mock_ssh_file('testmaster', 'some_file', b'file contents')
_run_on_all_nodes(self.runner, self.output_dir, ['cat', 'some_file'],
print_stderr=False)
with open(os.path.join(self.output_dir, 'master', 'stdout'), 'r') as f:
self.assertEqual(f.read().rstrip(), 'file contents')
self.assertEqual(os.listdir(self.output_dir), ['master'])
