2 * Copyright (c) 2017 Inocybe Technologies 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.actors;
10 import static org.mockito.Mockito.mock;
11 import static org.mockito.Mockito.timeout;
12 import static org.mockito.Mockito.verify;
14 import akka.actor.ActorRef;
15 import akka.testkit.javadsl.TestKit;
16 import java.time.Duration;
17 import org.junit.Before;
18 import org.junit.Test;
19 import org.mockito.Mock;
20 import org.mockito.MockitoAnnotations;
21 import org.opendaylight.controller.cluster.datastore.AbstractActorTest;
22 import org.opendaylight.controller.cluster.datastore.messages.CloseDataTreeNotificationListenerRegistration;
23 import org.opendaylight.controller.cluster.datastore.messages.CloseDataTreeNotificationListenerRegistrationReply;
24 import org.opendaylight.yangtools.concepts.ListenerRegistration;
26 public class DataTreeNotificationListenerRegistrationActorTest extends AbstractActorTest {
28 private ListenerRegistration<?> mockListenerReg;
31 private Runnable mockOnClose;
37 MockitoAnnotations.initMocks(this);
38 DataTreeNotificationListenerRegistrationActor.killDelay = 100;
39 kit = new TestKit(getSystem());
43 public void testOnReceiveCloseListenerRegistrationAfterSetRegistration() {
44 final ActorRef subject = getSystem().actorOf(DataTreeNotificationListenerRegistrationActor.props(),
45 "testOnReceiveCloseListenerRegistrationAfterSetRegistration");
48 subject.tell(new DataTreeNotificationListenerRegistrationActor.SetRegistration(mockListenerReg,
49 mockOnClose), ActorRef.noSender());
50 subject.tell(CloseDataTreeNotificationListenerRegistration.getInstance(), kit.getRef());
52 kit.expectMsgClass(Duration.ofSeconds(5), CloseDataTreeNotificationListenerRegistrationReply.class);
54 verify(mockListenerReg, timeout(5000)).close();
55 verify(mockOnClose, timeout(5000)).run();
57 kit.expectTerminated(Duration.ofSeconds(5), subject);
61 public void testOnReceiveCloseListenerRegistrationBeforeSetRegistration() {
62 final ActorRef subject = getSystem().actorOf(DataTreeNotificationListenerRegistrationActor.props(),
63 "testOnReceiveSetRegistrationAfterPriorClose");
66 subject.tell(CloseDataTreeNotificationListenerRegistration.getInstance(), kit.getRef());
67 kit.expectMsgClass(Duration.ofSeconds(5), CloseDataTreeNotificationListenerRegistrationReply.class);
69 subject.tell(new DataTreeNotificationListenerRegistrationActor.SetRegistration(mockListenerReg,
70 mockOnClose), ActorRef.noSender());
72 verify(mockListenerReg, timeout(5000)).close();
73 verify(mockOnClose, timeout(5000)).run();
75 kit.expectTerminated(Duration.ofSeconds(5), subject);
79 public void testOnReceiveSetRegistrationAfterPriorClose() {
80 DataTreeNotificationListenerRegistrationActor.killDelay = 1000;
81 final ListenerRegistration<?> mockListenerReg2 = mock(ListenerRegistration.class);
82 final Runnable mockOnClose2 = mock(Runnable.class);
84 final ActorRef subject = getSystem().actorOf(DataTreeNotificationListenerRegistrationActor.props(),
85 "testOnReceiveSetRegistrationAfterPriorClose");
88 subject.tell(new DataTreeNotificationListenerRegistrationActor.SetRegistration(mockListenerReg,
89 mockOnClose), ActorRef.noSender());
90 subject.tell(CloseDataTreeNotificationListenerRegistration.getInstance(), ActorRef.noSender());
91 subject.tell(new DataTreeNotificationListenerRegistrationActor.SetRegistration(mockListenerReg2,
92 mockOnClose2), ActorRef.noSender());
94 verify(mockListenerReg, timeout(5000)).close();
95 verify(mockOnClose, timeout(5000)).run();
96 verify(mockListenerReg2, timeout(5000)).close();
97 verify(mockOnClose2, timeout(5000)).run();
99 kit.expectTerminated(Duration.ofSeconds(5), subject);