commit 86182d0355c89bd632d5617ba517cea469128fb3
author László Bodor <bodorlaszlo0202@gmail.com> Mon Feb 10 08:56:26 2020 -0600
committer Jonathan Eagles <jeagles@apache.org> Mon Feb 10 08:56:26 2020 -0600
tree 33a39c2e0bd4c4929312a3997aa60a0f599d54cd
parent b078e3a25f892fcd610a1cc15688e825e41a9bcf

TEZ-4122. TestMRRJobsDAGApi should set TezClassLoader

Signed-off-by: Jonathan Eagles <jeagles@apache.org>

tez-tests/src/test/java/org/apache/tez/mapreduce/TestMRRJobsDAGApi.java

diff --git a/tez-tests/src/test/java/org/apache/tez/mapreduce/TestMRRJobsDAGApi.java b/tez-tests/src/test/java/org/apache/tez/mapreduce/TestMRRJobsDAGApi.java
index a756cf9..96b7bbf 100644
--- a/tez-tests/src/test/java/org/apache/tez/mapreduce/TestMRRJobsDAGApi.java
+++ b/tez-tests/src/test/java/org/apache/tez/mapreduce/TestMRRJobsDAGApi.java
@@ -71,6 +71,7 @@
 import org.apache.tez.client.TezClient;
 import org.apache.tez.client.TezAppMasterStatus;
 import org.apache.tez.common.ReflectionUtils;
+import org.apache.tez.common.TezClassLoader;
 import org.apache.tez.common.TezUtils;
 import org.apache.tez.common.counters.FileSystemCounter;
 import org.apache.tez.common.counters.TaskCounter;
@@ -833,6 +834,7 @@
           .getConfigurationBytes());
 
       try {
+        Thread.currentThread().setContextClassLoader(TezClassLoader.getInstance());
         ReflectionUtils.getClazz(RELOCALIZATION_TEST_CLASS_NAME);
         LOG.info("Class found");
         FileSystem fs = FileSystem.get(conf);