CorcInputFormat implements InputFormat<NullWritable, Corc> { public static void setSearchArgument(Configuration conf, SearchArgument searchArgument) { if (searchArgument != null) { setSearchArgumentKryo(conf, toKryo(searchArgument)); } }  static void setConverterFactoryClass(Configuration conf,
      Class<? extends ConverterFactory> converterFactoryClass); static void setTypeInfo(Configuration conf, StructTypeInfo typeInfo); static void setSchemaTypeInfo(Configuration conf, StructTypeInfo schemaTypeInfo); static void setSearchArgument(Configuration conf, SearchArgument searchArgument); static void setSearchArgumentKryo(Configuration conf, String searchArgumentKryo); @Override InputSplit[] getSplits(JobConf conf, int numSplits); @Override RecordReader<NullWritable, Corc> getRecordReader(InputSplit inputSplit, JobConf conf, Reporter reporter); static String toKryo(SearchArgument sarg); static final String ENABLE_ROW_LEVEL_SEARCH_ARGUMENT; }
@Test public void setSearchArgument() { SearchArgument searchArgument = SearchArgumentFactory.newBuilder().startAnd().equals("a", PredicateLeaf.Type.STRING, "b").end().build(); CorcInputFormat.setSearchArgument(conf, searchArgument); String kryo = conf.get(CorcInputFormat.SEARCH_ARGUMENT); assertThat(kryo, is(CorcInputFormat.toKryo(searchArgument))); }
@Test public void setSearchArgumentNull() { CorcInputFormat.setSearchArgument(conf, null); String kryo = conf.get(CorcInputFormat.SEARCH_ARGUMENT); assertThat(kryo, is(nullValue())); }