/**
* 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.jena.tdbloader4;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import java.io.IOException;
import java.util.Arrays;
import java.util.Collection;
import org.apache.hadoop.util.ToolRunner;
import org.apache.jena.tdbloader4.Constants;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.junit.runners.Parameterized.Parameters;
import org.openjena.atlas.lib.FileOps;
import cmd.tdbloader4;
import com.hp.hpl.jena.tdb.base.file.Location;
import com.hp.hpl.jena.tdb.store.DatasetGraphTDB;
import com.hp.hpl.jena.tdb.sys.SetupTDB;
@RunWith(Parameterized.class)
public class TestScriptedMiniMRCluster extends AbstractMiniMRClusterTest {
@BeforeClass public static void before() throws IOException {
startCluster() ;
}
@AfterClass public static void after() throws IOException {
stopCluster() ;
}
@Before public void setup() {
if ( FileOps.exists(output) ) {
FileOps.clearDirectory(output) ;
} else {
FileOps.ensureDir(output);
}
}
@Parameters
public static Collection<Object[]> data() {
return Arrays.asList(TestScriptedLocal.TEST_DATA);
}
private String input ;
private String output ;
public TestScriptedMiniMRCluster ( String input, String output ) {
this.input = input ;
this.output = output ;
}
@Test public void test() throws Exception {
run (input, output);
}
private void run ( String input, String output ) throws Exception {
String[] args = new String[] {
"-conf", config,
"-D", Constants.OPTION_OVERRIDE_OUTPUT + "=true",
"-D", Constants.OPTION_USE_COMPRESSION + "=" + Constants.OPTION_USE_COMPRESSION_DEFAULT,
"-D", Constants.OPTION_COPY_TO_LOCAL + "=true",
"-D", Constants.OPTION_VERIFY + "=false",
"-D", Constants.OPTION_RUN_LOCAL + "=false",
"-D", Constants.OPTION_NUM_REDUCERS + "=3",
"-D", Constants.OPTION_NUM_SAMPLES + "numSamples=30",
input,
output
};
assertEquals ( 0, ToolRunner.run(new tdbloader4(), args) );
DatasetGraphTDB dsgMem = tdbloader4.load(input);
DatasetGraphTDB dsgDisk = SetupTDB.buildDataset(new Location(output)) ;
assertTrue ( tdbloader4.dump(dsgMem, dsgDisk), tdbloader4.isomorphic ( dsgMem, dsgDisk ) );
}
}