2 * Copyright IBM Corporation, 2013. All rights reserved.
\r
4 * This program and the accompanying materials are made available under the
\r
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
\r
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
\r
8 package org.opendaylight.controller.samples.loadbalancer.internal;
\r
11 import org.junit.Assert;
\r
12 import org.junit.Test;
\r
13 import org.opendaylight.controller.samples.loadbalancer.ConfigManager;
\r
14 import org.opendaylight.controller.samples.loadbalancer.entities.Client;
\r
15 import org.opendaylight.controller.samples.loadbalancer.entities.Pool;
\r
16 import org.opendaylight.controller.samples.loadbalancer.entities.PoolMember;
\r
17 import org.opendaylight.controller.samples.loadbalancer.entities.VIP;
\r
18 import org.opendaylight.controller.samples.loadbalancer.policies.RoundRobinLBPolicy;
\r
20 import junit.framework.TestCase;
\r
24 * Class to unit test the load balancing policies.
\r
27 public class LoadBalancerTest extends TestCase {
\r
29 public void testRoundRobinPolicy() {
\r
30 ConfigManager cm = null;
\r
31 cm = new ConfigManager();
\r
32 Assert.assertFalse(cm== null);
\r
34 Pool pool = cm.createPool("TestPool","roundrobin");
\r
35 VIP vip = cm.createVIP("TestVIP","10.0.0.9","TCP",(short)5550,"TestPool");
\r
36 PoolMember host1 = new PoolMember("host1","10.0.0.1","TestPool");
\r
37 PoolMember host2 = new PoolMember("host2","10.0.0.2","TestPool");
\r
38 PoolMember host3 = new PoolMember("host3","10.0.0.3","TestPool");
\r
39 PoolMember host4 = new PoolMember("host4","10.0.0.4","TestPool");
\r
40 PoolMember host5 = new PoolMember("host5","10.0.0.5","TestPool");
\r
41 PoolMember host6 = new PoolMember("host6","10.0.0.6","TestPool");
\r
42 PoolMember host7 = new PoolMember("host7","10.0.0.7","TestPool");
\r
44 pool.addMember(host1);
\r
45 pool.addMember(host2);
\r
46 pool.addMember(host3);
\r
47 pool.addMember(host4);
\r
48 pool.addMember(host5);
\r
49 pool.addMember(host6);
\r
50 pool.addMember(host7);
\r
53 Assert.assertTrue(cm.getAllPoolMembers("TestPool").size() == pool.getAllMembers().size());
\r
55 RoundRobinLBPolicy rrp = new RoundRobinLBPolicy(cm);
\r
57 Client c1 = new Client("10.0.0.1","TCP",(short)5000);
\r
58 Assert.assertTrue(rrp.getPoolMemberForClient(c1, vip).equals(host1.getIp()));
\r
60 c1 = new Client("10.0.0.1","TCP",(short)5001);
\r
61 Assert.assertTrue(rrp.getPoolMemberForClient(c1, vip).equals(host2.getIp()));
\r
63 c1 = new Client("10.0.0.1","TCP",(short)5002);
\r
64 Assert.assertTrue(rrp.getPoolMemberForClient(c1, vip).equals(host3.getIp()));
\r
66 c1 = new Client("10.0.0.1","TCP",(short)5003);
\r
67 Assert.assertTrue(rrp.getPoolMemberForClient(c1, vip).equals(host4.getIp()));
\r