class TestStepRunner(unittest.TestCase):
def setUp(self):
self.test_server = TestServer()
self.test_server.start()
self._run_helloworld()
r = requests.get(self.test_server.server_url+'/api/step_runs/')
self.step1_run_id = r.json()['step_runs'][0].get('_id')
self.file_root = tempfile.mkdtemp()
parser = StepRunner._get_parser()
args = parser.parse_args(['--run_id', self.step1_run_id, '--master_url', self.test_server.server_url])
self.step_runner = StepRunner(args=args)
self.step_runner.settings['WORKING_DIR'] = self.file_root
def tearDown(self):
shutil.rmtree(self.file_root)
self.test_server.stop()
def _run_helloworld(self):
url = self.test_server.server_url+'/api/submitrequest/'
response = requests.post(url, data=fixtures.helloworld_json)
self.assertEqual(response.status_code, 201, 'Expected 201 but got %d trying to post to %s' % (response.status_code, url))
self.test_server.dry_run_job_queues()
# Given steprun ID, retrieve the steprun
def test_get_step_run(self):
step_run = self.step_runner.step_run
self.assertEqual(step_run.get('_id'), self.step1_run_id)
def test_execute(self):
process = self.step_runner._execute()
self.step_runner._wait_for_process(process)
def setUp(self):
self.test_server = TestServer()
self.test_server.start()
self._run_helloworld()
r = requests.get(self.test_server.server_url+'/api/step_runs/')
self.step1_run_id = r.json()['step_runs'][0].get('_id')
self.file_root = tempfile.mkdtemp()
parser = StepRunner._get_parser()
args = parser.parse_args(['--run_id', self.step1_run_id, '--master_url', self.test_server.server_url])
self.step_runner = StepRunner(args=args)
self.step_runner.settings['WORKING_DIR'] = self.file_root
