Add null check and throw exception

Author: kfaraz

indexing-service/src/main/java/org/apache/druid/indexing/common/actions/UpdateLocationAction.java

@@ -23,6 +23,7 @@
 import com.fasterxml.jackson.annotation.JsonProperty;
 import com.fasterxml.jackson.core.type.TypeReference;
 import com.google.common.base.Optional;
+import org.apache.druid.error.InvalidInput;
 import org.apache.druid.indexer.TaskLocation;
 import org.apache.druid.indexing.common.task.Task;
 import org.apache.druid.indexing.overlord.TaskRunner;
@@ -42,6 +43,7 @@ public UpdateLocationAction(
       @JsonProperty("location") TaskLocation location
   )
   {
+    InvalidInput.conditionalException(location != null, "No task location specified");
     this.location = location;
   }
 

indexing-service/src/test/java/org/apache/druid/indexing/common/actions/UpdateLocationActionTest.java

@@ -21,6 +21,7 @@
 
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.google.common.base.Optional;
+import org.apache.druid.error.DruidException;
 import org.apache.druid.indexer.TaskLocation;
 import org.apache.druid.indexing.common.task.NoopTask;
 import org.apache.druid.indexing.common.task.Task;
@@ -82,4 +83,13 @@ public void testSerde() throws JsonProcessingException
     final TaskAction<?> deserialized = TestHelper.JSON_MAPPER.readValue(json, TaskAction.class);
     Assert.assertEquals(original, deserialized);
   }
+
+  @Test
+  public void test_actionWithNullLocation_throwsException()
+  {
+    Assert.assertThrows(
+        DruidException.class,
+        () -> new UpdateLocationAction(null)
+    );
+  }
 }