opendaylight/md-sal/sal-remoterpc-connector/src/test/java/org/opendaylight/controller/remote/rpc/registry/gossip/BucketStoreTest.java

   1 /*
   2  * Copyright (c) 2014 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.controller.remote.rpc.registry.gossip;
   9
  10 import akka.actor.ActorSystem;
  11 import akka.actor.Address;
  12 import akka.actor.Props;
  13 import akka.testkit.TestActorRef;
  14 import com.typesafe.config.ConfigFactory;
  15 import java.util.HashMap;
  16 import java.util.Map;
  17 import org.junit.AfterClass;
  18 import org.junit.Assert;
  19 import org.junit.BeforeClass;
  20 import org.junit.Test;
  21 import org.opendaylight.controller.remote.rpc.RemoteRpcProviderConfig;
  22 import org.opendaylight.controller.remote.rpc.TerminationMonitor;
  23
  24 public class BucketStoreTest {
  25
  26     private static ActorSystem system;
  27
  28     @BeforeClass
  29     public static void setup() {
  30
  31         system = ActorSystem.create("opendaylight-rpc", ConfigFactory.load().getConfig("unit-test"));
  32         system.actorOf(Props.create(TerminationMonitor.class), "termination-monitor");
  33     }
  34
  35     @AfterClass
  36     public static void teardown() {
  37         system.shutdown();
  38     }
  39
  40     /**
  41      * Given remote buckets
  42      * Should merge with local copy of remote buckets
  43      */
  44     @Test
  45     public void testReceiveUpdateRemoteBuckets(){
  46
  47         BucketStore store = createStore();
  48
  49         Address localAddress = system.provider().getDefaultAddress();
  50         Bucket localBucket = new BucketImpl();
  51
  52         Address a1 = new Address("tcp", "system1");
  53         Address a2 = new Address("tcp", "system2");
  54         Address a3 = new Address("tcp", "system3");
  55
  56         Bucket b1 = new BucketImpl();
  57         Bucket b2 = new BucketImpl();
  58         Bucket b3 = new BucketImpl();
  59
  60         Map<Address, Bucket> remoteBuckets = new HashMap<>(3);
  61         remoteBuckets.put(a1, b1);
  62         remoteBuckets.put(a2, b2);
  63         remoteBuckets.put(a3, b3);
  64         remoteBuckets.put(localAddress, localBucket);
  65
  66         //Given remote buckets
  67         store.receiveUpdateRemoteBuckets(remoteBuckets);
  68
  69         //Should NOT contain local bucket
  70         //Should contain ONLY 3 entries i.e a1, a2, a3
  71         Map<Address, Bucket<?>> remoteBucketsInStore = store.getRemoteBuckets();
  72         Assert.assertFalse("remote buckets contains local bucket", remoteBucketsInStore.containsKey(localAddress));
  73         Assert.assertTrue(remoteBucketsInStore.size() == 3);
  74
  75         //Add a new remote bucket
  76         Address a4 = new Address("tcp", "system4");
  77         Bucket b4 = new BucketImpl();
  78         remoteBuckets.clear();
  79         remoteBuckets.put(a4, b4);
  80         store.receiveUpdateRemoteBuckets(remoteBuckets);
  81
  82         //Should contain a4
  83         //Should contain 4 entries now i.e a1, a2, a3, a4
  84         remoteBucketsInStore = store.getRemoteBuckets();
  85         Assert.assertTrue("Does not contain a4", remoteBucketsInStore.containsKey(a4));
  86         Assert.assertTrue(remoteBucketsInStore.size() == 4);
  87
  88         //Update a bucket
  89         Bucket b3_new = new BucketImpl();
  90         remoteBuckets.clear();
  91         remoteBuckets.put(a3, b3_new);
  92         remoteBuckets.put(a1, null);
  93         remoteBuckets.put(a2, null);
  94         store.receiveUpdateRemoteBuckets(remoteBuckets);
  95
  96         //Should only update a3
  97         remoteBucketsInStore = store.getRemoteBuckets();
  98         Bucket b3_inStore = remoteBucketsInStore.get(a3);
  99         Assert.assertEquals(b3_new.getVersion(), b3_inStore.getVersion());
 100
 101         //Should NOT update a1 and a2
 102         Bucket b1_inStore = remoteBucketsInStore.get(a1);
 103         Bucket b2_inStore = remoteBucketsInStore.get(a2);
 104         Assert.assertEquals(b1.getVersion(), b1_inStore.getVersion());
 105         Assert.assertEquals(b2.getVersion(), b2_inStore.getVersion());
 106         Assert.assertTrue(remoteBucketsInStore.size() == 4);
 107
 108         //Should update versions map
 109         //versions map contains versions for all remote buckets (4).
 110         Map<Address, Long> versionsInStore = store.getVersions();
 111         Assert.assertEquals(4, versionsInStore.size());
 112         Assert.assertEquals(b1.getVersion(), versionsInStore.get(a1));
 113         Assert.assertEquals(b2.getVersion(), versionsInStore.get(a2));
 114         Assert.assertEquals(b3_new.getVersion(), versionsInStore.get(a3));
 115         Assert.assertEquals(b4.getVersion(), versionsInStore.get(a4));
 116
 117         //Send older version of bucket
 118         remoteBuckets.clear();
 119         remoteBuckets.put(a3, b3);
 120         store.receiveUpdateRemoteBuckets(remoteBuckets);
 121
 122         //Should NOT update a3
 123         remoteBucketsInStore = store.getRemoteBuckets();
 124         b3_inStore = remoteBucketsInStore.get(a3);
 125         Assert.assertTrue(b3_inStore.getVersion().longValue() == b3_new.getVersion().longValue());
 126
 127     }
 128
 129     /**
 130      * Create BucketStore actor and returns the underlying instance of BucketStore class.
 131      *
 132      * @return instance of BucketStore class
 133      */
 134     private static BucketStore createStore(){
 135         final Props props = Props.create(BucketStore.class, new RemoteRpcProviderConfig(system.settings().config()));
 136         final TestActorRef<BucketStore> testRef = TestActorRef.create(system, props, "testStore");
 137         return testRef.underlyingActor();
 138     }
 139 }