TestOldMethodsJobID.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.hadoop.mapred;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;

import org.apache.hadoop.mapred.TaskCompletionEvent.Status;
import org.apache.hadoop.mapreduce.TaskType;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.Timeout;

import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNull;

/**
 * Test deprecated methods
 *
 */
public class TestOldMethodsJobID {

  /**
   * test deprecated methods of TaskID
   * @throws IOException In the process of unit testing, an IO exception occurred.
   */
  @SuppressWarnings("deprecation")
  @Test
  @Timeout(value = 5)
  public void testDeprecatedMethods() throws IOException {
    JobID jid = new JobID();
    TaskID test = new TaskID(jid, true, 1);
    assertThat(test.getTaskType()).isEqualTo(TaskType.MAP);
    test = new TaskID(jid, false, 1);
    assertThat(test.getTaskType()).isEqualTo(TaskType.REDUCE);

    test = new TaskID("001", 1, false, 1);
    assertThat(test.getTaskType()).isEqualTo(TaskType.REDUCE);
    test = new TaskID("001", 1, true, 1);
    assertThat(test.getTaskType()).isEqualTo(TaskType.MAP);
    ByteArrayOutputStream out = new ByteArrayOutputStream();

    test.write(new DataOutputStream(out));
    TaskID ti = TaskID.read(new DataInputStream(new ByteArrayInputStream(out
        .toByteArray())));
    assertEquals(ti.toString(), test.toString());
    assertEquals("task_001_0001_m_000002",
        TaskID.getTaskIDsPattern("001", 1, true, 2));
    assertEquals("task_003_0001_m_000004",
        TaskID.getTaskIDsPattern("003", 1, TaskType.MAP, 4));
    assertEquals("003_0001_m_000004",
        TaskID.getTaskIDsPatternWOPrefix("003", 1, TaskType.MAP, 4).toString());
   
  }
  
  /**
   * test JobID
   * @throws IOException
   */
  @SuppressWarnings("deprecation")
  @Test
  @Timeout(value = 5)
  public void testJobID() throws IOException{
    JobID jid = new JobID("001",2);
    ByteArrayOutputStream out = new ByteArrayOutputStream();
    jid.write(new DataOutputStream(out));
   assertEquals(jid,JobID.read(new DataInputStream(new ByteArrayInputStream(out.toByteArray()))));
   assertEquals("job_001_0001",JobID.getJobIDsPattern("001",1));
  }
  /**
   * test deprecated methods of TaskCompletionEvent
   */
  @SuppressWarnings("deprecation")
  @Test
  @Timeout(value = 5)
  public void testTaskCompletionEvent() {
    TaskAttemptID taid = new TaskAttemptID("001", 1, TaskType.REDUCE, 2, 3);
    TaskCompletionEvent template = new TaskCompletionEvent(12, taid, 13, true,
        Status.SUCCEEDED, "httptracker");
    TaskCompletionEvent testEl = TaskCompletionEvent.downgrade(template);
    testEl.setTaskAttemptId(taid);
    testEl.setTaskTrackerHttp("httpTracker");

    testEl.setTaskId("attempt_001_0001_m_000002_04");
    assertEquals("attempt_001_0001_m_000002_4",testEl.getTaskId());

    testEl.setTaskStatus(Status.OBSOLETE);
    assertEquals(Status.OBSOLETE.toString(), testEl.getStatus().toString());

    testEl.setTaskRunTime(20);
    assertThat(testEl.getTaskRunTime()).isEqualTo(20);
    testEl.setEventId(16);
    assertThat(testEl.getEventId()).isEqualTo(16);

  }

  /**
   * test depricated methods of JobProfile
   * @throws IOException
   */
  @SuppressWarnings("deprecation")
  @Test
  @Timeout(value = 5)
  public void testJobProfile() throws IOException {

    JobProfile profile = new JobProfile("user", "job_001_03", "jobFile", "uri",
        "name");
    assertEquals("job_001_0003", profile.getJobId());
    assertEquals("default", profile.getQueueName());
   // serialization test
    ByteArrayOutputStream out = new ByteArrayOutputStream();
    profile.write(new DataOutputStream(out));

    JobProfile profile2 = new JobProfile();
    profile2.readFields(new DataInputStream(new ByteArrayInputStream(out
        .toByteArray())));

    assertEquals(profile2.name, profile.name);
    assertEquals(profile2.jobFile, profile.jobFile);
    assertEquals(profile2.queueName, profile.queueName);
    assertEquals(profile2.url, profile.url);
    assertEquals(profile2.user, profile.user);
  }
  /**
   * test TaskAttemptID 
   */
  @SuppressWarnings( "deprecation" )
  @Test
  @Timeout(value = 5)
  public void testTaskAttemptID (){
    TaskAttemptID task  = new TaskAttemptID("001",2,true,3,4);
    assertEquals("attempt_001_0002_m_000003_4", TaskAttemptID.getTaskAttemptIDsPattern("001", 2, true, 3, 4));
    assertEquals("task_001_0002_m_000003", task.getTaskID().toString());
    assertEquals("attempt_001_0001_r_000002_3",TaskAttemptID.getTaskAttemptIDsPattern("001", 1, TaskType.REDUCE, 2, 3));
    assertEquals("001_0001_m_000001_2", TaskAttemptID.getTaskAttemptIDsPatternWOPrefix("001",1, TaskType.MAP, 1, 2).toString());
    
  }
  
  /**
   * test Reporter.NULL
   *
   */
  
  @Test
  @Timeout(value = 5)
  public void testReporter(){
    Reporter nullReporter=Reporter.NULL;
    assertNull(nullReporter.getCounter(null));
    assertNull(nullReporter.getCounter("group", "name"));
    // getInputSplit method removed
    try{
      assertNull(nullReporter.getInputSplit());
    }catch(UnsupportedOperationException e){
      assertEquals( "NULL reporter has no input",e.getMessage());
    }
    assertEquals(0,nullReporter.getProgress(),0.01);

  }
}