2 * Copyright © 2019 FRINX s.r.o. 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.restconf.nb.rfc8040.streams;
10 import static org.junit.jupiter.api.Assertions.assertInstanceOf;
11 import static org.junit.jupiter.api.Assertions.assertNull;
12 import static org.mockito.Mockito.doReturn;
13 import static org.mockito.Mockito.verify;
16 import java.util.concurrent.ScheduledExecutorService;
17 import org.eclipse.jetty.websocket.servlet.ServletUpgradeRequest;
18 import org.eclipse.jetty.websocket.servlet.ServletUpgradeResponse;
19 import org.junit.jupiter.api.BeforeEach;
20 import org.junit.jupiter.api.Test;
21 import org.junit.jupiter.api.extension.ExtendWith;
22 import org.mockito.Mock;
23 import org.mockito.junit.jupiter.MockitoExtension;
24 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
25 import org.opendaylight.yang.gen.v1.urn.sal.restconf.event.subscription.rev140708.CreateDataChangeEventSubscriptionInput1.Scope;
26 import org.opendaylight.yang.gen.v1.urn.sal.restconf.event.subscription.rev140708.NotificationOutputTypeGrouping;
27 import org.opendaylight.yangtools.yang.common.QName;
28 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
30 @ExtendWith(MockitoExtension.class)
31 class WebSocketFactoryTest extends AbstractNotificationListenerTest {
32 private static final String REGISTERED_STREAM_NAME = "data-change-event-subscription/"
33 + "toaster:toaster/datastore=CONFIGURATION/scope=SUBTREE/JSON";
35 private final ListenersBroker listenersBroker = new ListenersBroker.ServerSentEvents();
38 private ScheduledExecutorService execService;
40 private ServletUpgradeRequest upgradeRequest;
42 private ServletUpgradeResponse upgradeResponse;
44 private WebSocketFactory webSocketFactory;
47 void prepareListenersBroker() {
48 webSocketFactory = new WebSocketFactory(execService, listenersBroker, 5000, 2000);
50 listenersBroker.registerDataChangeListener(MODEL_CONTEXT, LogicalDatastoreType.CONFIGURATION,
51 YangInstanceIdentifier.of(QName.create("http://netconfcentral.org/ns/toaster", "2009-11-20", "toaster")),
52 Scope.SUBTREE, NotificationOutputTypeGrouping.NotificationOutputType.JSON);
56 void createWebSocketSuccessfully() {
57 doReturn(URI.create("https://localhost:8181/rests/streams/" + REGISTERED_STREAM_NAME))
58 .when(upgradeRequest).getRequestURI();
60 assertInstanceOf(WebSocketSessionHandler.class,
61 webSocketFactory.createWebSocket(upgradeRequest, upgradeResponse));
62 verify(upgradeResponse).setSuccess(true);
63 verify(upgradeResponse).setStatusCode(101);
67 void createWebSocketUnsuccessfully() {
68 doReturn(URI.create("https://localhost:8181/rests/streams/" + REGISTERED_STREAM_NAME + "/toasterStatus"))
69 .when(upgradeRequest).getRequestURI();
71 assertNull(webSocketFactory.createWebSocket(upgradeRequest, upgradeResponse));
72 verify(upgradeResponse).setSuccess(false);
73 verify(upgradeResponse).setStatusCode(404);