X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-distributed-datastore%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fdatastore%2FShardManagerTest.java;h=02201f7cd1672d2c8d02415ae4d65d8b71432f8e;hp=fa436c16053bc42ad9835e7ecbccd1cb202fc4b8;hb=c33b2b55b2eae406df001619885a0610800cb951;hpb=81aa5072801e6453306e296b91dba3dbeeaf046d diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/ShardManagerTest.java b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/ShardManagerTest.java index fa436c1605..02201f7cd1 100644 --- a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/ShardManagerTest.java +++ b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/ShardManagerTest.java @@ -1,27 +1,39 @@ package org.opendaylight.controller.cluster.datastore; +import akka.actor.ActorRef; import akka.actor.ActorSystem; import akka.actor.Props; import akka.testkit.JavaTestKit; import akka.testkit.TestActorRef; +import junit.framework.Assert; import org.junit.AfterClass; import org.junit.BeforeClass; import org.junit.Test; +import org.opendaylight.controller.cluster.datastore.messages.FindLocalShard; import org.opendaylight.controller.cluster.datastore.messages.FindPrimary; +import org.opendaylight.controller.cluster.datastore.messages.LocalShardFound; +import org.opendaylight.controller.cluster.datastore.messages.LocalShardNotFound; import org.opendaylight.controller.cluster.datastore.messages.PrimaryFound; import org.opendaylight.controller.cluster.datastore.messages.PrimaryNotFound; +import org.opendaylight.controller.cluster.datastore.messages.UpdateSchemaContext; +import org.opendaylight.controller.cluster.datastore.utils.MockClusterWrapper; +import org.opendaylight.controller.cluster.datastore.utils.MockConfiguration; +import org.opendaylight.controller.md.cluster.datastore.model.TestModel; import scala.concurrent.duration.Duration; +import static junit.framework.Assert.assertEquals; +import static org.junit.Assert.assertTrue; + public class ShardManagerTest { private static ActorSystem system; @BeforeClass - public static void setUp(){ + public static void setUp() { system = ActorSystem.create("test"); } @AfterClass - public static void tearDown(){ + public static void tearDown() { JavaTestKit.shutdownActorSystem(system); system = null; } @@ -30,41 +42,200 @@ public class ShardManagerTest { public void testOnReceiveFindPrimaryForNonExistentShard() throws Exception { new JavaTestKit(system) {{ - final Props props = ShardManager.props("config"); - final TestActorRef subject = TestActorRef.create(system, props); + final Props props = ShardManager + .props("config", new MockClusterWrapper(), + new MockConfiguration(), new DatastoreContext()); + final TestActorRef subject = + TestActorRef.create(system, props); + + new Within(duration("10 seconds")) { + @Override + protected void run() { + + subject.tell(new FindPrimary("inventory").toSerializable(), getRef()); + + expectMsgEquals(Duration.Zero(), + new PrimaryNotFound("inventory").toSerializable()); + + expectNoMsg(); + } + }; + }}; + } + + @Test + public void testOnReceiveFindPrimaryForExistentShard() throws Exception { + + new JavaTestKit(system) {{ + final Props props = ShardManager + .props("config", new MockClusterWrapper(), + new MockConfiguration(), new DatastoreContext()); + final TestActorRef subject = + TestActorRef.create(system, props); + + subject.tell(new UpdateSchemaContext(TestModel.createTestContext()), getRef()); + + new Within(duration("10 seconds")) { + @Override + protected void run() { + + subject.tell(new FindPrimary(Shard.DEFAULT_NAME).toSerializable(), getRef()); + + expectMsgClass(duration("1 seconds"), PrimaryFound.SERIALIZABLE_CLASS); - new Within(duration("1 seconds")) { + expectNoMsg(); + } + }; + }}; + } + + @Test + public void testOnReceiveFindLocalShardForNonExistentShard() throws Exception { + + new JavaTestKit(system) {{ + final Props props = ShardManager + .props("config", new MockClusterWrapper(), + new MockConfiguration(), new DatastoreContext()); + final TestActorRef subject = + TestActorRef.create(system, props); + + new Within(duration("10 seconds")) { + @Override protected void run() { - subject.tell(new FindPrimary("inventory"), getRef()); + subject.tell(new FindLocalShard("inventory"), getRef()); + + final String out = new ExpectMsg(duration("10 seconds"), "find local") { + @Override + protected String match(Object in) { + if (in instanceof LocalShardNotFound) { + return ((LocalShardNotFound) in).getShardName(); + } else { + throw noMatch(); + } + } + }.get(); // this extracts the received message - expectMsgEquals(Duration.Zero(), new PrimaryNotFound("inventory")); + assertEquals("inventory", out); - // Will wait for the rest of the 3 seconds expectNoMsg(); } }; }}; } - @Test - public void testOnReceiveFindPrimaryForExistentShard() throws Exception { + @Test + public void testOnReceiveFindLocalShardForExistentShard() throws Exception { + + final MockClusterWrapper mockClusterWrapper = new MockClusterWrapper(); + + new JavaTestKit(system) {{ + final Props props = ShardManager + .props("config", mockClusterWrapper, + new MockConfiguration(), new DatastoreContext()); + final TestActorRef subject = + TestActorRef.create(system, props); + + subject.tell(new UpdateSchemaContext(TestModel.createTestContext()), getRef()); + + new Within(duration("10 seconds")) { + @Override + protected void run() { + + subject.tell(new FindLocalShard(Shard.DEFAULT_NAME), getRef()); + + final ActorRef out = new ExpectMsg(duration("10 seconds"), "find local") { + @Override + protected ActorRef match(Object in) { + if (in instanceof LocalShardFound) { + return ((LocalShardFound) in).getPath(); + } else { + throw noMatch(); + } + } + }.get(); // this extracts the received message - new JavaTestKit(system) {{ - final Props props = ShardManager.props("config"); - final TestActorRef subject = TestActorRef.create(system, props); + assertTrue(out.path().toString(), out.path().toString().contains("member-1-shard-default-config")); - // the run() method needs to finish within 3 seconds - new Within(duration("1 seconds")) { - protected void run() { - subject.tell(new FindPrimary(Shard.DEFAULT_NAME), getRef()); + expectNoMsg(); + } + }; + }}; + } + + @Test + public void testOnReceiveMemberUp() throws Exception { + + new JavaTestKit(system) {{ + final Props props = ShardManager + .props("config", new MockClusterWrapper(), + new MockConfiguration(), new DatastoreContext()); + final TestActorRef subject = + TestActorRef.create(system, props); + + // the run() method needs to finish within 3 seconds + new Within(duration("10 seconds")) { + @Override + protected void run() { + + MockClusterWrapper.sendMemberUp(subject, "member-2", getRef().path().toString()); + + subject.tell(new FindPrimary("astronauts").toSerializable(), getRef()); + + final String out = new ExpectMsg(duration("1 seconds"), "primary found") { + // do not put code outside this method, will run afterwards + @Override + protected String match(Object in) { + if (in.getClass().equals(PrimaryFound.SERIALIZABLE_CLASS)) { + PrimaryFound f = PrimaryFound.fromSerializable(in); + return f.getPrimaryPath(); + } else { + throw noMatch(); + } + } + }.get(); // this extracts the received message + + Assert.assertTrue(out, out.contains("member-2-shard-astronauts-config")); + + expectNoMsg(); + } + }; + }}; + } + + @Test + public void testOnReceiveMemberDown() throws Exception { + + new JavaTestKit(system) {{ + final Props props = ShardManager + .props("config", new MockClusterWrapper(), + new MockConfiguration(), new DatastoreContext()); + final TestActorRef subject = + TestActorRef.create(system, props); + + // the run() method needs to finish within 3 seconds + new Within(duration("10 seconds")) { + @Override + protected void run() { + + MockClusterWrapper.sendMemberUp(subject, "member-2", getRef().path().toString()); + + subject.tell(new FindPrimary("astronauts").toSerializable(), getRef()); + + expectMsgClass(duration("1 seconds"), PrimaryFound.SERIALIZABLE_CLASS); + + MockClusterWrapper.sendMemberRemoved(subject, "member-2", getRef().path().toString()); + + subject.tell(new FindPrimary("astronauts").toSerializable(), getRef()); + + expectMsgClass(duration("1 seconds"), PrimaryNotFound.SERIALIZABLE_CLASS); + + expectNoMsg(); + } + }; + }}; + } - expectMsgClass(PrimaryFound.class); - expectNoMsg(); - } - }; - }}; - } -} \ No newline at end of file +}