@Test(dependsOnMethods = {"testSetAndGet"})
public void testSerDe() throws IOException {
Closer closer = Closer.create();
try {
ByteArrayOutputStream baos = closer.register(new ByteArrayOutputStream());
DataOutputStream dos = closer.register(new DataOutputStream(baos));
this.taskState.write(dos);
ByteArrayInputStream bais = closer.register((new ByteArrayInputStream(baos.toByteArray())));
DataInputStream dis = closer.register((new DataInputStream(bais)));
TaskState newTaskState = new TaskState();
newTaskState.readFields(dis);
Assert.assertEquals(newTaskState.getJobId(), "Job-1");
Assert.assertEquals(newTaskState.getTaskId(), "Task-1");
Assert.assertEquals(this.taskState.getHighWaterMark(), 2000);
Assert.assertEquals(newTaskState.getStartTime(), this.startTime);
Assert.assertEquals(newTaskState.getEndTime(), this.startTime + 1000);
Assert.assertEquals(newTaskState.getTaskDuration(), 1000);
Assert.assertEquals(newTaskState.getWorkingState(), WorkUnitState.WorkingState.COMMITTED);
Assert.assertEquals(newTaskState.getProp("foo"), "bar");
} catch (Throwable t) {
throw closer.rethrow(t);
} finally {
closer.close();
}
}
