affinity/api/src/test/java/org/opendaylight/affinity/affinity/AffinityGroupTest.java

   1 package org.opendaylight.affinity.affinity;
   2
   3 import java.net.Inet6Address;
   4 import java.net.InetAddress;
   5 import java.util.ArrayList;
   6 import java.util.List;
   7
   8 import junit.framework.TestCase;
   9 import org.junit.Assert;
  10 import org.junit.Test;
  11
  12 import org.opendaylight.controller.sal.utils.Status;
  13 import org.opendaylight.controller.sal.utils.StatusCode;
  14
  15 //import org.opendaylight.controller.sal.core.MacAddress;
  16
  17 public class AffinityGroupTest extends TestCase {
  18     @Test
  19     public void testAffinityGroup() throws Exception {
  20         InetAddress ipAddress1 = InetAddress.getByName("10.0.0.10");
  21         InetAddress ipAddress2 = InetAddress.getByName("10.0.0.20");
  22         Integer int1 = new Integer(10);
  23
  24         AffinityGroup ag1 = new AffinityGroup("group1");
  25
  26         // Add a valid IP and confirm.
  27         Status ret1 = ag1.add("10.0.0.10");
  28         Assert.assertTrue(ret1.isSuccess());
  29
  30         Status ret2 = ag1.add("10.0.0.20");
  31         Assert.assertTrue(ret2.isSuccess());
  32
  33         // Add an invalid element.
  34         Status ret3 = ag1.add("10");
  35         System.out.println(ret3);
  36         Assert.assertTrue(!ret3.isSuccess());
  37
  38         // Second affinity group.
  39         AffinityGroup ag2 = new AffinityGroup("group2");
  40         ag2.add("20.0.0.10");
  41         ag2.add("20.0.0.20");
  42
  43         // Add an affinity link from ag1 to ag2.
  44         AffinityLink al1 = new AffinityLink();
  45         al1.setFromGroup(ag1);
  46         al1.setToGroup(ag2);
  47         al1.setName("link1");
  48         al1.setAttribute("isolate");
  49
  50         // Add a self loop for ag2.
  51         AffinityLink al2 = new AffinityLink("link2", ag2, ag2);
  52         al2.setFromGroup(ag2);
  53         al2.setToGroup(ag2);
  54         al2.setName("link2");
  55         al2.setAttribute("hopcount");
  56
  57         System.out.println("Affinity group size is " + ag1.size());
  58         Assert.assertTrue(ag1.size() == 2);
  59         ag1.print();
  60     }
  61 }
  62