2 * Copyright (c) 2018 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.netconf.topology.singleton.impl.actors;
10 import static org.junit.Assert.assertEquals;
11 import static org.mockito.Mockito.timeout;
12 import static org.mockito.Mockito.verify;
13 import static org.mockito.Mockito.when;
14 import static org.opendaylight.netconf.topology.singleton.impl.actors.ReadTransactionActorTestAdapter.NODE;
15 import static org.opendaylight.netconf.topology.singleton.impl.actors.ReadTransactionActorTestAdapter.PATH;
16 import static org.opendaylight.netconf.topology.singleton.impl.actors.ReadTransactionActorTestAdapter.STORE;
17 import static org.opendaylight.netconf.topology.singleton.impl.actors.ReadTransactionActorTestAdapter.TIMEOUT;
19 import akka.actor.ActorRef;
20 import akka.actor.ActorSystem;
21 import akka.actor.Status.Failure;
22 import akka.actor.Status.Success;
23 import akka.testkit.TestProbe;
24 import com.google.common.util.concurrent.Futures;
25 import org.junit.Test;
26 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
27 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
28 import org.opendaylight.netconf.topology.singleton.messages.NormalizedNodeMessage;
29 import org.opendaylight.netconf.topology.singleton.messages.transactions.CancelRequest;
30 import org.opendaylight.netconf.topology.singleton.messages.transactions.DeleteRequest;
31 import org.opendaylight.netconf.topology.singleton.messages.transactions.MergeRequest;
32 import org.opendaylight.netconf.topology.singleton.messages.transactions.PutRequest;
33 import org.opendaylight.netconf.topology.singleton.messages.transactions.SubmitRequest;
34 import org.opendaylight.yangtools.yang.common.RpcError;
35 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
38 * Adapter for write transaction tests.
40 * @author Thomas Pantelis
42 public abstract class WriteTransactionActorTestAdapter {
43 private DOMDataWriteTransaction mockWriteTx;
44 private TestProbe probe;
45 private ActorRef actorRef;
46 private ActorSystem system;
48 public void init(DOMDataWriteTransaction inMockWriteTx, ActorSystem inSystem, ActorRef inActorRef) {
49 this.mockWriteTx = inMockWriteTx;
50 this.probe = TestProbe.apply(inSystem);
51 this.actorRef = inActorRef;
52 this.system = inSystem;
56 public void testPut() {
57 final NormalizedNodeMessage normalizedNodeMessage = new NormalizedNodeMessage(PATH, NODE);
58 actorRef.tell(new PutRequest(STORE, normalizedNodeMessage), probe.ref());
59 verify(mockWriteTx).put(STORE, PATH, NODE);
63 public void testMerge() {
64 final NormalizedNodeMessage normalizedNodeMessage = new NormalizedNodeMessage(PATH, NODE);
65 actorRef.tell(new MergeRequest(STORE, normalizedNodeMessage), probe.ref());
66 verify(mockWriteTx).merge(STORE, PATH, NODE);
70 public void testDelete() {
71 actorRef.tell(new DeleteRequest(STORE, PATH), probe.ref());
72 verify(mockWriteTx).delete(STORE, PATH);
76 public void testCancel() throws Exception {
77 when(mockWriteTx.cancel()).thenReturn(true);
78 actorRef.tell(new CancelRequest(), probe.ref());
80 verify(mockWriteTx).cancel();
81 probe.expectMsg(true);
85 public void testSubmit() throws Exception {
86 when(mockWriteTx.submit()).thenReturn(Futures.immediateCheckedFuture(null));
87 actorRef.tell(new SubmitRequest(), probe.ref());
89 verify(mockWriteTx).submit();
90 probe.expectMsgClass(Success.class);
94 public void testSubmitFail() throws Exception {
95 final RpcError rpcError =
96 RpcResultBuilder.newError(RpcError.ErrorType.APPLICATION, "fail", "fail");
97 final TransactionCommitFailedException cause = new TransactionCommitFailedException("fail", rpcError);
98 when(mockWriteTx.submit()).thenReturn(Futures.immediateFailedCheckedFuture(cause));
99 actorRef.tell(new SubmitRequest(), probe.ref());
101 verify(mockWriteTx).submit();
102 final Failure response = probe.expectMsgClass(Failure.class);
103 assertEquals(cause, response.cause());
107 public void testIdleTimeout() throws Exception {
108 final TestProbe testProbe = new TestProbe(system);
109 testProbe.watch(actorRef);
110 verify(mockWriteTx, timeout(3000)).cancel();
111 testProbe.expectTerminated(actorRef, TIMEOUT.duration());