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.mockito.ArgumentMatchers.any;
11 import static org.mockito.ArgumentMatchers.same;
12 import static org.mockito.Mockito.doReturn;
13 import static org.mockito.Mockito.mock;
14 import static org.mockito.Mockito.verify;
16 import java.util.Optional;
17 import java.util.function.Consumer;
18 import org.junit.Assert;
19 import org.junit.Test;
20 import org.opendaylight.controller.cluster.access.ABIVersion;
21 import org.opendaylight.controller.cluster.access.commands.ModifyTransactionRequestBuilder;
22 import org.opendaylight.controller.cluster.access.commands.TransactionWrite;
23 import org.opendaylight.controller.cluster.access.concepts.LocalHistoryIdentifier;
24 import org.opendaylight.controller.cluster.access.concepts.Request;
25 import org.opendaylight.controller.cluster.access.concepts.RequestException;
26 import org.opendaylight.controller.cluster.access.concepts.Response;
27 import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
28 import org.opendaylight.controller.cluster.messaging.MessageSlice;
29 import org.opendaylight.yangtools.yang.common.QName;
30 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
31 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
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 final Optional<Long> timeout = connection.checkTimeout(now);
42 Assert.assertNull(timeout);
46 protected ConnectedClientConnection<BackendInfo> createConnection() {
47 final BackendInfo backend = new BackendInfo(backendProbe.ref(), "test", 0L, ABIVersion.BORON, 10);
48 final ConnectingClientConnection<BackendInfo> connectingConn = new ConnectingClientConnection<>(context, 0L,
50 return new ConnectedClientConnection<>(connectingConn, backend);
55 public void testReconnectConnection() {
56 final ClientActorBehavior<BackendInfo> behavior = mock(ClientActorBehavior.class);
57 connection.lockedReconnect(behavior, mock(RequestException.class));
58 verify(behavior).reconnectConnection(same(connection), any(ReconnectingClientConnection.class));
61 @SuppressWarnings("unchecked")
63 public void testSendSliceableMessageRequest() {
64 final ClientActorConfig config = AccessClientUtil.newMockClientActorConfig();
65 doReturn(5).when(config).getMaximumMessageSliceSize();
66 context = new ClientActorContext(contextProbe.ref(), PERSISTENCE_ID, system, CLIENT_ID, config);
67 connection = createConnection();
69 final Consumer<Response<?, ?>> callback = mock(Consumer.class);
71 final TransactionIdentifier identifier =
72 new TransactionIdentifier(new LocalHistoryIdentifier(CLIENT_ID, 0L), 0L);
73 ModifyTransactionRequestBuilder reqBuilder =
74 new ModifyTransactionRequestBuilder(identifier, replyToProbe.ref());
75 reqBuilder.addModification(new TransactionWrite(YangInstanceIdentifier.EMPTY, Builders.containerBuilder()
76 .withNodeIdentifier(YangInstanceIdentifier.NodeIdentifier.create(
77 QName.create("namespace", "localName"))).build()));
78 reqBuilder.setSequence(0L);
79 final Request<?, ?> request = reqBuilder.build();
80 connection.sendRequest(request, callback);
82 backendProbe.expectMsgClass(MessageSlice.class);