Skip to content

Commit

Permalink
MAPREDUCE-7363. Rename JobClientUnitTest to TestJobClients (#3487)
Browse files Browse the repository at this point in the history
  • Loading branch information
dongjoon-hyun committed Sep 28, 2021
1 parent 1f8f53f commit f5148ca
Showing 1 changed file with 18 additions and 18 deletions.
Expand Up @@ -44,8 +44,8 @@
import org.junit.Test;

@SuppressWarnings("deprecation")
public class JobClientUnitTest {
public class TestJobClients {

public class TestJobClient extends JobClient {

TestJobClient(JobConf jobConf) throws IOException {
Expand Down Expand Up @@ -99,57 +99,57 @@ public void testMapTaskReportsWithNullJob() throws Exception {
Cluster mockCluster = mock(Cluster.class);
client.setCluster(mockCluster);
JobID id = new JobID("test",0);

when(mockCluster.getJob(id)).thenReturn(null);

TaskReport[] result = client.getMapTaskReports(id);
assertEquals(0, result.length);

verify(mockCluster).getJob(id);
}

@Test
public void testReduceTaskReportsWithNullJob() throws Exception {
TestJobClient client = new TestJobClient(new JobConf());
Cluster mockCluster = mock(Cluster.class);
client.setCluster(mockCluster);
JobID id = new JobID("test",0);

when(mockCluster.getJob(id)).thenReturn(null);

TaskReport[] result = client.getReduceTaskReports(id);
assertEquals(0, result.length);

verify(mockCluster).getJob(id);
}

@Test
public void testSetupTaskReportsWithNullJob() throws Exception {
TestJobClient client = new TestJobClient(new JobConf());
Cluster mockCluster = mock(Cluster.class);
client.setCluster(mockCluster);
JobID id = new JobID("test",0);

when(mockCluster.getJob(id)).thenReturn(null);

TaskReport[] result = client.getSetupTaskReports(id);
assertEquals(0, result.length);

verify(mockCluster).getJob(id);
}

@Test
public void testCleanupTaskReportsWithNullJob() throws Exception {
TestJobClient client = new TestJobClient(new JobConf());
Cluster mockCluster = mock(Cluster.class);
client.setCluster(mockCluster);
JobID id = new JobID("test",0);

when(mockCluster.getJob(id)).thenReturn(null);

TaskReport[] result = client.getCleanupTaskReports(id);
assertEquals(0, result.length);

verify(mockCluster).getJob(id);
}

Expand Down Expand Up @@ -184,7 +184,7 @@ public void testShowJob() throws Exception {
when(mockCluster.getJob(jobID)).thenReturn(mockJob);

client.setCluster(mockCluster);

ByteArrayOutputStream out = new ByteArrayOutputStream();
client.displayJobList(new JobStatus[] {mockJobStatus}, new PrintWriter(out));
String commandLineOutput = out.toString();
Expand Down

0 comments on commit f5148ca

Please sign in to comment.