Apache
/
druid
mirror of https://github.com/apache/druid.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Make DruidCoordinatorTest fail fast (#4142)

This commit is contained in:
Jihoon Son 2017-04-05 05:58:32 +09:00 committed by Jonathan Wei
parent 81ba0224c5
commit b166e13d2b
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
server/src/test/java/io/druid/server/coordinator/DruidCoordinatorTest.java
View File

@ -26,7 +26,6 @@ import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Lists;
import com.google.common.collect.MapMaker;
import com.google.common.collect.Maps;
import io.druid.client.DruidDataSource;
import io.druid.client.DruidServer;
import io.druid.client.ImmutableDruidDataSource;
@ -77,7 +76,6 @@ import java.util.concurrent.atomic.AtomicReference;
public class DruidCoordinatorTest extends CuratorTestBase
{
private DruidCoordinator coordinator;
private LoadQueueTaskMaster taskMaster;
private MetadataSegmentManager databaseSegmentManager;
private SingleServerInventoryView serverInventoryView;
private ScheduledExecutorFactory scheduledExecutorFactory;
@ -101,7 +99,6 @@ public class DruidCoordinatorTest extends CuratorTestBase
@Before
public void setUp() throws Exception
{
taskMaster = EasyMock.createMock(LoadQueueTaskMaster.class);
druidServer = EasyMock.createMock(DruidServer.class);
serverInventoryView = EasyMock.createMock(SingleServerInventoryView.class);
databaseSegmentManager = EasyMock.createNiceMock(MetadataSegmentManager.class);
@ -174,7 +171,7 @@ public class DruidCoordinatorTest extends CuratorTestBase
new NoopServiceEmitter(),
scheduledExecutorFactory,
null,
taskMaster,
null,
new NoopServiceAnnouncer(){
@Override
public void announce(DruidNode node)
@ -328,10 +325,14 @@ public class DruidCoordinatorTest extends CuratorTestBase
) throws Exception
{
if(pathChildrenCacheEvent.getType().equals(PathChildrenCacheEvent.Type.CHILD_ADDED)){
//Coordinator should try to assign segment to druidServer historical
//Simulate historical loading segment
druidServer.addDataSegment(dataSegment.getIdentifier(), dataSegment);
assignSegmentLatch.countDown();
if (assignSegmentLatch.getCount() > 0) {
//Coordinator should try to assign segment to druidServer historical
//Simulate historical loading segment
druidServer.addDataSegment(dataSegment.getIdentifier(), dataSegment);
assignSegmentLatch.countDown();
} else {
Assert.fail("The same segment is assigned to the same server multiple times");
}
}
}
}
@ -371,9 +372,7 @@ public class DruidCoordinatorTest extends CuratorTestBase
Assert.assertNotNull(dataSourceMap.get(dataSource));
// Simulated the adding of segment to druidServer during SegmentChangeRequestLoad event
// The load rules asks for 2 replicas, therefore 1 replica should still be pending
while(dataSourceMap.get(dataSource).get() != 1L) {
Thread.sleep(50);
}
Assert.assertEquals(1L, dataSourceMap.get(dataSource).get());
coordinator.stop();
leaderUnannouncerLatch.await();