From 0d5534430c203620417c501c0fbfbd783ae75680 Mon Sep 17 00:00:00 2001
From: Swaroopa Kadam <s.kadam@salesforce.com>
Date: Thu, 8 Oct 2020 14:56:27 -0700
Subject: [PATCH] MAPREDUCE-7301: Expose Mini MR Cluster attribute for testing

Signed-off-by: Mingliang Liu <liuml07@apache.org>
---
 .../src/test/java/org/apache/hadoop/mapred/MiniMRCluster.java | 4 ++++
 .../org/apache/hadoop/mapred/MiniMRYarnClusterAdapter.java    | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/hadoop-mapreduce-project/hadoop-mapreduce-client/hadoop-mapreduce-client-jobclient/src/test/java/org/apache/hadoop/mapred/MiniMRCluster.java b/hadoop-mapreduce-project/hadoop-mapreduce-client/hadoop-mapreduce-client-jobclient/src/test/java/org/apache/hadoop/mapred/MiniMRCluster.java
index e7df5b304b5..2de885f8615 100644
--- a/hadoop-mapreduce-project/hadoop-mapreduce-client/hadoop-mapreduce-client-jobclient/src/test/java/org/apache/hadoop/mapred/MiniMRCluster.java
+++ b/hadoop-mapreduce-project/hadoop-mapreduce-client/hadoop-mapreduce-client-jobclient/src/test/java/org/apache/hadoop/mapred/MiniMRCluster.java
@@ -271,4 +271,8 @@ public class MiniMRCluster {
     }
   }
 
+  public MiniMRClientCluster getMrClientCluster() {
+    return mrClientCluster;
+  }
+
 }
diff --git a/hadoop-mapreduce-project/hadoop-mapreduce-client/hadoop-mapreduce-client-jobclient/src/test/java/org/apache/hadoop/mapred/MiniMRYarnClusterAdapter.java b/hadoop-mapreduce-project/hadoop-mapreduce-client/hadoop-mapreduce-client-jobclient/src/test/java/org/apache/hadoop/mapred/MiniMRYarnClusterAdapter.java
index 4f89840f300..684587d7c87 100644
--- a/hadoop-mapreduce-project/hadoop-mapreduce-client/hadoop-mapreduce-client-jobclient/src/test/java/org/apache/hadoop/mapred/MiniMRYarnClusterAdapter.java
+++ b/hadoop-mapreduce-project/hadoop-mapreduce-client/hadoop-mapreduce-client-jobclient/src/test/java/org/apache/hadoop/mapred/MiniMRYarnClusterAdapter.java
@@ -74,4 +74,8 @@ public class MiniMRYarnClusterAdapter implements MiniMRClientCluster {
     miniMRYarnCluster.start();
   }
 
+  public MiniMRYarnCluster getMiniMRYarnCluster() {
+    return miniMRYarnCluster;
+  }
+
 }