Homec4science
Diffusion warcbase (master) Browse

R1473/warcbase-core/src/test/java/org/warcbase/io/ArcRecordWritableTest.java
62b8c4e57800master

Last Change
Raw File

ArcRecordWritableTest.java
View Options

/*
* Warcbase: an open-source platform for managing web archives
*
* Licensed 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.warcbase.io;
import static org.junit.Assert.assertEquals;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.Path;
import org.apache.hadoop.io.LongWritable;
import org.apache.hadoop.mapreduce.InputFormat;
import org.apache.hadoop.mapreduce.RecordReader;
import org.apache.hadoop.mapreduce.TaskAttemptContext;
import org.apache.hadoop.mapreduce.TaskAttemptID;
import org.apache.hadoop.mapreduce.lib.input.FileSplit;
import org.apache.hadoop.mapreduce.task.TaskAttemptContextImpl;
import org.apache.hadoop.util.ReflectionUtils;
import org.junit.Test;
import org.warcbase.mapreduce.WacArcInputFormat;
import com.google.common.io.Resources;
public class ArcRecordWritableTest {
@Test
public void testInputFormat() throws Exception {
String arcFile = Resources.getResource("arc/example.arc.gz").getPath();
Configuration conf = new Configuration(false);
conf.set("fs.defaultFS", "file:///");
File testFile = new File(arcFile);
Path path = new Path(testFile.getAbsoluteFile().toURI());
FileSplit split = new FileSplit(path, 0, testFile.length(), null);
InputFormat<LongWritable, ArcRecordWritable> inputFormat = ReflectionUtils.newInstance(
WacArcInputFormat.class, conf);
TaskAttemptContext context = new TaskAttemptContextImpl(conf, new TaskAttemptID());
RecordReader<LongWritable, ArcRecordWritable> reader = inputFormat.createRecordReader(split,
context);
reader.initialize(split, context);
int cnt = 0;
while (reader.nextKeyValue()) {
ArcRecordWritable record = reader.getCurrentValue();
cnt++;
ByteArrayOutputStream bytesOut = new ByteArrayOutputStream();
DataOutputStream dataOut = new DataOutputStream(bytesOut);
record.write(dataOut);
ArcRecordWritable reconstructed = new ArcRecordWritable();
reconstructed.readFields(new DataInputStream(new ByteArrayInputStream(bytesOut.toByteArray())));
assertEquals(record.getRecord().getMetaData().getUrl(),
reconstructed.getRecord().getMetaData().getUrl());
}
assertEquals(300, cnt);
}
}
c4science · Help