2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.controller.cluster.datastore;
10 import akka.actor.ActorRef;
11 import akka.actor.DeadLetter;
12 import akka.actor.Props;
13 import akka.testkit.JavaTestKit;
14 import com.google.common.collect.ImmutableList;
15 import org.junit.Assert;
16 import org.junit.Test;
17 import org.mockito.Matchers;
18 import org.mockito.Mockito;
19 import org.opendaylight.controller.cluster.datastore.messages.DataTreeChanged;
20 import org.opendaylight.controller.cluster.datastore.messages.DataTreeChangedReply;
21 import org.opendaylight.controller.cluster.datastore.messages.EnableNotification;
22 import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeChangeListener;
23 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidate;
25 public class DataTreeChangeListenerActorTest extends AbstractActorTest {
28 public void testDataChangedWhenNotificationsAreEnabled() {
29 new JavaTestKit(getSystem()) {
31 final DataTreeCandidate mockTreeCandidate = Mockito.mock(DataTreeCandidate.class);
32 final ImmutableList<DataTreeCandidate> mockCandidates = ImmutableList.of(mockTreeCandidate);
33 final DOMDataTreeChangeListener mockListener = Mockito.mock(DOMDataTreeChangeListener.class);
34 final Props props = DataTreeChangeListenerActor.props(mockListener);
35 final ActorRef subject = getSystem().actorOf(props, "testDataTreeChangedNotificationsEnabled");
37 // Let the DataChangeListener know that notifications should be
39 subject.tell(new EnableNotification(true), getRef());
41 subject.tell(new DataTreeChanged(mockCandidates), getRef());
43 expectMsgClass(DataTreeChangedReply.class);
45 Mockito.verify(mockListener).onDataTreeChanged(mockCandidates);
51 public void testDataChangedWhenNotificationsAreDisabled() {
52 new JavaTestKit(getSystem()) {
54 final DataTreeCandidate mockTreeCandidate = Mockito.mock(DataTreeCandidate.class);
55 final ImmutableList<DataTreeCandidate> mockCandidates = ImmutableList.of(mockTreeCandidate);
56 final DOMDataTreeChangeListener mockListener = Mockito.mock(DOMDataTreeChangeListener.class);
57 final Props props = DataTreeChangeListenerActor.props(mockListener);
58 final ActorRef subject = getSystem().actorOf(props, "testDataTreeChangedNotificationsDisabled");
60 subject.tell(new DataTreeChanged(mockCandidates), getRef());
62 new Within(duration("1 seconds")) {
64 protected void run() {
67 Mockito.verify(mockListener, Mockito.never())
68 .onDataTreeChanged(Matchers.anyCollectionOf(DataTreeCandidate.class));
76 public void testDataChangedWithNoSender() {
77 new JavaTestKit(getSystem()) {
79 final DataTreeCandidate mockTreeCandidate = Mockito.mock(DataTreeCandidate.class);
80 final ImmutableList<DataTreeCandidate> mockCandidates = ImmutableList.of(mockTreeCandidate);
81 final DOMDataTreeChangeListener mockListener = Mockito.mock(DOMDataTreeChangeListener.class);
82 final Props props = DataTreeChangeListenerActor.props(mockListener);
83 final ActorRef subject = getSystem().actorOf(props, "testDataTreeChangedWithNoSender");
85 getSystem().eventStream().subscribe(getRef(), DeadLetter.class);
87 subject.tell(new DataTreeChanged(mockCandidates), ActorRef.noSender());
89 // Make sure no DataChangedReply is sent to DeadLetters.
91 DeadLetter deadLetter;
93 deadLetter = expectMsgClass(duration("1 seconds"), DeadLetter.class);
94 } catch (AssertionError e) {
95 // Timed out - got no DeadLetter - this is good
99 // We may get DeadLetters for other messages we don't care
101 Assert.assertFalse("Unexpected DataTreeChangedReply",
102 deadLetter.message() instanceof DataTreeChangedReply);
109 public void testDataChangedWithListenerRuntimeEx() {
110 new JavaTestKit(getSystem()) {
112 final DataTreeCandidate mockTreeCandidate1 = Mockito.mock(DataTreeCandidate.class);
113 final ImmutableList<DataTreeCandidate> mockCandidates1 = ImmutableList.of(mockTreeCandidate1);
114 final DataTreeCandidate mockTreeCandidate2 = Mockito.mock(DataTreeCandidate.class);
115 final ImmutableList<DataTreeCandidate> mockCandidates2 = ImmutableList.of(mockTreeCandidate2);
116 final DataTreeCandidate mockTreeCandidate3 = Mockito.mock(DataTreeCandidate.class);
117 final ImmutableList<DataTreeCandidate> mockCandidates3 = ImmutableList.of(mockTreeCandidate3);
119 final DOMDataTreeChangeListener mockListener = Mockito.mock(DOMDataTreeChangeListener.class);
120 Mockito.doThrow(new RuntimeException("mock")).when(mockListener).onDataTreeChanged(mockCandidates2);
122 Props props = DataTreeChangeListenerActor.props(mockListener);
123 ActorRef subject = getSystem().actorOf(props, "testDataTreeChangedWithListenerRuntimeEx");
125 // Let the DataChangeListener know that notifications should be
127 subject.tell(new EnableNotification(true), getRef());
129 subject.tell(new DataTreeChanged(mockCandidates1), getRef());
130 expectMsgClass(DataTreeChangedReply.class);
132 subject.tell(new DataTreeChanged(mockCandidates2), getRef());
133 expectMsgClass(DataTreeChangedReply.class);
135 subject.tell(new DataTreeChanged(mockCandidates3), getRef());
136 expectMsgClass(DataTreeChangedReply.class);
138 Mockito.verify(mockListener).onDataTreeChanged(mockCandidates1);
139 Mockito.verify(mockListener).onDataTreeChanged(mockCandidates2);
140 Mockito.verify(mockListener).onDataTreeChanged(mockCandidates3);