From 056a678fa88ecf4c12a54fdaf4044bada9a40abd Mon Sep 17 00:00:00 2001
From: Shay Banon <kimchy@gmail.com>
Date: Fri, 13 Sep 2013 22:06:12 +0200
Subject: [PATCH] use built in flush/refresh helper methods

---
 .../broadcast/BroadcastActionsTests.java          | 15 +++------------
 1 file changed, 3 insertions(+), 12 deletions(-)

diff --git a/src/test/java/org/elasticsearch/broadcast/BroadcastActionsTests.java b/src/test/java/org/elasticsearch/broadcast/BroadcastActionsTests.java
index a87211c19e8..0226d501982 100644
--- a/src/test/java/org/elasticsearch/broadcast/BroadcastActionsTests.java
+++ b/src/test/java/org/elasticsearch/broadcast/BroadcastActionsTests.java
@@ -20,23 +20,20 @@
 package org.elasticsearch.broadcast;
 
 import com.google.common.base.Charsets;
+import org.elasticsearch.AbstractSharedClusterTest;
 import org.elasticsearch.action.ShardOperationFailedException;
 import org.elasticsearch.action.admin.cluster.health.ClusterHealthResponse;
 import org.elasticsearch.action.admin.cluster.health.ClusterHealthStatus;
-import org.elasticsearch.action.admin.indices.flush.FlushResponse;
-import org.elasticsearch.action.admin.indices.refresh.RefreshResponse;
 import org.elasticsearch.action.count.CountResponse;
 import org.elasticsearch.action.support.broadcast.BroadcastOperationThreading;
 import org.elasticsearch.common.xcontent.XContentBuilder;
 import org.elasticsearch.common.xcontent.XContentFactory;
-import org.elasticsearch.AbstractSharedClusterTest;
 import org.junit.Test;
 
 import java.io.IOException;
 
 import static org.elasticsearch.client.Requests.*;
 import static org.elasticsearch.index.query.QueryBuilders.termQuery;
-import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.containsString;
 import static org.hamcrest.Matchers.equalTo;
 
@@ -56,15 +53,9 @@ public class BroadcastActionsTests extends AbstractSharedClusterTest {
         assertThat(clusterHealth.getStatus(), equalTo(ClusterHealthStatus.YELLOW));
 
         client().index(indexRequest("test").type("type1").id("1").source(source("1", "test"))).actionGet();
-        FlushResponse flushResponse = client().admin().indices().flush(flushRequest("test")).actionGet();
-        assertThat(flushResponse.getTotalShards(), equalTo(10));
-        assertThat(flushResponse.getSuccessfulShards(), equalTo(5));
-        assertThat(flushResponse.getFailedShards(), equalTo(0));
+        flush();
         client().index(indexRequest("test").type("type1").id("2").source(source("2", "test"))).actionGet();
-        RefreshResponse refreshResponse = client().admin().indices().refresh(refreshRequest("test")).actionGet();
-        assertThat(refreshResponse.getTotalShards(), equalTo(10));
-        assertThat(refreshResponse.getSuccessfulShards(), equalTo(5));
-        assertThat(refreshResponse.getFailedShards(), equalTo(0));
+        refresh();
 
         logger.info("Count");
         // check count