Fix Feature Test Errors

Author: faizalrub-datastax

src/main/scala/com/datastax/cdm/job/BaseJob.scala

@@ -32,6 +32,7 @@ abstract class BaseJob[T: ClassTag] extends App {
   var maxPartition: BigInteger = _
   var coveragePercent: Int = _
   var numSplits: Int = _
+  var partitionFile: String = _
 
   var parts: util.Collection[T] = _
   var slices: RDD[T] = _
@@ -69,8 +70,9 @@ abstract class BaseJob[T: ClassTag] extends App {
     maxPartition = getMaxPartition(propertyHelper.getString(KnownProperties.PARTITION_MAX), hasRandomPartitioner)
     coveragePercent = propertyHelper.getInteger(KnownProperties.TOKEN_COVERAGE_PERCENT)
     numSplits = propertyHelper.getInteger(KnownProperties.PERF_NUM_PARTS)
-    if ("".equals(this.fileName)) {
-      this.fileName = propertyHelper.getString(KnownProperties.PARTITIONS_TOKEN_RANGE_FILE)
+    partitionFile = propertyHelper.getString(KnownProperties.PARTITIONS_TOKEN_RANGE_FILE)
+    if (partitionFile.trim.nonEmpty && jobName.contains("Partitions from File")) {
+      this.fileName = partitionFile
     }
     abstractLogger.info("PARAM -- Min Partition: " + minPartition)
     abstractLogger.info("PARAM -- Max Partition: " + maxPartition)

src/main/scala/com/datastax/cdm/job/DiffPartitionsFromFile.scala

@@ -1,14 +1,9 @@
 package com.datastax.cdm.job
 
 object DiffPartitionsFromFile extends BasePartitionJob {
-  setup("Diff Partitions from File Job", new DiffJobSessionFactory())
-  if ("".equals(this.fileName)) {
-    abstractLogger.error("Please set conf for spark.input.partitionFile ")
-  }
-  else {
-    execute()
-    finish()
-  }
+  setup("Diff Partitions from File Job", new DiffJobSessionFactory(), "./partitions.csv")
+  execute()
+  finish()
 
   override def execute(): Unit = {
     slices.foreach(slice => {