소스 검색

[jobsub] Fix running job status flaky test

Romain Rigaux 13 년 전
부모
커밋
4a52276
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      apps/jobsub/src/jobsub/tests.py

+ 2 - 2
apps/jobsub/src/jobsub/tests.py

@@ -257,7 +257,7 @@ class TestJobsubWithHadoop(OozieServerProvider):
         'map_sleep_time': 1,
         'reduce_sleep_time': 1}, follow=True)
 
-    assert_true('PREP' in response.content or 'OK' in response.content, response.content)
+    assert_true(any([status in response.content for status in ('PREP', 'OK', 'DONE')]), response.content)
     assert_true(str(jobid) in response.content)
 
     oozie_job_id = response.context['jobid']
@@ -282,7 +282,7 @@ class TestJobsubWithHadoop(OozieServerProvider):
         'map_sleep_time': 1,
         'reduce_sleep_time': 1}, follow=True)
 
-    assert_true('PREP' in response.content or 'DONE' in response.content, response.content)
+    assert_true(any([status in response.content for status in ('PREP', 'OK', 'DONE')]), response.content)
     assert_true(str(jobid) in response.content)
 
     oozie_job_id = response.context['jobid']