|
@@ -0,0 +1,139 @@
|
|
|
+/**
|
|
|
+ * 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.mapreduce.v2.api.records;
|
|
|
+
|
|
|
+import static org.junit.Assert.assertEquals;
|
|
|
+import static org.junit.Assert.assertFalse;
|
|
|
+import static org.junit.Assert.assertTrue;
|
|
|
+
|
|
|
+import org.apache.hadoop.mapreduce.v2.util.MRBuilderUtils;
|
|
|
+import org.apache.hadoop.yarn.util.BuilderUtils;
|
|
|
+import org.junit.Test;
|
|
|
+
|
|
|
+public class TestIds {
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testJobId() {
|
|
|
+ long ts1 = 1315890136000l;
|
|
|
+ long ts2 = 1315890136001l;
|
|
|
+ JobId j1 = createJobId(ts1, 2);
|
|
|
+ JobId j2 = createJobId(ts1, 1);
|
|
|
+ JobId j3 = createJobId(ts2, 1);
|
|
|
+ JobId j4 = createJobId(ts1, 2);
|
|
|
+
|
|
|
+ assertTrue(j1.equals(j4));
|
|
|
+ assertFalse(j1.equals(j2));
|
|
|
+ assertFalse(j1.equals(j3));
|
|
|
+
|
|
|
+ assertTrue(j1.compareTo(j4) == 0);
|
|
|
+ assertTrue(j1.compareTo(j2) > 0);
|
|
|
+ assertTrue(j1.compareTo(j3) < 0);
|
|
|
+
|
|
|
+ assertTrue(j1.hashCode() == j4.hashCode());
|
|
|
+ assertFalse(j1.hashCode() == j2.hashCode());
|
|
|
+ assertFalse(j1.hashCode() == j3.hashCode());
|
|
|
+
|
|
|
+ JobId j5 = createJobId(ts1, 231415);
|
|
|
+ assertEquals("job_" + ts1 + "_0002", j1.toString());
|
|
|
+ assertEquals("job_" + ts1 + "_231415", j5.toString());
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testTaskId() {
|
|
|
+ long ts1 = 1315890136000l;
|
|
|
+ long ts2 = 1315890136001l;
|
|
|
+ TaskId t1 = createTaskId(ts1, 1, 2, TaskType.MAP);
|
|
|
+ TaskId t2 = createTaskId(ts1, 1, 2, TaskType.REDUCE);
|
|
|
+ TaskId t3 = createTaskId(ts1, 1, 1, TaskType.MAP);
|
|
|
+ TaskId t4 = createTaskId(ts1, 1, 2, TaskType.MAP);
|
|
|
+ TaskId t5 = createTaskId(ts2, 1, 1, TaskType.MAP);
|
|
|
+
|
|
|
+ assertTrue(t1.equals(t4));
|
|
|
+ assertFalse(t1.equals(t2));
|
|
|
+ assertFalse(t1.equals(t3));
|
|
|
+ assertFalse(t1.equals(t5));
|
|
|
+
|
|
|
+ assertTrue(t1.compareTo(t4) == 0);
|
|
|
+ assertTrue(t1.compareTo(t2) < 0);
|
|
|
+ assertTrue(t1.compareTo(t3) > 0);
|
|
|
+ assertTrue(t1.compareTo(t5) < 0);
|
|
|
+
|
|
|
+ assertTrue(t1.hashCode() == t4.hashCode());
|
|
|
+ assertFalse(t1.hashCode() == t2.hashCode());
|
|
|
+ assertFalse(t1.hashCode() == t3.hashCode());
|
|
|
+ assertFalse(t1.hashCode() == t5.hashCode());
|
|
|
+
|
|
|
+ TaskId t6 = createTaskId(ts1, 324151, 54643747, TaskType.REDUCE);
|
|
|
+ assertEquals("task_" + ts1 + "_0001_m_000002", t1.toString());
|
|
|
+ assertEquals("task_" + ts1 + "_324151_r_54643747", t6.toString());
|
|
|
+ }
|
|
|
+
|
|
|
+ @Test
|
|
|
+ public void testTaskAttemptId() {
|
|
|
+ long ts1 = 1315890136000l;
|
|
|
+ long ts2 = 1315890136001l;
|
|
|
+ TaskAttemptId t1 = createTaskAttemptId(ts1, 2, 2, TaskType.MAP, 2);
|
|
|
+ TaskAttemptId t2 = createTaskAttemptId(ts1, 2, 2, TaskType.REDUCE, 2);
|
|
|
+ TaskAttemptId t3 = createTaskAttemptId(ts1, 2, 2, TaskType.MAP, 3);
|
|
|
+ TaskAttemptId t4 = createTaskAttemptId(ts1, 2, 2, TaskType.MAP, 1);
|
|
|
+ TaskAttemptId t5 = createTaskAttemptId(ts1, 2, 1, TaskType.MAP, 3);
|
|
|
+ TaskAttemptId t6 = createTaskAttemptId(ts1, 2, 2, TaskType.MAP, 2);
|
|
|
+
|
|
|
+ assertTrue(t1.equals(t6));
|
|
|
+ assertFalse(t1.equals(t2));
|
|
|
+ assertFalse(t1.equals(t3));
|
|
|
+ assertFalse(t1.equals(t5));
|
|
|
+
|
|
|
+ assertTrue(t1.compareTo(t6) == 0);
|
|
|
+ assertTrue(t1.compareTo(t2) < 0);
|
|
|
+ assertTrue(t1.compareTo(t3) < 0);
|
|
|
+ assertTrue(t1.compareTo(t4) > 0);
|
|
|
+ assertTrue(t1.compareTo(t5) > 0);
|
|
|
+
|
|
|
+ assertTrue(t1.hashCode() == t6.hashCode());
|
|
|
+ assertFalse(t1.hashCode() == t2.hashCode());
|
|
|
+ assertFalse(t1.hashCode() == t3.hashCode());
|
|
|
+ assertFalse(t1.hashCode() == t5.hashCode());
|
|
|
+
|
|
|
+ TaskAttemptId t7 =
|
|
|
+ createTaskAttemptId(ts2, 5463346, 4326575, TaskType.REDUCE, 54375);
|
|
|
+ assertEquals("attempt_" + ts1 + "_0002_m_000002_2", t1.toString());
|
|
|
+ assertEquals("attempt_" + ts2 + "_5463346_r_4326575_54375", t7.toString());
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+ private JobId createJobId(long clusterTimestamp, int idInt) {
|
|
|
+ return MRBuilderUtils.newJobId(
|
|
|
+ BuilderUtils.newApplicationId(clusterTimestamp, idInt), idInt);
|
|
|
+ }
|
|
|
+
|
|
|
+ private TaskId createTaskId(long clusterTimestamp, int jobIdInt,
|
|
|
+ int taskIdInt, TaskType taskType) {
|
|
|
+ return MRBuilderUtils.newTaskId(createJobId(clusterTimestamp, jobIdInt),
|
|
|
+ taskIdInt, taskType);
|
|
|
+ }
|
|
|
+
|
|
|
+ private TaskAttemptId createTaskAttemptId(long clusterTimestamp,
|
|
|
+ int jobIdInt, int taskIdInt, TaskType taskType, int taskAttemptIdInt) {
|
|
|
+ return MRBuilderUtils.newTaskAttemptId(
|
|
|
+ createTaskId(clusterTimestamp, jobIdInt, taskIdInt, taskType),
|
|
|
+ taskAttemptIdInt);
|
|
|
+ }
|
|
|
+}
|