From 41347c82bb8f96a734c0db96cfa9616ea9809678 Mon Sep 17 00:00:00 2001 From: fjy Date: Tue, 13 Aug 2013 12:07:45 -0700 Subject: [PATCH] fix broken server manager tests --- .../druid/coordination/ServerManagerTest.java | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/server/src/test/java/com/metamx/druid/coordination/ServerManagerTest.java b/server/src/test/java/com/metamx/druid/coordination/ServerManagerTest.java index e791eba8764..adb9cb0c88f 100644 --- a/server/src/test/java/com/metamx/druid/coordination/ServerManagerTest.java +++ b/server/src/test/java/com/metamx/druid/coordination/ServerManagerTest.java @@ -267,7 +267,7 @@ public class ServerManagerTest Assert.assertEquals(1, factory.getSegmentReferences().size()); for (ReferenceCountingSegment referenceCountingSegment : factory.getSegmentReferences()) { - Assert.assertEquals(0, referenceCountingSegment.getNumReferences()); + Assert.assertEquals(1, referenceCountingSegment.getNumReferences()); } queryWaitYieldLatch.countDown(); @@ -311,7 +311,7 @@ public class ServerManagerTest queryWaitYieldLatch.countDown(); - Assert.assertTrue(factory.getAdapters().size() == 1); + Assert.assertEquals(1, factory.getAdapters().size()); for (SegmentForTesting segmentForTesting : factory.getAdapters()) { Assert.assertFalse(segmentForTesting.isClosed()); @@ -346,7 +346,15 @@ public class ServerManagerTest queryNotifyLatch.await(); - Assert.assertTrue(factory.getAdapters().size() == 1); + Assert.assertEquals(1, factory.getSegmentReferences().size()); + + for (ReferenceCountingSegment referenceCountingSegment : factory.getSegmentReferences()) { + Assert.assertEquals(1, referenceCountingSegment.getNumReferences()); + } + + queryWaitYieldLatch.countDown(); + + Assert.assertEquals(1, factory.getAdapters().size()); for (SegmentForTesting segmentForTesting : factory.getAdapters()) { Assert.assertFalse(segmentForTesting.isClosed());