2 * Copyright (c) 2017 Pantheon Technologies s.r.o. 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.access.client;
10 import static org.junit.Assert.assertNull;
11 import static org.mockito.ArgumentMatchers.any;
12 import static org.mockito.ArgumentMatchers.same;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.mock;
15 import static org.mockito.Mockito.verify;
17 import java.util.function.Consumer;
18 import org.junit.Test;
19 import org.opendaylight.controller.cluster.access.ABIVersion;
20 import org.opendaylight.controller.cluster.access.commands.ModifyTransactionRequestBuilder;
21 import org.opendaylight.controller.cluster.access.commands.TransactionWrite;
22 import org.opendaylight.controller.cluster.access.concepts.LocalHistoryIdentifier;
23 import org.opendaylight.controller.cluster.access.concepts.Request;
24 import org.opendaylight.controller.cluster.access.concepts.RequestException;
25 import org.opendaylight.controller.cluster.access.concepts.Response;
26 import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
27 import org.opendaylight.controller.cluster.messaging.MessageSlice;
28 import org.opendaylight.yangtools.yang.common.QName;
29 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
30 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
31 import org.opendaylight.yangtools.yang.data.spi.node.ImmutableNodes;
33 public class ConnectedClientConnectionTest
34 extends AbstractClientConnectionTest<ConnectedClientConnection<BackendInfo>, BackendInfo> {
37 public void testCheckTimeoutConnectionTimedout() {
38 final Consumer<Response<?, ?>> callback = mock(Consumer.class);
39 connection.sendRequest(createRequest(replyToProbe.ref()), callback);
40 final long now = context.ticker().read() + ConnectedClientConnection.DEFAULT_BACKEND_ALIVE_TIMEOUT_NANOS;
41 assertNull(connection.checkTimeout(now));
45 protected ConnectedClientConnection<BackendInfo> createConnection() {
46 final BackendInfo backend = new BackendInfo(backendProbe.ref(), "test", 0L, ABIVersion.current(), 10);
47 final ConnectingClientConnection<BackendInfo> connectingConn = new ConnectingClientConnection<>(context, 0L,
49 return new ConnectedClientConnection<>(connectingConn, backend);
54 public void testReconnectConnection() {
55 final ClientActorBehavior<BackendInfo> behavior = mock(ClientActorBehavior.class);
56 connection.lockedReconnect(behavior, mock(RequestException.class));
57 verify(behavior).reconnectConnection(same(connection), any(ReconnectingClientConnection.class));
60 @SuppressWarnings("unchecked")
62 public void testSendSliceableMessageRequest() {
63 final ClientActorConfig config = AccessClientUtil.newMockClientActorConfig();
64 doReturn(5).when(config).getMaximumMessageSliceSize();
65 context = new ClientActorContext(contextProbe.ref(), PERSISTENCE_ID, system, CLIENT_ID, config);
66 connection = createConnection();
68 final Consumer<Response<?, ?>> callback = mock(Consumer.class);
70 final TransactionIdentifier identifier =
71 new TransactionIdentifier(new LocalHistoryIdentifier(CLIENT_ID, 0L), 0L);
72 ModifyTransactionRequestBuilder reqBuilder =
73 new ModifyTransactionRequestBuilder(identifier, replyToProbe.ref());
74 reqBuilder.addModification(new TransactionWrite(YangInstanceIdentifier.of(),
75 ImmutableNodes.newContainerBuilder()
76 .withNodeIdentifier(new NodeIdentifier(QName.create("namespace", "localName")))
78 reqBuilder.setSequence(0L);
79 final Request<?, ?> request = reqBuilder.build();
80 connection.sendRequest(request, callback);
82 backendProbe.expectMsgClass(MessageSlice.class);