opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/datastore/shardstrategy/ModuleShardStrategyTest.java

   1 package org.opendaylight.controller.cluster.datastore.shardstrategy;
   2
   3 import org.junit.BeforeClass;
   4 import org.junit.Rule;
   5 import org.junit.Test;
   6 import org.junit.rules.ExpectedException;
   7 import org.opendaylight.controller.cluster.datastore.Configuration;
   8 import org.opendaylight.controller.cluster.datastore.ConfigurationImpl;
   9 import org.opendaylight.controller.md.cluster.datastore.model.CarsModel;
  10 import org.opendaylight.yangtools.yang.common.QName;
  11 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  12
  13 import static junit.framework.Assert.assertEquals;
  14
  15 public class ModuleShardStrategyTest {
  16     @Rule
  17     public ExpectedException expectedEx = ExpectedException.none();
  18
  19     private static Configuration configuration;
  20
  21     @BeforeClass
  22     public static void setUpClass(){
  23         configuration = new ConfigurationImpl("module-shards.conf", "modules.conf");
  24     }
  25
  26
  27     @Test
  28     public void testFindShard() throws Exception {
  29         ModuleShardStrategy moduleShardStrategy =
  30             new ModuleShardStrategy("cars", configuration);
  31
  32         String shard = moduleShardStrategy.findShard(CarsModel.BASE_PATH);
  33
  34         assertEquals("cars-1", shard);
  35     }
  36
  37     @Test
  38     public void testFindShardWhenModuleConfigurationPresentInModulesButMissingInModuleShards() {
  39
  40         final QName BASE_QNAME = QName.create("urn:opendaylight:params:xml:ns:yang:controller:md:sal:dom:store:test:missing", "2014-03-13",
  41             "missing");
  42
  43         final YangInstanceIdentifier BASE_PATH = YangInstanceIdentifier.of(BASE_QNAME);
  44
  45         ModuleShardStrategy moduleShardStrategy =
  46             new ModuleShardStrategy("missing", configuration);
  47
  48         String shard = moduleShardStrategy.findShard(BASE_PATH);
  49
  50         assertEquals(DefaultShardStrategy.DEFAULT_SHARD, shard);
  51
  52     }
  53 }