git.opendaylight Code Review - controller.git/blob

   1 /*
   2  * Copyright (c) 2017 Inocybe Technologies and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.controller.cluster.datastore.actors;
   9
  10 import static org.mockito.Mockito.mock;
  11 import static org.mockito.Mockito.timeout;
  12 import static org.mockito.Mockito.verify;
  13
  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.junit.runner.RunWith;
  20 import org.mockito.Mock;
  21 import org.mockito.junit.MockitoJUnitRunner;
  22 import org.opendaylight.controller.cluster.datastore.AbstractActorTest;
  23 import org.opendaylight.controller.cluster.datastore.messages.CloseDataTreeNotificationListenerRegistration;
  24 import org.opendaylight.controller.cluster.datastore.messages.CloseDataTreeNotificationListenerRegistrationReply;
  25 import org.opendaylight.yangtools.concepts.Registration;
  26
  27 @RunWith(MockitoJUnitRunner.StrictStubs.class)
  28 public class DataTreeNotificationListenerRegistrationActorTest extends AbstractActorTest {
  29     @Mock
  30     private Registration mockListenerReg;
  31
  32     @Mock
  33     private Runnable mockOnClose;
  34
  35     private TestKit kit;
  36
  37     @Before
  38     public void setup() {
  39         DataTreeNotificationListenerRegistrationActor.killDelay = 100;
  40         kit = new TestKit(getSystem());
  41     }
  42
  43     @Test
  44     public void testOnReceiveCloseListenerRegistrationAfterSetRegistration() {
  45         final ActorRef subject = getSystem().actorOf(DataTreeNotificationListenerRegistrationActor.props(),
  46                 "testOnReceiveCloseListenerRegistrationAfterSetRegistration");
  47         kit.watch(subject);
  48
  49         subject.tell(new DataTreeNotificationListenerRegistrationActor.SetRegistration(mockListenerReg,
  50             mockOnClose), ActorRef.noSender());
  51         subject.tell(CloseDataTreeNotificationListenerRegistration.getInstance(), kit.getRef());
  52
  53         kit.expectMsgClass(Duration.ofSeconds(5), CloseDataTreeNotificationListenerRegistrationReply.class);
  54
  55         verify(mockListenerReg, timeout(5000)).close();
  56         verify(mockOnClose, timeout(5000)).run();
  57
  58         kit.expectTerminated(Duration.ofSeconds(5), subject);
  59     }
  60
  61     @Test
  62     public void testOnReceiveCloseListenerRegistrationBeforeSetRegistration() {
  63         final ActorRef subject = getSystem().actorOf(DataTreeNotificationListenerRegistrationActor.props(),
  64                 "testOnReceiveSetRegistrationAfterPriorClose");
  65         kit.watch(subject);
  66
  67         subject.tell(CloseDataTreeNotificationListenerRegistration.getInstance(), kit.getRef());
  68         kit.expectMsgClass(Duration.ofSeconds(5), CloseDataTreeNotificationListenerRegistrationReply.class);
  69
  70         subject.tell(new DataTreeNotificationListenerRegistrationActor.SetRegistration(mockListenerReg,
  71             mockOnClose), ActorRef.noSender());
  72
  73         verify(mockListenerReg, timeout(5000)).close();
  74         verify(mockOnClose, timeout(5000)).run();
  75
  76         kit.expectTerminated(Duration.ofSeconds(5), subject);
  77     }
  78
  79     @Test
  80     public void testOnReceiveSetRegistrationAfterPriorClose() {
  81         DataTreeNotificationListenerRegistrationActor.killDelay = 1000;
  82         final Registration mockListenerReg2 = mock(Registration.class);
  83         final Runnable mockOnClose2 = mock(Runnable.class);
  84
  85         final ActorRef subject = getSystem().actorOf(DataTreeNotificationListenerRegistrationActor.props(),
  86             "testOnReceiveSetRegistrationAfterPriorClose");
  87         kit.watch(subject);
  88
  89         subject.tell(new DataTreeNotificationListenerRegistrationActor.SetRegistration(mockListenerReg,
  90             mockOnClose), ActorRef.noSender());
  91         subject.tell(CloseDataTreeNotificationListenerRegistration.getInstance(), ActorRef.noSender());
  92         subject.tell(new DataTreeNotificationListenerRegistrationActor.SetRegistration(mockListenerReg2,
  93             mockOnClose2), ActorRef.noSender());
  94
  95         verify(mockListenerReg, timeout(5000)).close();
  96         verify(mockOnClose, timeout(5000)).run();
  97         verify(mockListenerReg2, timeout(5000)).close();
  98         verify(mockOnClose2, timeout(5000)).run();
  99
 100         kit.expectTerminated(Duration.ofSeconds(5), subject);
 101     }
 102 }