2 * Copyright (c) 2018 ZTE, 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.broker;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertNotNull;
12 import static org.mockito.ArgumentMatchers.any;
13 import static org.mockito.Mockito.doNothing;
14 import static org.mockito.Mockito.mock;
15 import static org.mockito.Mockito.timeout;
16 import static org.mockito.Mockito.times;
17 import static org.mockito.Mockito.verify;
19 import com.google.common.collect.Iterables;
20 import com.google.common.util.concurrent.MoreExecutors;
21 import java.util.Collection;
22 import java.util.concurrent.ExecutionException;
23 import org.junit.Before;
24 import org.junit.Test;
25 import org.mockito.ArgumentCaptor;
26 import org.opendaylight.mdsal.dom.api.DOMDataTreeChangeListener;
27 import org.opendaylight.mdsal.dom.broker.util.TestModel;
28 import org.opendaylight.mdsal.dom.spi.store.DOMStoreReadWriteTransaction;
29 import org.opendaylight.mdsal.dom.spi.store.DOMStoreThreePhaseCommitCohort;
30 import org.opendaylight.mdsal.dom.store.inmemory.InMemoryDOMDataStore;
31 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
32 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
33 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
34 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidate;
35 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
36 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableContainerNodeBuilder;
38 public class DOMDataTreeChangeListenerTest {
40 private InMemoryDOMDataStore domStore;
44 domStore = new InMemoryDOMDataStore("Mdsal217", MoreExecutors.newDirectExecutorService());
45 domStore.onGlobalContextUpdated(TestModel.createTestContext());
49 public void receiveOnDataInitialEventForEmptyRoot() throws InterruptedException, ExecutionException {
50 final DOMDataTreeChangeListener listener = mock(DOMDataTreeChangeListener.class);
51 doNothing().when(listener).onInitialData();
53 domStore.registerTreeChangeListener(YangInstanceIdentifier.EMPTY, listener);
54 verify(listener, timeout(1000)).onInitialData();
58 public void receiveOnDataInitialEventForNonExistingData() throws InterruptedException, ExecutionException {
59 final DOMDataTreeChangeListener listener = mock(DOMDataTreeChangeListener.class);
60 final ArgumentCaptor<Collection> candidateCapture = ArgumentCaptor.forClass(Collection.class);
61 doNothing().when(listener).onInitialData();
62 doNothing().when(listener).onDataTreeChanged(any());
64 domStore.registerTreeChangeListener(TestModel.TEST_PATH, listener);
65 verify(listener, times(1)).onInitialData();
67 final NormalizedNode<?, ?> testNode = ImmutableNodes.containerNode(TestModel.TEST_QNAME);
68 DOMStoreReadWriteTransaction writeTx = domStore.newReadWriteTransaction();
69 assertNotNull(writeTx);
70 writeTx.write(TestModel.TEST_PATH, testNode);
71 DOMStoreThreePhaseCommitCohort cohort = writeTx.ready();
72 cohort.preCommit().get();
73 cohort.commit().get();
75 verify(listener, timeout(1000)).onDataTreeChanged(candidateCapture.capture());
76 final DataTreeCandidate candidate = (DataTreeCandidate) Iterables.getOnlyElement(candidateCapture.getValue());
77 assertEquals(TestModel.TEST_PATH, candidate.getRootPath());
81 public void receiveOnDataTreeChangedEventForPreExistingEmptyData() throws InterruptedException, ExecutionException {
82 final DOMDataTreeChangeListener listener = mock(DOMDataTreeChangeListener.class);
83 final ArgumentCaptor<Collection> candidateCapture = ArgumentCaptor.forClass(Collection.class);
84 doNothing().when(listener).onDataTreeChanged(any());
86 final NormalizedNode<?, ?> testNode = ImmutableNodes.containerNode(TestModel.TEST_QNAME);
88 DOMStoreReadWriteTransaction writeTx = domStore.newReadWriteTransaction();
89 assertNotNull(writeTx);
90 writeTx.write(TestModel.TEST_PATH, testNode);
91 DOMStoreThreePhaseCommitCohort cohort = writeTx.ready();
92 cohort.preCommit().get();
93 cohort.commit().get();
95 domStore.registerTreeChangeListener(TestModel.TEST_PATH, listener);
97 verify(listener, timeout(1000)).onDataTreeChanged(candidateCapture.capture());
98 final DataTreeCandidate candidate = (DataTreeCandidate) Iterables.getOnlyElement(candidateCapture.getValue());
99 assertEquals(TestModel.TEST_PATH, candidate.getRootPath());
103 public void receiveOnDataTreeChangeEventForPreExistingData() throws InterruptedException, ExecutionException {
104 final DOMDataTreeChangeListener listener = mock(DOMDataTreeChangeListener.class);
105 final ArgumentCaptor<Collection> candidateCapture = ArgumentCaptor.forClass(Collection.class);
106 doNothing().when(listener).onDataTreeChanged(any());
108 final ContainerNode testNode = ImmutableContainerNodeBuilder.create()
109 .withNodeIdentifier(new YangInstanceIdentifier.NodeIdentifier(TestModel.TEST_QNAME))
110 .addChild(ImmutableNodes.mapNodeBuilder(TestModel.OUTER_LIST_QNAME)
111 .addChild(ImmutableNodes.mapEntry(TestModel.OUTER_LIST_QNAME,
112 TestModel.ID_QNAME, 1)).build()).build();
113 DOMStoreReadWriteTransaction writeTx = domStore.newReadWriteTransaction();
114 assertNotNull(writeTx);
115 writeTx.write(TestModel.TEST_PATH, testNode);
116 DOMStoreThreePhaseCommitCohort cohort = writeTx.ready();
117 cohort.preCommit().get();
118 cohort.commit().get();
120 domStore.registerTreeChangeListener(TestModel.TEST_PATH, listener);
122 verify(listener, timeout(1000)).onDataTreeChanged(candidateCapture.capture());
123 final DataTreeCandidate firstItem = (DataTreeCandidate) candidateCapture.getValue().iterator().next();
124 assertEquals(TestModel.TEST_PATH, firstItem.getRootPath());