From 2c6da09d9d17f33ebc025c9ec9fa949605288bb7 Mon Sep 17 00:00:00 2001
From: Balaji Varadarajan <varadarb@uber.com>
Date: Mon, 16 Sep 2019 23:42:14 -0700
Subject: [PATCH] [HUDI-257] Fix Bloom Index unit-test failures

---
 .../java/org/apache/hudi/index/bloom/TestHoodieBloomIndex.java  | 2 ++
 .../test/java/org/apache/hudi/table/TestCopyOnWriteTable.java   | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/hudi-client/src/test/java/org/apache/hudi/index/bloom/TestHoodieBloomIndex.java b/hudi-client/src/test/java/org/apache/hudi/index/bloom/TestHoodieBloomIndex.java
index 89766835d..28d19eb1c 100644
--- a/hudi-client/src/test/java/org/apache/hudi/index/bloom/TestHoodieBloomIndex.java
+++ b/hudi-client/src/test/java/org/apache/hudi/index/bloom/TestHoodieBloomIndex.java
@@ -91,6 +91,7 @@ public class TestHoodieBloomIndex extends HoodieClientTestHarness {
   public void setUp() throws Exception {
     initSparkContexts("TestHoodieBloomIndex");
     initTempFolderAndPath();
+    initFileSystem();
     HoodieTestUtils.init(jsc.hadoopConfiguration(), basePath);
     // We have some records to be tagged (two different partitions)
     schemaStr = FileIOUtils.readAsUTFString(getClass().getResourceAsStream("/exampleSchema.txt"));
@@ -100,6 +101,7 @@ public class TestHoodieBloomIndex extends HoodieClientTestHarness {
   @After
   public void tearDown() throws Exception {
     cleanupSparkContexts();
+    cleanupFileSystem();
     cleanupTempFolderAndPath();
   }
 
diff --git a/hudi-client/src/test/java/org/apache/hudi/table/TestCopyOnWriteTable.java b/hudi-client/src/test/java/org/apache/hudi/table/TestCopyOnWriteTable.java
index da7a0e8e4..6439d7520 100644
--- a/hudi-client/src/test/java/org/apache/hudi/table/TestCopyOnWriteTable.java
+++ b/hudi-client/src/test/java/org/apache/hudi/table/TestCopyOnWriteTable.java
@@ -75,6 +75,7 @@ public class TestCopyOnWriteTable extends HoodieClientTestHarness {
     initTempFolderAndPath();
     initTableType();
     initTestDataGenerator();
+    initFileSystem();
   }
 
   @After
@@ -82,6 +83,7 @@ public class TestCopyOnWriteTable extends HoodieClientTestHarness {
     cleanupSparkContexts();
     cleanupTempFolderAndPath();
     cleanupTableType();
+    cleanupFileSystem();
     cleanupTestDataGenerator();
   }