def test_availability_at_first_attempt(self, sleep_mock, failed_mock):
SSH.await_availability('')
self.assertEqual(1, failed_mock.call_count)
self.assertFalse(sleep_mock.called)
def test_availability_at_second_attempt(self, sleep_mock, failed_mock):
SSH.await_availability('')
self.assertEqual(2, failed_mock.call_count)
self.assertTrue(sleep_mock.called)
def wait_ssh(name, states):
wait_state(name, 'Running', states)
SSH.await_availability(name)
#!/usr/bin/env python3
from expyrimenter import SSH, Executor
# Suppose we have a cluster with hostnames vm0, vm1 and vm2
cluster = ['vm%d' % i for i in range(0, 3)]
# Let's run the command below in all VMs:
cmd = 'echo "$(date +%S) Hello by $(hostname)"'
# Blocking version, serial execution
print('Serial execution\n================')
for vm in cluster:
output = SSH(vm, cmd, stdout=True).run()
print(output)
# Create a pool for parallel execution
pool = Executor()
# Non-blocking version, parallel execution
print('\nParallel execution\n==================')
for vm in cluster:
ssh = SSH(vm, cmd, stdout=True, stderr=False)
pool.run(ssh)
# Block until all parallel calls are done
# and then print the results
pool.wait()
for result in pool.results:
print(result)
def wait_ssh(name, states):
wait_state(name, 'Running', states)
SSH.await_availability(name)
