From 17e994f2e438d98c7c29600a44f8cc9fbfaa581a Mon Sep 17 00:00:00 2001
From: nishantmonu51 <nishant.monu51@gmail.com>
Date: Fri, 9 May 2014 23:53:25 +0530
Subject: [PATCH] review comment

---
 .../src/test/java/io/druid/db/DatabaseSegmentManagerTest.java  | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/server/src/test/java/io/druid/db/DatabaseSegmentManagerTest.java b/server/src/test/java/io/druid/db/DatabaseSegmentManagerTest.java
index 329f5490ad1..553c1b3946d 100644
--- a/server/src/test/java/io/druid/db/DatabaseSegmentManagerTest.java
+++ b/server/src/test/java/io/druid/db/DatabaseSegmentManagerTest.java
@@ -106,8 +106,7 @@ public class DatabaseSegmentManagerTest
   @Test
   public void testPoll()
   {
-    EasyMock.expect(dbi.withHandle(EasyMock.<HandleCallback>anyObject())).andReturn(testRows);
-    EasyMock.expect(dbi.withHandle(EasyMock.<HandleCallback>anyObject())).andReturn(testRows);
+    EasyMock.expect(dbi.withHandle(EasyMock.<HandleCallback>anyObject())).andReturn(testRows).times(2);
     EasyMock.replay(dbi);
 
     manager.start();