Add OnDemandShardState to report additional Shard state
[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / test / java / org / opendaylight / controller / cluster / datastore / DataChangeListenerTest.java
index 25d47388fe49579fdb66271e4f33ca7effeef314..f3dcaa2b709810d169ae1ae1cc0b54c716051fe9 100644 (file)
@@ -1,5 +1,15 @@
+/*
+ * Copyright (c) 2014, 2015 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
 package org.opendaylight.controller.cluster.datastore;
 
+import static org.opendaylight.controller.md.cluster.datastore.model.TestModel.TEST_PATH;
+
 import akka.actor.ActorRef;
 import akka.actor.DeadLetter;
 import akka.actor.Props;
@@ -10,121 +20,123 @@ import org.mockito.Mockito;
 import org.opendaylight.controller.cluster.datastore.messages.DataChanged;
 import org.opendaylight.controller.cluster.datastore.messages.DataChangedReply;
 import org.opendaylight.controller.cluster.datastore.messages.EnableNotification;
-import org.opendaylight.controller.md.cluster.datastore.model.CompositeModel;
 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeListener;
-import org.opendaylight.yangtools.yang.model.api.SchemaContext;
 
 public class DataChangeListenerTest extends AbstractActorTest {
 
     @SuppressWarnings({ "rawtypes", "unchecked" })
     @Test
-    public void testDataChangedWhenNotificationsAreEnabled(){
-        new JavaTestKit(getSystem()) {{
-            final AsyncDataChangeEvent mockChangeEvent = Mockito.mock(AsyncDataChangeEvent.class);
-            final AsyncDataChangeListener mockListener = Mockito.mock(AsyncDataChangeListener.class);
-            final Props props = DataChangeListener.props(mockListener);
-            final ActorRef subject = getSystem().actorOf(props, "testDataChangedNotificationsEnabled");
+    public void testDataChangedWhenNotificationsAreEnabled() {
+        new JavaTestKit(getSystem()) {
+            {
+                final AsyncDataChangeEvent mockChangeEvent = Mockito.mock(AsyncDataChangeEvent.class);
+                final AsyncDataChangeListener mockListener = Mockito.mock(AsyncDataChangeListener.class);
+                final Props props = DataChangeListener.props(mockListener, TEST_PATH);
+                final ActorRef subject = getSystem().actorOf(props, "testDataChangedNotificationsEnabled");
 
-            // Let the DataChangeListener know that notifications should be enabled
-            subject.tell(new EnableNotification(true), getRef());
+                // Let the DataChangeListener know that notifications should be
+                // enabled
+                subject.tell(new EnableNotification(true), getRef());
 
-            subject.tell(new DataChanged(CompositeModel.createTestContext(), mockChangeEvent),
-                    getRef());
+                subject.tell(new DataChanged(mockChangeEvent), getRef());
 
-            expectMsgClass(DataChangedReply.class);
+                expectMsgClass(DataChangedReply.class);
 
-            Mockito.verify(mockListener).onDataChanged(mockChangeEvent);
-        }};
+                Mockito.verify(mockListener).onDataChanged(mockChangeEvent);
+            }
+        };
     }
 
     @SuppressWarnings({ "rawtypes", "unchecked" })
     @Test
-    public void testDataChangedWhenNotificationsAreDisabled(){
-        new JavaTestKit(getSystem()) {{
-            final AsyncDataChangeEvent mockChangeEvent = Mockito.mock(AsyncDataChangeEvent.class);
-            final AsyncDataChangeListener mockListener = Mockito.mock(AsyncDataChangeListener.class);
-            final Props props = DataChangeListener.props(mockListener);
-            final ActorRef subject =
-                getSystem().actorOf(props, "testDataChangedNotificationsDisabled");
-
-            subject.tell(new DataChanged(CompositeModel.createTestContext(), mockChangeEvent),
-                    getRef());
-
-            new Within(duration("1 seconds")) {
-                @Override
-                protected void run() {
-                    expectNoMsg();
-
-                    Mockito.verify(mockListener, Mockito.never()).onDataChanged(
-                            Mockito.any(AsyncDataChangeEvent.class));
-                }
-            };
-        }};
+    public void testDataChangedWhenNotificationsAreDisabled() {
+        new JavaTestKit(getSystem()) {
+            {
+                final AsyncDataChangeEvent mockChangeEvent = Mockito.mock(AsyncDataChangeEvent.class);
+                final AsyncDataChangeListener mockListener = Mockito.mock(AsyncDataChangeListener.class);
+                final Props props = DataChangeListener.props(mockListener, TEST_PATH);
+                final ActorRef subject = getSystem().actorOf(props, "testDataChangedNotificationsDisabled");
+
+                subject.tell(new DataChanged(mockChangeEvent), getRef());
+
+                new Within(duration("1 seconds")) {
+                    @Override
+                    protected void run() {
+                        expectNoMsg();
+
+                        Mockito.verify(mockListener, Mockito.never())
+                                .onDataChanged(Mockito.any(AsyncDataChangeEvent.class));
+                    }
+                };
+            }
+        };
     }
 
     @SuppressWarnings({ "rawtypes", "unchecked" })
     @Test
-    public void testDataChangedWithNoSender(){
-        new JavaTestKit(getSystem()) {{
-            final AsyncDataChangeEvent mockChangeEvent = Mockito.mock(AsyncDataChangeEvent.class);
-            final AsyncDataChangeListener mockListener = Mockito.mock(AsyncDataChangeListener.class);
-            final Props props = DataChangeListener.props(mockListener);
-            final ActorRef subject = getSystem().actorOf(props, "testDataChangedWithNoSender");
-
-            getSystem().eventStream().subscribe(getRef(), DeadLetter.class);
-
-            subject.tell(new DataChanged(CompositeModel.createTestContext(), mockChangeEvent),
-                    ActorRef.noSender());
-
-            // Make sure no DataChangedReply is sent to DeadLetters.
-            while(true) {
-                DeadLetter deadLetter;
-                try {
-                    deadLetter = expectMsgClass(duration("1 seconds"), DeadLetter.class);
-                } catch (AssertionError e) {
-                    // Timed out - got no DeadLetter - this is good
-                    break;
+    public void testDataChangedWithNoSender() {
+        new JavaTestKit(getSystem()) {
+            {
+                final AsyncDataChangeEvent mockChangeEvent = Mockito.mock(AsyncDataChangeEvent.class);
+                final AsyncDataChangeListener mockListener = Mockito.mock(AsyncDataChangeListener.class);
+                final Props props = DataChangeListener.props(mockListener, TEST_PATH);
+                final ActorRef subject = getSystem().actorOf(props, "testDataChangedWithNoSender");
+
+                getSystem().eventStream().subscribe(getRef(), DeadLetter.class);
+
+                subject.tell(new DataChanged(mockChangeEvent), ActorRef.noSender());
+
+                // Make sure no DataChangedReply is sent to DeadLetters.
+                while (true) {
+                    DeadLetter deadLetter;
+                    try {
+                        deadLetter = expectMsgClass(duration("1 seconds"), DeadLetter.class);
+                    } catch (AssertionError e) {
+                        // Timed out - got no DeadLetter - this is good
+                        break;
+                    }
+
+                    // We may get DeadLetters for other messages we don't care
+                    // about.
+                    Assert.assertFalse("Unexpected DataChangedReply", deadLetter.message() instanceof DataChangedReply);
                 }
-
-                // We may get DeadLetters for other messages we don't care about.
-                Assert.assertFalse("Unexpected DataChangedReply",
-                        deadLetter.message() instanceof DataChangedReply);
             }
-        }};
+        };
     }
 
     @SuppressWarnings({ "rawtypes", "unchecked" })
     @Test
-    public void testDataChangedWithListenerRuntimeEx(){
-        new JavaTestKit(getSystem()) {{
-            AsyncDataChangeEvent mockChangeEvent1 = Mockito.mock(AsyncDataChangeEvent.class);
-            AsyncDataChangeEvent mockChangeEvent2 = Mockito.mock(AsyncDataChangeEvent.class);
-            AsyncDataChangeEvent mockChangeEvent3 = Mockito.mock(AsyncDataChangeEvent.class);
+    public void testDataChangedWithListenerRuntimeEx() {
+        new JavaTestKit(getSystem()) {
+            {
+                final AsyncDataChangeEvent mockChangeEvent1 = Mockito.mock(AsyncDataChangeEvent.class);
+                final AsyncDataChangeEvent mockChangeEvent2 = Mockito.mock(AsyncDataChangeEvent.class);
+                final AsyncDataChangeEvent mockChangeEvent3 = Mockito.mock(AsyncDataChangeEvent.class);
 
-            AsyncDataChangeListener mockListener = Mockito.mock(AsyncDataChangeListener.class);
-            Mockito.doThrow(new RuntimeException("mock")).when(mockListener).onDataChanged(mockChangeEvent2);
+                AsyncDataChangeListener mockListener = Mockito.mock(AsyncDataChangeListener.class);
+                Mockito.doThrow(new RuntimeException("mock")).when(mockListener).onDataChanged(mockChangeEvent2);
 
-            Props props = DataChangeListener.props(mockListener);
-            ActorRef subject = getSystem().actorOf(props, "testDataChangedWithListenerRuntimeEx");
+                Props props = DataChangeListener.props(mockListener, TEST_PATH);
+                ActorRef subject = getSystem().actorOf(props, "testDataChangedWithListenerRuntimeEx");
 
-            // Let the DataChangeListener know that notifications should be enabled
-            subject.tell(new EnableNotification(true), getRef());
+                // Let the DataChangeListener know that notifications should be
+                // enabled
+                subject.tell(new EnableNotification(true), getRef());
 
-            SchemaContext schemaContext = CompositeModel.createTestContext();
+                subject.tell(new DataChanged(mockChangeEvent1), getRef());
+                expectMsgClass(DataChangedReply.class);
 
-            subject.tell(new DataChanged(schemaContext, mockChangeEvent1),getRef());
-            expectMsgClass(DataChangedReply.class);
+                subject.tell(new DataChanged(mockChangeEvent2), getRef());
+                expectMsgClass(DataChangedReply.class);
 
-            subject.tell(new DataChanged(schemaContext, mockChangeEvent2),getRef());
-            expectMsgClass(DataChangedReply.class);
+                subject.tell(new DataChanged(mockChangeEvent3), getRef());
+                expectMsgClass(DataChangedReply.class);
 
-            subject.tell(new DataChanged(schemaContext, mockChangeEvent3),getRef());
-            expectMsgClass(DataChangedReply.class);
-
-            Mockito.verify(mockListener).onDataChanged(mockChangeEvent1);
-            Mockito.verify(mockListener).onDataChanged(mockChangeEvent2);
-            Mockito.verify(mockListener).onDataChanged(mockChangeEvent3);
-        }};
+                Mockito.verify(mockListener).onDataChanged(mockChangeEvent1);
+                Mockito.verify(mockListener).onDataChanged(mockChangeEvent2);
+                Mockito.verify(mockListener).onDataChanged(mockChangeEvent3);
+            }
+        };
     }
 }