From 33c5cad5c9ae173fda4097a5035bb81e61fd60a4 Mon Sep 17 00:00:00 2001
From: Andy Galkin <1936087+galkk@users.noreply.github.com>
Date: Thu, 14 Nov 2019 11:40:42 -0800
Subject: [PATCH] Fixed flaky test suite, that was breaking github action build

---
 .../sql/unittest/LocalClusterStateTest.java                 | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/src/test/java/com/amazon/opendistroforelasticsearch/sql/unittest/LocalClusterStateTest.java b/src/test/java/com/amazon/opendistroforelasticsearch/sql/unittest/LocalClusterStateTest.java
index 0c1859eeca..5528468fd8 100644
--- a/src/test/java/com/amazon/opendistroforelasticsearch/sql/unittest/LocalClusterStateTest.java
+++ b/src/test/java/com/amazon/opendistroforelasticsearch/sql/unittest/LocalClusterStateTest.java
@@ -122,12 +122,8 @@ public class LocalClusterStateTest {
 
     @Before
     public void init() {
-        mockLocalClusterState(MAPPING);
-    }
-
-    @After
-    public void cleanUp() {
         LocalClusterState.state(null);
+        mockLocalClusterState(MAPPING);
     }
 
     @Test