+++ /dev/null
-/*
- * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
- *
- * This program and the accompanying materials are made available under the
- * terms of the Eclipse Public License v1.0 which accompanies this distribution,
- * and is available at http://www.eclipse.org/legal/epl-v10.html
- */
-package org.opendaylight.mdsal.dom.broker.test;
-
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertFalse;
-import static org.junit.Assert.assertNotNull;
-import static org.mockito.Matchers.any;
-import static org.mockito.Matchers.eq;
-import static org.mockito.Mockito.doNothing;
-import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.times;
-import static org.mockito.Mockito.verify;
-import static org.opendaylight.mdsal.dom.broker.test.MockingUtilities.captorFor;
-
-import java.util.Collection;
-import java.util.Collections;
-import java.util.Map;
-import org.junit.Before;
-import org.junit.Test;
-import org.mockito.ArgumentCaptor;
-import org.mockito.Mock;
-import org.mockito.Mockito;
-import org.mockito.MockitoAnnotations;
-import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeChangeListener;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeListener;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeLoopException;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeProducer;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeService;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeShard;
-import org.opendaylight.mdsal.dom.api.DOMDataTreeShardingConflictException;
-import org.opendaylight.mdsal.dom.broker.ShardedDOMDataTree;
-import org.opendaylight.mdsal.dom.broker.test.util.TestModel;
-import org.opendaylight.mdsal.dom.spi.store.DOMStore;
-import org.opendaylight.mdsal.dom.spi.store.DOMStoreTransactionChain;
-import org.opendaylight.mdsal.dom.spi.store.DOMStoreTreeChangePublisher;
-import org.opendaylight.mdsal.dom.spi.store.DOMStoreWriteTransaction;
-import org.opendaylight.yangtools.concepts.ListenerRegistration;
-import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
-import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
-import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
-import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidate;
-import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidates;
-import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
-
-public class ShardedDOMDataTreeListenerWithProducerTest {
-
-
- private static final DOMDataTreeIdentifier ROOT_ID = new DOMDataTreeIdentifier(LogicalDatastoreType.OPERATIONAL,
- YangInstanceIdentifier.EMPTY);
- private static final DOMDataTreeIdentifier TEST_ID = new DOMDataTreeIdentifier(LogicalDatastoreType.OPERATIONAL,
- TestModel.TEST_PATH);
-
- private static final DOMDataTreeIdentifier TEST2_ID = new DOMDataTreeIdentifier(LogicalDatastoreType.OPERATIONAL,
- TestModel.TEST2_PATH);
-
-
- private static final Collection<DOMDataTreeIdentifier> SUBTREES_ROOT = Collections.singleton(ROOT_ID);
- private static final Collection<DOMDataTreeIdentifier> SUBTREES_TEST = Collections.singleton(TEST_ID);
- private static final Collection<DOMDataTreeIdentifier> SUBTREES_TEST2 = Collections.singleton(TEST2_ID);
- private static final ContainerNode TEST_CONTAINER = ImmutableNodes.containerNode(TestModel.TEST_QNAME);
-
- private interface ListenableShard extends DOMDataTreeShard, DOMStoreTreeChangePublisher, DOMStore {
-
-
- }
-
- @Mock(name = "rootShard")
- private ListenableShard rootShard;
-
- @Mock(name = "childShard")
- private ListenableShard childShard;
-
- @Mock
- private ListenerRegistration<?> storeListenerReg;
-
- @Mock(name = "storeWriteTx")
- private DOMStoreWriteTransaction writeTxMock;
-
- @Mock(name = "storeTxChain")
- private DOMStoreTransactionChain txChainMock;
-
- private DOMDataTreeService treeService;
-
- private ListenerRegistration<ListenableShard> shardReg;
-
- @Before
- public void setUp() throws DOMDataTreeShardingConflictException {
- MockitoAnnotations.initMocks(this);
- final ShardedDOMDataTree impl = new ShardedDOMDataTree();
- treeService = impl;
- shardReg = impl.registerDataTreeShard(ROOT_ID, rootShard);
- doReturn("rootShard").when(rootShard).toString();
- doReturn("childShard").when(childShard).toString();
-
- doReturn(txChainMock).when(rootShard).createTransactionChain();
- doReturn(writeTxMock).when(txChainMock).newWriteOnlyTransaction();
- doReturn(TestCommitCohort.ALLWAYS_SUCCESS).when(writeTxMock).ready();
-
- doReturn(storeListenerReg).when(rootShard).registerTreeChangeListener(any(YangInstanceIdentifier.class),
- any(DOMDataTreeChangeListener.class));
- doNothing().when(storeListenerReg).close();
- }
-
- @Test
- public void registerListenerWithOneProducer() throws DOMDataTreeLoopException {
- DOMDataTreeListener listener = Mockito.mock(DOMDataTreeListener.class);
- DOMDataTreeProducer producer = treeService.createProducer(SUBTREES_TEST2);
- treeService.registerListener(listener, SUBTREES_TEST, true, Collections.singleton(producer));
- }
-
- @Test(expected = IllegalStateException.class)
- public void registerListenerWithAlreadyBoundProducer() throws DOMDataTreeLoopException {
- DOMDataTreeListener listener1 = MockingUtilities.mock(DOMDataTreeListener.class, "listener1");
- DOMDataTreeProducer producer = treeService.createProducer(SUBTREES_TEST2);
- treeService.registerListener(listener1, SUBTREES_TEST, true, Collections.singleton(producer));
-
- DOMDataTreeListener listener2 = MockingUtilities.mock(DOMDataTreeListener.class, "listener2");
- treeService.registerListener(listener2, SUBTREES_TEST, true, Collections.singleton(producer));
- }
-
- @Test(expected = DOMDataTreeLoopException.class)
- public void loopSameSubtree() throws DOMDataTreeLoopException {
- DOMDataTreeListener listener = Mockito.mock(DOMDataTreeListener.class);
- DOMDataTreeProducer producer = treeService.createProducer(SUBTREES_TEST);
- treeService.registerListener(listener, SUBTREES_TEST, true, Collections.singleton(producer));
- }
-
- @Test(expected = DOMDataTreeLoopException.class)
- public void loopListenParentWritesChild() throws DOMDataTreeLoopException {
- DOMDataTreeListener listener = Mockito.mock(DOMDataTreeListener.class);
- DOMDataTreeProducer producer = treeService.createProducer(SUBTREES_TEST);
- treeService.registerListener(listener, SUBTREES_ROOT, true, Collections.singleton(producer));
- }
-
- @Test(expected = DOMDataTreeLoopException.class)
- public void loopListenChildWritesParent() throws DOMDataTreeLoopException {
- DOMDataTreeListener listener = Mockito.mock(DOMDataTreeListener.class);
- DOMDataTreeProducer producer = treeService.createProducer(SUBTREES_ROOT);
- treeService.registerListener(listener, SUBTREES_ROOT, true, Collections.singleton(producer));
- }
-
- @Test
- public void receiveChangeEvent() throws DOMDataTreeLoopException {
- DOMDataTreeListener listener = Mockito.mock(DOMDataTreeListener.class);
- ArgumentCaptor<DOMDataTreeChangeListener> storeListener =
- ArgumentCaptor.forClass(DOMDataTreeChangeListener.class);
- treeService.registerListener(listener, SUBTREES_TEST, true, Collections.<DOMDataTreeProducer>emptyList());
- verify(rootShard, times(1)).registerTreeChangeListener(eq(TEST_ID.getRootIdentifier()),
- storeListener.capture());
-
- DataTreeCandidate sentStoreCandidate =
- DataTreeCandidates.fromNormalizedNode(TEST_ID.getRootIdentifier(), TEST_CONTAINER);
- Collection<DataTreeCandidate> changes = Collections.singleton(sentStoreCandidate);
-
- doNothing().when(listener).onDataTreeChanged(Mockito.<Collection<DataTreeCandidate>>any(), Mockito.anyMap());
- storeListener.getValue().onDataTreeChanged(changes);
-
- ArgumentCaptor<Collection<DataTreeCandidate>> candidateCapture = captorFor(Collection.class);
- ArgumentCaptor<Map<DOMDataTreeIdentifier, NormalizedNode<?, ?>>> mapCapture = captorFor(Map.class);
- verify(listener, times(1)).onDataTreeChanged(candidateCapture.capture(), mapCapture.capture());
-
- Collection<DataTreeCandidate> receivedCandidate = candidateCapture.getValue();
- Map<DOMDataTreeIdentifier, NormalizedNode<?, ?>> receivedMap = mapCapture.getValue();
-
- assertNotNull("receivedCandidate", receivedCandidate);
- assertNotNull("receivedMap", receivedMap);
- assertFalse("candidate collection must not be empty", receivedCandidate.isEmpty());
- assertEquals(1, receivedCandidate.size());
- DataTreeCandidate firstItem = receivedCandidate.iterator().next();
- assertEquals(TEST_ID.getRootIdentifier(), firstItem.getRootPath());
- assertEquals(TEST_CONTAINER, receivedMap.get(TEST_ID));
- }
-
-}