MAPREDUCE-6801. Fix flaky TestKill.testKillJob (Haibo Chen via Varun Saxena)
This commit is contained in:
parent
f6ffa11635
commit
7584fbf4cb
@ -22,6 +22,7 @@
|
||||
import java.util.Map;
|
||||
import java.util.concurrent.CountDownLatch;
|
||||
|
||||
import org.apache.hadoop.service.Service;
|
||||
import org.junit.Assert;
|
||||
|
||||
import org.apache.hadoop.conf.Configuration;
|
||||
@ -66,7 +67,7 @@ public void testKillJob() throws Exception {
|
||||
Job job = app.submit(new Configuration());
|
||||
|
||||
//wait and vailidate for Job to become RUNNING
|
||||
app.waitForState(job, JobState.RUNNING);
|
||||
app.waitForInternalState((JobImpl) job, JobStateInternal.RUNNING);
|
||||
|
||||
//send the kill signal to Job
|
||||
app.getContext().getEventHandler().handle(
|
||||
@ -77,6 +78,10 @@ public void testKillJob() throws Exception {
|
||||
|
||||
//wait and validate for Job to be KILLED
|
||||
app.waitForState(job, JobState.KILLED);
|
||||
// make sure all events are processed. The AM is stopped
|
||||
// only when all tasks and task attempts have been killed
|
||||
app.waitForState(Service.STATE.STOPPED);
|
||||
|
||||
Map<TaskId,Task> tasks = job.getTasks();
|
||||
Assert.assertEquals("No of tasks is not correct", 1,
|
||||
tasks.size());
|
||||
|
Loading…
Reference in New Issue
Block a user