X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-distributed-datastore%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fdatastore%2FShardManagerTest.java;h=02201f7cd1672d2c8d02415ae4d65d8b71432f8e;hb=0ef97b34b82e9987ada3ad2a318fd03731385aa4;hp=87d257a3f217bf0bfac0b3c8ade1cbe8d4555b04;hpb=516a4b2ea78179c9bd6ebb584862e8fc686ebf08;p=controller.git 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 87d257a3f2..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,5 +1,6 @@ package org.opendaylight.controller.cluster.datastore; +import akka.actor.ActorRef; import akka.actor.ActorSystem; import akka.actor.Props; import akka.testkit.JavaTestKit; @@ -8,13 +9,21 @@ 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; @@ -35,11 +44,12 @@ public class ShardManagerTest { new JavaTestKit(system) {{ final Props props = ShardManager .props("config", new MockClusterWrapper(), - new MockConfiguration()); + new MockConfiguration(), new DatastoreContext()); final TestActorRef subject = TestActorRef.create(system, props); - new Within(duration("1 seconds")) { + new Within(duration("10 seconds")) { + @Override protected void run() { subject.tell(new FindPrimary("inventory").toSerializable(), getRef()); @@ -47,7 +57,6 @@ public class ShardManagerTest { expectMsgEquals(Duration.Zero(), new PrimaryNotFound("inventory").toSerializable()); - // Will wait for the rest of the 3 seconds expectNoMsg(); } }; @@ -60,17 +69,94 @@ public class ShardManagerTest { new JavaTestKit(system) {{ final Props props = ShardManager .props("config", new MockClusterWrapper(), - new MockConfiguration()); + new MockConfiguration(), new DatastoreContext()); final TestActorRef subject = TestActorRef.create(system, props); - // the run() method needs to finish within 3 seconds - new Within(duration("1 seconds")) { + 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(PrimaryFound.SERIALIZABLE_CLASS); + expectMsgClass(duration("1 seconds"), PrimaryFound.SERIALIZABLE_CLASS); + + 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 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 + + assertEquals("inventory", out); + + expectNoMsg(); + } + }; + }}; + } + + @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 + + assertTrue(out.path().toString(), out.path().toString().contains("member-1-shard-default-config")); + expectNoMsg(); } @@ -84,20 +170,22 @@ public class ShardManagerTest { new JavaTestKit(system) {{ final Props props = ShardManager .props("config", new MockClusterWrapper(), - new MockConfiguration()); + new MockConfiguration(), new DatastoreContext()); final TestActorRef subject = TestActorRef.create(system, props); // the run() method needs to finish within 3 seconds - new Within(duration("1 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("primary found") { + 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); @@ -122,25 +210,26 @@ public class ShardManagerTest { new JavaTestKit(system) {{ final Props props = ShardManager .props("config", new MockClusterWrapper(), - new MockConfiguration()); + new MockConfiguration(), new DatastoreContext()); final TestActorRef subject = TestActorRef.create(system, props); // the run() method needs to finish within 3 seconds - new Within(duration("1 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(PrimaryFound.SERIALIZABLE_CLASS); + expectMsgClass(duration("1 seconds"), PrimaryFound.SERIALIZABLE_CLASS); MockClusterWrapper.sendMemberRemoved(subject, "member-2", getRef().path().toString()); subject.tell(new FindPrimary("astronauts").toSerializable(), getRef()); - expectMsgClass(PrimaryNotFound.SERIALIZABLE_CLASS); + expectMsgClass(duration("1 seconds"), PrimaryNotFound.SERIALIZABLE_CLASS); expectNoMsg(); }