dom/mdsal-dom-broker/src/test/java/org/opendaylight/mdsal/dom/broker/ShardedDOMDataTreeShardTest.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.mdsal.dom.broker;
   9
  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;
  15
  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;
  29
  30 public class ShardedDOMDataTreeShardTest {
  31
  32
  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,
  36             TestModel.TEST_PATH);
  37
  38     @Mock(name = "rootShard")
  39     private DOMDataTreeShard rootShard;
  40
  41     @Mock(name = "rootShard")
  42     private DOMDataTreeShard childShard;
  43
  44     @Mock
  45     private ShardedDOMDataTreeProducer rootProducer;
  46
  47     @Mock
  48     private ShardedDOMDataTreeProducer testProducer;
  49
  50     private DOMDataTreeShardingService shardingService;
  51     private ListenerRegistration<DOMDataTreeShard> shardReg;
  52
  53     @Before
  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();
  65     }
  66
  67     @Test
  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);
  72     }
  73
  74     @Test
  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);
  80
  81         doNothing().when(rootShard).onChildDetached(TEST_ID, childShard);
  82         doNothing().when(childShard).onChildDetached(TEST_ID, childShard);
  83
  84         reg.close();
  85         verify(rootShard, times(1)).onChildDetached(TEST_ID, childShard);
  86     }
  87
  88     @Test
  89     public void removeShard() {
  90         doNothing().when(rootShard).onChildDetached(ROOT_ID, rootShard);
  91         shardReg.close();
  92     }
  93
  94 }