diff --git a/test/unit/voldemort/store/system/AsyncMetadataVersionManagerTest.java b/test/unit/voldemort/store/system/AsyncMetadataVersionManagerTest.java index be730842e5..98b25c9bc5 100644 --- a/test/unit/voldemort/store/system/AsyncMetadataVersionManagerTest.java +++ b/test/unit/voldemort/store/system/AsyncMetadataVersionManagerTest.java @@ -20,6 +20,7 @@ import static org.junit.Assert.fail; import java.util.Date; +import java.util.List; import java.util.Properties; import java.util.concurrent.Callable; @@ -27,13 +28,17 @@ import org.junit.Before; import org.junit.Test; +import voldemort.ClusterTestUtils; import voldemort.ServerTestUtils; +import voldemort.TestUtils; import voldemort.client.SystemStore; import voldemort.client.SystemStoreRepository; import voldemort.client.scheduler.AsyncMetadataVersionManager; import voldemort.cluster.Cluster; import voldemort.common.service.SchedulerService; +import voldemort.server.VoldemortConfig; import voldemort.server.VoldemortServer; +import voldemort.store.StoreDefinition; import voldemort.store.socket.SocketStoreFactory; import voldemort.store.socket.clientrequest.ClientRequestExecutorPool; import voldemort.utils.MetadataVersionStoreUtils; @@ -70,17 +75,24 @@ public class AsyncMetadataVersionManagerTest { @Before public void setUp() throws Exception { - final int numServers = 2; - servers = new VoldemortServer[numServers]; - int partitionMap[][] = { { 0, 1, 2, 3 }, { 4, 5, 6, 7 } }; - cluster = ServerTestUtils.startVoldemortCluster(numServers, - servers, - partitionMap, - socketStoreFactory, - true, // useNio - null, - storesXmlfile, - new Properties()); + cluster = ClusterTestUtils.getZZZCluster(); + servers = new VoldemortServer[cluster.getNodeIds().size()]; + + List storeDefs = ClusterTestUtils.getZZZ322StoreDefs("bdb"); + int i = 0; + + for(Integer nodeId: cluster.getNodeIds()) { + VoldemortConfig config = ServerTestUtils.createServerConfigWithDefs(true, + nodeId, + TestUtils.createTempDir() + .getAbsolutePath(), + cluster, + storeDefs, + new Properties()); + VoldemortServer server = ServerTestUtils.startVoldemortServer(socketStoreFactory, + config); + servers[i++] = server; + } socketUrl = servers[0].getIdentityNode().getSocketUrl().toString(); diff --git a/test/unit/voldemort/store/system/SystemStoreTest.java b/test/unit/voldemort/store/system/SystemStoreTest.java index b47dc79c54..c0b29c503b 100644 --- a/test/unit/voldemort/store/system/SystemStoreTest.java +++ b/test/unit/voldemort/store/system/SystemStoreTest.java @@ -19,19 +19,24 @@ import static org.junit.Assert.assertEquals; import static org.junit.Assert.fail; +import java.util.List; import java.util.Properties; import org.junit.After; import org.junit.Before; import org.junit.Test; +import voldemort.ClusterTestUtils; import voldemort.ServerTestUtils; +import voldemort.TestUtils; import voldemort.client.AbstractStoreClientFactory; import voldemort.client.ClientConfig; import voldemort.client.SocketStoreClientFactory; import voldemort.client.SystemStore; import voldemort.cluster.Cluster; +import voldemort.server.VoldemortConfig; import voldemort.server.VoldemortServer; +import voldemort.store.StoreDefinition; import voldemort.store.metadata.MetadataStore; import voldemort.store.socket.SocketStoreFactory; import voldemort.store.socket.clientrequest.ClientRequestExecutorPool; @@ -60,20 +65,25 @@ public class SystemStoreTest { @Before public void setUp() throws Exception { - final int numServers = 2; - servers = new VoldemortServer[numServers]; - int partitionMap[][] = { { 0, 1, 2, 3 }, { 4, 5, 6, 7 } }; - - cluster = ServerTestUtils.getLocalCluster(2, new int[][] { { 0, 1, 2, 3 }, { 4, 5, 6, 7 } }); - servers = new VoldemortServer[2]; - cluster = ServerTestUtils.startVoldemortCluster(numServers, - servers, - partitionMap, - socketStoreFactory, - true, // useNio - null, - storesXmlfile, - new Properties()); + + cluster = ClusterTestUtils.getZZZCluster(); + servers = new VoldemortServer[cluster.getNodeIds().size()]; + + List storeDefs = ClusterTestUtils.getZZZ322StoreDefs("bdb"); + int i = 0; + + for(Integer nodeId: cluster.getNodeIds()) { + VoldemortConfig config = ServerTestUtils.createServerConfigWithDefs(true, + nodeId, + TestUtils.createTempDir() + .getAbsolutePath(), + cluster, + storeDefs, + new Properties()); + VoldemortServer server = ServerTestUtils.startVoldemortServer(socketStoreFactory, + config); + servers[i++] = server; + } socketUrl = servers[0].getIdentityNode().getSocketUrl().toString();