From 2e46ef9417e31e38c632d8a966a07c45496755b2 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 e7df5b304b..2de885f861 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 void shutdown() {
     }
   }
 
+  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 4f89840f30..684587d7c8 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 void restart() {
     miniMRYarnCluster.start();
   }
 
+  public MiniMRYarnCluster getMiniMRYarnCluster() {
+    return miniMRYarnCluster;
+  }
+
 }