Class TestCheckTool

java/tools/src/test/org/apache/orc/tools/TestCheckTool.java:42–210

import static org.junit.jupiter.api.Assertions.assertTrue;

public class TestCheckTool implements TestConf {
  private Path workDir = new Path(System.getProperty("test.tmp.dir"));
  private FileSystem fs;
  private Path testFilePath;

  @BeforeEach
  public void openFileSystem() throws Exception {
    fs = FileSystem.getLocal(conf);
    testFilePath = new Path(workDir + File.separator + "TestCheckTool.testCheckTool.orc");
    fs.delete(testFilePath, false);
    createFile();
  }

  private void createFile() throws IOException {
    TypeDescription schema = TypeDescription.fromString("struct<x:int,y:string,z:string>");
    Writer writer = OrcFile.createWriter(testFilePath,
        OrcFile.writerOptions(conf)
            .bloomFilterColumns("x,y")
            .rowIndexStride(5000)
            .setSchema(schema));
    VectorizedRowBatch batch = schema.createRowBatch();
    LongColumnVector x = (LongColumnVector) batch.cols[0];
    BytesColumnVector y = (BytesColumnVector) batch.cols[1];
    BytesColumnVector z = (BytesColumnVector) batch.cols[2];
    for (int r = 0; r < 10000; ++r) {
      int row = batch.size++;
      x.vector[row] = r;
      byte[] yBuffer = ("y-byte-" + r).getBytes();
      byte[] zBuffer = ("z-byte-" + r).getBytes();
      y.setRef(row, yBuffer, 0, yBuffer.length);
      z.setRef(row, zBuffer, 0, zBuffer.length);
      if (batch.size == batch.getMaxSize()) {
        writer.addRowBatch(batch);
        batch.reset();
      }
    }
    if (batch.size != 0) {
      writer.addRowBatch(batch);
    }
    writer.close();
  }

  @Test
  public void testPredicate() throws Exception {
    PrintStream origOut = System.out;
    ByteArrayOutputStream myOut = new ByteArrayOutputStream();
    System.setOut(new PrintStream(myOut, false, StandardCharsets.UTF_8));

    CheckTool.main(conf, new String[]{
        "--type", "predicate",
        "--values", "0", "--values", "5566",
        "--column", "x",
        testFilePath.toString()});

    CheckTool.main(conf, new String[]{
        "--type", "predicate",
        "--values", "y-byte-1234", "--values", "y-byte-5566",
        "--column", "y",