/*
* 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.flink.test.hadoopcompatibility.mapred.record;
import org.apache.flink.api.common.Plan;
import org.apache.flink.hadoopcompatibility.mapred.record.example.WordCountWithOutputFormat;
import org.apache.flink.test.testdata.WordCountData;
import org.apache.flink.test.util.RecordAPITestBase;
/**
* test the hadoop inputformat and outputformat
*/
public class HadoopRecordInputOutputITCase extends RecordAPITestBase {
protected String textPath;
protected String resultPath;
protected String counts;
@Override
protected void preSubmit() throws Exception {
textPath = createTempFile("text.txt", WordCountData.TEXT);
resultPath = getTempDirPath("result");
counts = WordCountData.COUNTS.replaceAll(" ", "\t");
}
@Override
protected Plan getTestJob() {
//WordCountWithHadoopOutputFormat takes hadoop TextInputFormat as input and output file in hadoop TextOutputFormat
WordCountWithOutputFormat wc = new WordCountWithOutputFormat();
return wc.getPlan("1", textPath, resultPath);
}
@Override
protected void postSubmit() throws Exception {
// Test results, append /1 to resultPath due to the generated _temproray file.
compareResultsByLinesInMemory(counts, resultPath + "/1");
}
}