From c814e58348308b05b215db427412cd6c0b21333e Mon Sep 17 00:00:00 2001
From: Boglarka Egyed <bogi@apache.org>
Date: Mon, 3 Sep 2018 11:52:44 +0200
Subject: [PATCH] SQOOP-3375: HiveMiniCluster does not restore hive-site.xml
 location

(Szabolcs Vasas via Boglarka Egyed)
---
 .../org/apache/sqoop/hive/minicluster/HiveMiniCluster.java  | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/test/org/apache/sqoop/hive/minicluster/HiveMiniCluster.java b/src/test/org/apache/sqoop/hive/minicluster/HiveMiniCluster.java
index 19bb7605..9dd54486 100644
--- a/src/test/org/apache/sqoop/hive/minicluster/HiveMiniCluster.java
+++ b/src/test/org/apache/sqoop/hive/minicluster/HiveMiniCluster.java
@@ -32,6 +32,7 @@
 import java.io.OutputStream;
 import java.net.InetAddress;
 import java.net.Socket;
+import java.net.URL;
 import java.security.PrivilegedAction;
 import java.util.Map;
 import java.util.concurrent.TimeoutException;
@@ -56,6 +57,8 @@ public class HiveMiniCluster {
 
   private HiveConf config;
 
+  private URL originalHiveSiteLocation;
+
   public HiveMiniCluster(AuthenticationConfiguration authenticationConfiguration) {
     this(DEFAULT_HOST, DEFAULT_PORT, authenticationConfiguration);
   }
@@ -102,6 +105,7 @@ private void createHiveSiteXml() throws IOException {
       config.writeXml(out);
     }
 
+    originalHiveSiteLocation = HiveConf.getHiveSiteLocation();
     HiveConf.setHiveSiteLocation(hiveSiteXmlFile.toURI().toURL());
   }
 
@@ -118,7 +122,7 @@ public Void run() {
 
   public void stop() {
     hiveServer2.stop();
-    HiveConf.setHiveSiteLocation(null);
+    HiveConf.setHiveSiteLocation(originalHiveSiteLocation);
     try {
       FileUtils.deleteDirectory(new File(tempFolderPath));
     } catch (IOException e) {