diff --git a/hbase-server/src/main/java/org/apache/hadoop/hbase/mapred/Driver.java b/hbase-server/src/main/java/org/apache/hadoop/hbase/mapred/Driver.java index 3c501c5..7e9a6f3 100644 --- a/hbase-server/src/main/java/org/apache/hadoop/hbase/mapred/Driver.java +++ b/hbase-server/src/main/java/org/apache/hadoop/hbase/mapred/Driver.java @@ -21,24 +21,31 @@ package org.apache.hadoop.hbase.mapred; import org.apache.hadoop.classification.InterfaceAudience; import org.apache.hadoop.classification.InterfaceStability; import org.apache.hadoop.util.ProgramDriver; +import com.google.common.annotations.VisibleForTesting; /** - * Driver for hbase mapreduce jobs. Select which to run by passing - * name of job to this main. + * Driver for hbase mapreduce jobs. Select which to run by passing name of job + * to this main. */ @Deprecated @InterfaceAudience.Public @InterfaceStability.Stable public class Driver { + + private static ProgramDriver pgd = new ProgramDriver(); + + @VisibleForTesting + static void setProgramDriver(ProgramDriver pgd0) { + pgd = pgd0; + } + /** * @param args * @throws Throwable */ public static void main(String[] args) throws Throwable { - ProgramDriver pgd = new ProgramDriver(); - pgd.addClass(RowCounter.NAME, RowCounter.class, - "Count rows in HBase table"); - ProgramDriver.class.getMethod("driver", new Class [] {String[].class}). - invoke(pgd, new Object[]{args}); + pgd.addClass(RowCounter.NAME, RowCounter.class, "Count rows in HBase table"); + ProgramDriver.class.getMethod("driver", new Class[] { String[].class }) + .invoke(pgd, new Object[] { args }); } }