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.doReturn;
12 import static org.mockito.Mockito.timeout;
13 import static org.mockito.Mockito.verify;
14 import static org.mockito.Mockito.when;
15 import static org.opendaylight.mdsal.common.api.CommitInfo.emptyFluentFuture;
16 import static org.opendaylight.netconf.topology.singleton.impl.actors.ReadTransactionActorTestAdapter.NODE;
17 import static org.opendaylight.netconf.topology.singleton.impl.actors.ReadTransactionActorTestAdapter.PATH;
18 import static org.opendaylight.netconf.topology.singleton.impl.actors.ReadTransactionActorTestAdapter.STORE;
19 import static org.opendaylight.netconf.topology.singleton.impl.actors.ReadTransactionActorTestAdapter.TIMEOUT;
21 import akka.actor.ActorRef;
22 import akka.actor.ActorSystem;
23 import akka.actor.Status.Failure;
24 import akka.actor.Status.Success;
25 import akka.testkit.TestProbe;
26 import org.junit.Test;
27 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
28 import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction;
29 import org.opendaylight.netconf.topology.singleton.messages.NormalizedNodeMessage;
30 import org.opendaylight.netconf.topology.singleton.messages.transactions.CancelRequest;
31 import org.opendaylight.netconf.topology.singleton.messages.transactions.DeleteRequest;
32 import org.opendaylight.netconf.topology.singleton.messages.transactions.MergeRequest;
33 import org.opendaylight.netconf.topology.singleton.messages.transactions.PutRequest;
34 import org.opendaylight.netconf.topology.singleton.messages.transactions.SubmitRequest;
35 import org.opendaylight.yangtools.util.concurrent.FluentFutures;
36 import org.opendaylight.yangtools.yang.common.RpcError;
37 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
40 * Adapter for write transaction tests.
42 * @author Thomas Pantelis
44 public abstract class WriteTransactionActorTestAdapter {
45 private DOMDataWriteTransaction mockWriteTx;
46 private TestProbe probe;
47 private ActorRef actorRef;
48 private ActorSystem system;
50 public void init(DOMDataWriteTransaction inMockWriteTx, ActorSystem inSystem, ActorRef inActorRef) {
51 this.mockWriteTx = inMockWriteTx;
52 this.probe = TestProbe.apply(inSystem);
53 this.actorRef = inActorRef;
54 this.system = inSystem;
58 public void testPut() {
59 final NormalizedNodeMessage normalizedNodeMessage = new NormalizedNodeMessage(PATH, NODE);
60 actorRef.tell(new PutRequest(STORE, normalizedNodeMessage), probe.ref());
61 verify(mockWriteTx).put(STORE, PATH, NODE);
65 public void testMerge() {
66 final NormalizedNodeMessage normalizedNodeMessage = new NormalizedNodeMessage(PATH, NODE);
67 actorRef.tell(new MergeRequest(STORE, normalizedNodeMessage), probe.ref());
68 verify(mockWriteTx).merge(STORE, PATH, NODE);
72 public void testDelete() {
73 actorRef.tell(new DeleteRequest(STORE, PATH), probe.ref());
74 verify(mockWriteTx).delete(STORE, PATH);
78 public void testCancel() {
79 when(mockWriteTx.cancel()).thenReturn(true);
80 actorRef.tell(new CancelRequest(), probe.ref());
82 verify(mockWriteTx).cancel();
83 probe.expectMsg(true);
87 public void testSubmit() {
88 doReturn(emptyFluentFuture()).when(mockWriteTx).commit();
89 actorRef.tell(new SubmitRequest(), probe.ref());
91 verify(mockWriteTx).commit();
92 probe.expectMsgClass(Success.class);
96 public void testSubmitFail() {
97 final RpcError rpcError =
98 RpcResultBuilder.newError(RpcError.ErrorType.APPLICATION, "fail", "fail");
99 final TransactionCommitFailedException cause = new TransactionCommitFailedException("fail", rpcError);
100 when(mockWriteTx.commit()).thenReturn(FluentFutures.immediateFailedFluentFuture(cause));
101 actorRef.tell(new SubmitRequest(), probe.ref());
103 verify(mockWriteTx).commit();
104 final Failure response = probe.expectMsgClass(Failure.class);
105 assertEquals(cause, response.cause());
109 public void testIdleTimeout() {
110 final TestProbe testProbe = new TestProbe(system);
111 testProbe.watch(actorRef);
112 verify(mockWriteTx, timeout(3000)).cancel();
113 testProbe.expectTerminated(actorRef, TIMEOUT.duration());