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
22 * Class to unit test the load balancing policies.
\r
25 public class LoadBalancerTest {
\r
27 public void testRoundRobinPolicy() {
\r
28 ConfigManager cm = null;
\r
29 cm = new ConfigManager();
\r
31 Pool pool = cm.createPool("TestPool","roundrobin");
\r
32 VIP vip = cm.createVIP("TestVIP","10.0.0.9","TCP",(short)5550,"TestPool");
\r
33 PoolMember host1 = new PoolMember("host1","10.0.0.1","TestPool");
\r
34 PoolMember host2 = new PoolMember("host2","10.0.0.2","TestPool");
\r
35 PoolMember host3 = new PoolMember("host3","10.0.0.3","TestPool");
\r
36 PoolMember host4 = new PoolMember("host4","10.0.0.4","TestPool");
\r
37 PoolMember host5 = new PoolMember("host5","10.0.0.5","TestPool");
\r
38 PoolMember host6 = new PoolMember("host6","10.0.0.6","TestPool");
\r
39 PoolMember host7 = new PoolMember("host7","10.0.0.7","TestPool");
\r
41 pool.addMember(host1);
\r
42 pool.addMember(host2);
\r
43 pool.addMember(host3);
\r
44 pool.addMember(host4);
\r
45 pool.addMember(host5);
\r
46 pool.addMember(host6);
\r
47 pool.addMember(host7);
\r
50 Assert.assertTrue(cm.getAllPoolMembers("TestPool").size() == pool.getAllMembers().size());
\r
52 RoundRobinLBPolicy rrp = new RoundRobinLBPolicy(cm);
\r
54 Client c1 = new Client("10.0.0.1","TCP",(short)5000);
\r
55 Assert.assertTrue(rrp.getPoolMemberForClient(c1, vip).equals(host1.getIp()));
\r
57 c1 = new Client("10.0.0.1","TCP",(short)5001);
\r
58 Assert.assertTrue(rrp.getPoolMemberForClient(c1, vip).equals(host2.getIp()));
\r
60 c1 = new Client("10.0.0.1","TCP",(short)5002);
\r
61 Assert.assertTrue(rrp.getPoolMemberForClient(c1, vip).equals(host3.getIp()));
\r
63 c1 = new Client("10.0.0.1","TCP",(short)5003);
\r
64 Assert.assertTrue(rrp.getPoolMemberForClient(c1, vip).equals(host4.getIp()));
\r