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.spi.store.DOMStoreReadWriteTransaction;
28 import org.opendaylight.mdsal.dom.spi.store.DOMStoreThreePhaseCommitCohort;
29 import org.opendaylight.mdsal.dom.store.inmemory.InMemoryDOMDataStore;
30 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
31 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
32 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
33 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidate;
34 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
35 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableContainerNodeBuilder;
37 public class DOMDataTreeChangeListenerTest extends AbstractDatastoreTest {
39 private InMemoryDOMDataStore domStore;
43 domStore = new InMemoryDOMDataStore("Mdsal217", MoreExecutors.newDirectExecutorService());
44 domStore.onModelContextUpdated(SCHEMA_CONTEXT);
48 public void receiveOnDataInitialEventForEmptyRoot() throws InterruptedException, ExecutionException {
49 final DOMDataTreeChangeListener listener = mock(DOMDataTreeChangeListener.class);
50 doNothing().when(listener).onInitialData();
52 domStore.registerTreeChangeListener(YangInstanceIdentifier.empty(), listener);
53 verify(listener, timeout(1000)).onInitialData();
57 public void receiveOnDataInitialEventForNonExistingData() throws InterruptedException, ExecutionException {
58 final DOMDataTreeChangeListener listener = mock(DOMDataTreeChangeListener.class);
59 final ArgumentCaptor<Collection> candidateCapture = ArgumentCaptor.forClass(Collection.class);
60 doNothing().when(listener).onInitialData();
61 doNothing().when(listener).onDataTreeChanged(any());
63 domStore.registerTreeChangeListener(TestModel.TEST_PATH, listener);
64 verify(listener, times(1)).onInitialData();
66 final NormalizedNode testNode = ImmutableNodes.containerNode(TestModel.TEST_QNAME);
67 DOMStoreReadWriteTransaction writeTx = domStore.newReadWriteTransaction();
68 assertNotNull(writeTx);
69 writeTx.write(TestModel.TEST_PATH, testNode);
70 DOMStoreThreePhaseCommitCohort cohort = writeTx.ready();
71 cohort.preCommit().get();
72 cohort.commit().get();
74 verify(listener, timeout(1000)).onDataTreeChanged(candidateCapture.capture());
75 final DataTreeCandidate candidate = (DataTreeCandidate) Iterables.getOnlyElement(candidateCapture.getValue());
76 assertEquals(TestModel.TEST_PATH, candidate.getRootPath());
80 public void receiveOnDataTreeChangedEventForPreExistingEmptyData() throws InterruptedException, ExecutionException {
81 final DOMDataTreeChangeListener listener = mock(DOMDataTreeChangeListener.class);
82 final ArgumentCaptor<Collection> candidateCapture = ArgumentCaptor.forClass(Collection.class);
83 doNothing().when(listener).onDataTreeChanged(any());
85 final NormalizedNode testNode = ImmutableNodes.containerNode(TestModel.TEST_QNAME);
87 DOMStoreReadWriteTransaction writeTx = domStore.newReadWriteTransaction();
88 assertNotNull(writeTx);
89 writeTx.write(TestModel.TEST_PATH, testNode);
90 DOMStoreThreePhaseCommitCohort cohort = writeTx.ready();
91 cohort.preCommit().get();
92 cohort.commit().get();
94 domStore.registerTreeChangeListener(TestModel.TEST_PATH, listener);
96 verify(listener, timeout(1000)).onDataTreeChanged(candidateCapture.capture());
97 final DataTreeCandidate candidate = (DataTreeCandidate) Iterables.getOnlyElement(candidateCapture.getValue());
98 assertEquals(TestModel.TEST_PATH, candidate.getRootPath());
102 public void receiveOnDataTreeChangeEventForPreExistingData() throws InterruptedException, ExecutionException {
103 final DOMDataTreeChangeListener listener = mock(DOMDataTreeChangeListener.class);
104 final ArgumentCaptor<Collection> candidateCapture = ArgumentCaptor.forClass(Collection.class);
105 doNothing().when(listener).onDataTreeChanged(any());
107 final ContainerNode testNode = ImmutableContainerNodeBuilder.create()
108 .withNodeIdentifier(new YangInstanceIdentifier.NodeIdentifier(TestModel.TEST_QNAME))
109 .addChild(ImmutableNodes.mapNodeBuilder(TestModel.OUTER_LIST_QNAME)
110 .addChild(ImmutableNodes.mapEntry(TestModel.OUTER_LIST_QNAME,
111 TestModel.ID_QNAME, 1)).build()).build();
112 DOMStoreReadWriteTransaction writeTx = domStore.newReadWriteTransaction();
113 assertNotNull(writeTx);
114 writeTx.write(TestModel.TEST_PATH, testNode);
115 DOMStoreThreePhaseCommitCohort cohort = writeTx.ready();
116 cohort.preCommit().get();
117 cohort.commit().get();
119 domStore.registerTreeChangeListener(TestModel.TEST_PATH, listener);
121 verify(listener, timeout(1000)).onDataTreeChanged(candidateCapture.capture());
122 final DataTreeCandidate firstItem = (DataTreeCandidate) candidateCapture.getValue().iterator().next();
123 assertEquals(TestModel.TEST_PATH, firstItem.getRootPath());