浏览代码

[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']