2 * Copyright (c) 2015 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.broker;
10 import static org.mockito.Matchers.anyMap;
11 import static org.mockito.Mockito.doNothing;
12 import static org.mockito.Mockito.doReturn;
13 import static org.mockito.Mockito.times;
14 import static org.mockito.Mockito.verify;
16 import java.util.Collections;
17 import org.junit.Before;
18 import org.junit.Test;
19 import org.mockito.Mock;
20 import org.mockito.MockitoAnnotations;
21 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
22 import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier;
23 import org.opendaylight.mdsal.dom.api.DOMDataTreeShard;
24 import org.opendaylight.mdsal.dom.api.DOMDataTreeShardingConflictException;
25 import org.opendaylight.mdsal.dom.api.DOMDataTreeShardingService;
26 import org.opendaylight.mdsal.dom.broker.util.TestModel;
27 import org.opendaylight.yangtools.concepts.ListenerRegistration;
28 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
30 public class ShardedDOMDataTreeShardTest {
33 private static final DOMDataTreeIdentifier ROOT_ID = new DOMDataTreeIdentifier(LogicalDatastoreType.OPERATIONAL,
34 YangInstanceIdentifier.EMPTY);
35 private static final DOMDataTreeIdentifier TEST_ID = new DOMDataTreeIdentifier(LogicalDatastoreType.OPERATIONAL,
38 @Mock(name = "rootShard")
39 private DOMDataTreeShard rootShard;
41 @Mock(name = "rootShard")
42 private DOMDataTreeShard childShard;
45 private ShardedDOMDataTreeProducer rootProducer;
48 private ShardedDOMDataTreeProducer testProducer;
50 private DOMDataTreeShardingService shardingService;
51 private ListenerRegistration<DOMDataTreeShard> shardReg;
54 public void setUp() throws DOMDataTreeShardingConflictException {
55 MockitoAnnotations.initMocks(this);
56 doReturn(Collections.singleton(ROOT_ID)).when(rootProducer).getSubtrees();
57 doReturn(Collections.singleton(TEST_ID)).when(testProducer).getSubtrees();
58 doNothing().when(rootProducer).subshardAdded(anyMap());
59 doNothing().when(testProducer).subshardAdded(anyMap());
60 final ShardedDOMDataTree impl = new ShardedDOMDataTree();
61 shardingService = impl;
62 shardReg = impl.registerDataTreeShard(ROOT_ID, rootShard, rootProducer);
63 doReturn("rootShard").when(rootShard).toString();
64 doReturn("childShard").when(childShard).toString();
68 public void attachChildShard() throws DOMDataTreeShardingConflictException {
69 doNothing().when(rootShard).onChildAttached(TEST_ID, childShard);
70 shardingService.registerDataTreeShard(TEST_ID, childShard, testProducer);
71 verify(rootShard, times(1)).onChildAttached(TEST_ID, childShard);
75 public void attachAndRemoveShard() throws DOMDataTreeShardingConflictException {
76 doNothing().when(rootShard).onChildAttached(TEST_ID, childShard);
77 final ListenerRegistration<DOMDataTreeShard> reg =
78 shardingService.registerDataTreeShard(TEST_ID, childShard, testProducer);
79 verify(rootShard, times(1)).onChildAttached(TEST_ID, childShard);
81 doNothing().when(rootShard).onChildDetached(TEST_ID, childShard);
82 doNothing().when(childShard).onChildDetached(TEST_ID, childShard);
85 verify(rootShard, times(1)).onChildDetached(TEST_ID, childShard);
89 public void removeShard() {
90 doNothing().when(rootShard).onChildDetached(ROOT_ID, rootShard);