import akka.testkit.TestActorRef;
import akka.testkit.javadsl.TestKit;
import com.google.common.base.Throwables;
+import java.time.Duration;
import java.util.concurrent.TimeUnit;
import org.junit.Before;
import org.junit.Test;
import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeModification;
import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableContainerNodeBuilder;
-import org.opendaylight.yangtools.yang.model.api.SchemaContext;
+import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
public class ShardTransactionTest extends AbstractActorTest {
private static final ShardIdentifier SHARD_IDENTIFIER =
ShardIdentifier.create("inventory", MEMBER_NAME, "config");
- private static final SchemaContext TEST_MODEL = TestModel.createTestContext();
+ private static final EffectiveModelContext TEST_MODEL = TestModel.createTestContext();
private DatastoreContext datastoreContext = DatastoreContext.newBuilder().persistent(false).build();
}
private void testOnReceiveReadData(final ActorRef transaction) {
- transaction.tell(new ReadData(YangInstanceIdentifier.EMPTY, DataStoreVersions.CURRENT_VERSION),
+ transaction.tell(new ReadData(YangInstanceIdentifier.empty(), DataStoreVersions.CURRENT_VERSION),
testKit.getRef());
- ReadDataReply reply = testKit.expectMsgClass(testKit.duration("5 seconds"), ReadDataReply.class);
+ ReadDataReply reply = testKit.expectMsgClass(Duration.ofSeconds(5), ReadDataReply.class);
assertNotNull(reply.getNormalizedNode());
}
private void testOnReceiveReadDataWhenDataNotFound(final ActorRef transaction) {
transaction.tell(new ReadData(TestModel.TEST_PATH, DataStoreVersions.CURRENT_VERSION), testKit.getRef());
- ReadDataReply reply = testKit.expectMsgClass(testKit.duration("5 seconds"), ReadDataReply.class);
+ ReadDataReply reply = testKit.expectMsgClass(Duration.ofSeconds(5), ReadDataReply.class);
assertNull(reply.getNormalizedNode());
}
}
private void testOnReceiveDataExistsPositive(final ActorRef transaction) {
- transaction.tell(new DataExists(YangInstanceIdentifier.EMPTY, DataStoreVersions.CURRENT_VERSION),
+ transaction.tell(new DataExists(YangInstanceIdentifier.empty(), DataStoreVersions.CURRENT_VERSION),
testKit.getRef());
- DataExistsReply reply = testKit.expectMsgClass(testKit.duration("5 seconds"), DataExistsReply.class);
+ DataExistsReply reply = testKit.expectMsgClass(Duration.ofSeconds(5), DataExistsReply.class);
assertTrue(reply.exists());
}
private void testOnReceiveDataExistsNegative(final ActorRef transaction) {
transaction.tell(new DataExists(TestModel.TEST_PATH, DataStoreVersions.CURRENT_VERSION), testKit.getRef());
- DataExistsReply reply = testKit.expectMsgClass(testKit.duration("5 seconds"), DataExistsReply.class);
+ DataExistsReply reply = testKit.expectMsgClass(Duration.ofSeconds(5), DataExistsReply.class);
assertFalse(reply.exists());
}
final ActorRef transaction = newTransactionActor(RW, mockWriteTx, "testOnReceiveBatchedModifications");
YangInstanceIdentifier writePath = TestModel.TEST_PATH;
- NormalizedNode<?, ?> writeData = ImmutableContainerNodeBuilder.create()
+ NormalizedNode writeData = ImmutableContainerNodeBuilder.create()
.withNodeIdentifier(new YangInstanceIdentifier.NodeIdentifier(TestModel.TEST_QNAME))
.withChild(ImmutableNodes.leafNode(TestModel.DESC_QNAME, "foo")).build();
YangInstanceIdentifier mergePath = TestModel.OUTER_LIST_PATH;
- NormalizedNode<?, ?> mergeData = ImmutableContainerNodeBuilder.create()
+ NormalizedNode mergeData = ImmutableContainerNodeBuilder.create()
.withNodeIdentifier(new YangInstanceIdentifier.NodeIdentifier(TestModel.OUTER_LIST_QNAME))
.build();
transaction.tell(batched, testKit.getRef());
- BatchedModificationsReply reply = testKit.expectMsgClass(testKit.duration("5 seconds"),
+ BatchedModificationsReply reply = testKit.expectMsgClass(Duration.ofSeconds(5),
BatchedModificationsReply.class);
assertEquals("getNumBatched", 3, reply.getNumBatched());
watcher.watch(transaction);
YangInstanceIdentifier writePath = TestModel.TEST_PATH;
- NormalizedNode<?, ?> writeData = ImmutableContainerNodeBuilder.create()
+ NormalizedNode writeData = ImmutableContainerNodeBuilder.create()
.withNodeIdentifier(new YangInstanceIdentifier.NodeIdentifier(TestModel.TEST_QNAME))
.withChild(ImmutableNodes.leafNode(TestModel.DESC_QNAME, "foo")).build();
batched.addModification(new WriteModification(writePath, writeData));
transaction.tell(batched, testKit.getRef());
- BatchedModificationsReply reply = testKit.expectMsgClass(testKit.duration("5 seconds"),
+ BatchedModificationsReply reply = testKit.expectMsgClass(Duration.ofSeconds(5),
BatchedModificationsReply.class);
assertEquals("getNumBatched", 1, reply.getNumBatched());
batched.setTotalMessagesSent(2);
transaction.tell(batched, testKit.getRef());
- testKit.expectMsgClass(testKit.duration("5 seconds"), ReadyTransactionReply.class);
- watcher.expectMsgClass(watcher.duration("5 seconds"), Terminated.class);
+ testKit.expectMsgClass(Duration.ofSeconds(5), ReadyTransactionReply.class);
+ watcher.expectMsgClass(Duration.ofSeconds(5), Terminated.class);
}
@Test
watcher.watch(transaction);
YangInstanceIdentifier writePath = TestModel.TEST_PATH;
- NormalizedNode<?, ?> writeData = ImmutableContainerNodeBuilder.create()
+ NormalizedNode writeData = ImmutableContainerNodeBuilder.create()
.withNodeIdentifier(new YangInstanceIdentifier.NodeIdentifier(TestModel.TEST_QNAME))
.withChild(ImmutableNodes.leafNode(TestModel.DESC_QNAME, "foo")).build();
batched.setTotalMessagesSent(1);
transaction.tell(batched, testKit.getRef());
- testKit.expectMsgClass(testKit.duration("5 seconds"), CommitTransactionReply.class);
- watcher.expectMsgClass(testKit.duration("5 seconds"), Terminated.class);
+ testKit.expectMsgClass(Duration.ofSeconds(5), CommitTransactionReply.class);
+ watcher.expectMsgClass(Duration.ofSeconds(5), Terminated.class);
}
@Test(expected = TestException.class)
batched.addModification(new WriteModification(path, node));
transaction.tell(batched, testKit.getRef());
- testKit.expectMsgClass(testKit.duration("5 seconds"), akka.actor.Status.Failure.class);
+ testKit.expectMsgClass(Duration.ofSeconds(5), akka.actor.Status.Failure.class);
batched = new BatchedModifications(tx1, DataStoreVersions.CURRENT_VERSION);
batched.setReady();
batched.setTotalMessagesSent(2);
transaction.tell(batched, testKit.getRef());
- Failure failure = testKit.expectMsgClass(testKit.duration("5 seconds"), akka.actor.Status.Failure.class);
- watcher.expectMsgClass(testKit.duration("5 seconds"), Terminated.class);
+ Failure failure = testKit.expectMsgClass(Duration.ofSeconds(5), akka.actor.Status.Failure.class);
+ watcher.expectMsgClass(Duration.ofSeconds(5), Terminated.class);
if (failure != null) {
Throwables.propagateIfPossible(failure.cause(), Exception.class);
transaction.tell(batched, testKit.getRef());
- Failure failure = testKit.expectMsgClass(testKit.duration("5 seconds"), akka.actor.Status.Failure.class);
- watcher.expectMsgClass(watcher.duration("5 seconds"), Terminated.class);
+ Failure failure = testKit.expectMsgClass(Duration.ofSeconds(5), akka.actor.Status.Failure.class);
+ watcher.expectMsgClass(Duration.ofSeconds(5), Terminated.class);
if (failure != null) {
Throwables.throwIfInstanceOf(failure.cause(), Exception.class);
transaction.tell(new CloseTransaction().toSerializable(), testKit.getRef());
- testKit.expectMsgClass(testKit.duration("3 seconds"), CloseTransactionReply.class);
- testKit.expectTerminated(testKit.duration("3 seconds"), transaction);
+ testKit.expectMsgClass(Duration.ofSeconds(3), CloseTransactionReply.class);
+ testKit.expectTerminated(Duration.ofSeconds(3), transaction);
}
@Test
transaction.tell(new CloseTransaction().toSerializable(), testKit.getRef());
- testKit.expectMsgClass(testKit.duration("3 seconds"), CloseTransactionReply.class);
- testKit.expectTerminated(testKit.duration("3 seconds"), transaction);
+ testKit.expectMsgClass(Duration.ofSeconds(3), CloseTransactionReply.class);
+ testKit.expectTerminated(Duration.ofSeconds(3), transaction);
}
@Test
transaction.tell(new CloseTransaction().toSerializable(), testKit.getRef());
- testKit.expectMsgClass(testKit.duration("3 seconds"), Terminated.class);
+ testKit.expectMsgClass(Duration.ofSeconds(3), Terminated.class);
}
@Test
testKit.watch(transaction);
- testKit.expectMsgClass(testKit.duration("3 seconds"), Terminated.class);
+ testKit.expectMsgClass(Duration.ofSeconds(3), Terminated.class);
}
public static class TestException extends RuntimeException {