2 * Copyright (c) 2016 Cisco Systems, Inc. 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.mdsal.dom.store.inmemory;
10 import static org.junit.Assert.assertFalse;
11 import static org.junit.Assert.assertTrue;
12 import static org.mockito.Mockito.doNothing;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.only;
15 import static org.mockito.Mockito.verify;
16 import static org.opendaylight.mdsal.dom.store.inmemory.TestUtils.DOM_DATA_TREE_IDENTIFIER;
17 import static org.opendaylight.mdsal.dom.store.inmemory.TestUtils.DOM_DATA_TREE_SHARD_PRODUCER;
18 import static org.opendaylight.mdsal.dom.store.inmemory.TestUtils.DOM_DATA_TREE_SHARD_WRITE_TRANSACTION;
19 import static org.opendaylight.mdsal.dom.store.inmemory.TestUtils.DOM_DATA_TREE_WRITE_CURSOR;
20 import static org.opendaylight.mdsal.dom.store.inmemory.TestUtils.resetMocks;
22 import org.junit.After;
23 import org.junit.Test;
24 import org.mockito.Mockito;
26 public class ForeignShardModificationContextTest {
29 public void basicTest() throws Exception {
30 final ForeignShardModificationContext foreignShardModificationContext =
31 new ForeignShardModificationContext(DOM_DATA_TREE_IDENTIFIER, DOM_DATA_TREE_SHARD_PRODUCER);
32 doReturn("testTransaction").when(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).toString();
33 assertFalse(foreignShardModificationContext.isModified());
34 doReturn(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).when(DOM_DATA_TREE_SHARD_PRODUCER).createTransaction();
35 doReturn(DOM_DATA_TREE_WRITE_CURSOR)
36 .when(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).createCursor(DOM_DATA_TREE_IDENTIFIER);
37 foreignShardModificationContext.getCursor();
38 verify(DOM_DATA_TREE_SHARD_PRODUCER).createTransaction();
39 verify(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).createCursor(DOM_DATA_TREE_IDENTIFIER);
40 assertTrue(foreignShardModificationContext.isModified());
42 doNothing().when(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).ready();
43 doNothing().when(DOM_DATA_TREE_WRITE_CURSOR).close();
44 foreignShardModificationContext.ready();
45 verify(DOM_DATA_TREE_WRITE_CURSOR, only()).close();
46 verify(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).ready();
48 doReturn(null).when(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).validate();
49 foreignShardModificationContext.validate();
50 verify(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).validate();
52 doReturn(null).when(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).prepare();
53 foreignShardModificationContext.prepare();
54 verify(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).prepare();
56 doReturn(null).when(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).commit();
57 foreignShardModificationContext.submit();
58 verify(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).commit();
60 Mockito.reset(DOM_DATA_TREE_WRITE_CURSOR);
61 doNothing().when(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).close();
62 doNothing().when(DOM_DATA_TREE_WRITE_CURSOR).close();
63 foreignShardModificationContext.getCursor();
64 foreignShardModificationContext.closeForeignTransaction();
65 verify(DOM_DATA_TREE_WRITE_CURSOR).close();
66 verify(DOM_DATA_TREE_SHARD_WRITE_TRANSACTION).close();