package org.ellab.elchat;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;

import junit.framework.TestCase;

/**
 * Test case for <code>SessionMap</code>.
 */
public class SessionMapTest extends TestCase {
    private SessionMap.SessionItem item1 = new SessionMap.SessionItem();
    private SessionMap.SessionItem item2 = new SessionMap.SessionItem();
    private SessionMap.SessionItem item3 = new SessionMap.SessionItem();
    private SessionMap.SessionItem item4 = new SessionMap.SessionItem();
    private SessionMap.SessionItem item5 = new SessionMap.SessionItem();
    private List list;

    public void setUp() {
        item1.setIp("10.101.1.1");
        item2.setIp("10.101.1.2");
        item3.setIp("10.101.1.3");
        item3.setUser("def");
        item4.setIp("10.101.1.4");
        item4.setUser("abc");
        item5.setIp("10.101.1.5");

        list = new ArrayList();
        list.add(item1);
        list.add(item2);
        list.add(item3);
        list.add(item4);
        list.add(item5);
    }

    public void testCompare() {
        assertTrue(item1.compareTo(item2) < 0);
        assertTrue(item3.compareTo(item2) < 0);
        assertTrue(item4.compareTo(item3) < 0);
    }

    public void testSort() {
        Collections.sort(list);
        assertEquals("10.101.1.4", ((SessionMap.SessionItem) list.get(0)).getIp());
        assertEquals("10.101.1.3", ((SessionMap.SessionItem) list.get(1)).getIp());
        assertEquals("10.101.1.1", ((SessionMap.SessionItem) list.get(2)).getIp());
        assertEquals("10.101.1.2", ((SessionMap.SessionItem) list.get(3)).getIp());
        assertEquals("10.101.1.5", ((SessionMap.SessionItem) list.get(4)).getIp());
    }

    public void testTreeSet() {
        Set set = new TreeSet();
        set.add(item1);
        set.add(item2);
        set.add(item3);
        set.add(item4);
        set.add(item5);
        int i = 0;
        for (Iterator it = set.iterator(); it.hasNext();) {
            SessionMap.SessionItem item = (SessionMap.SessionItem) it.next();
            switch (i) {
                case 0 :
                    assertEquals("10.101.1.4", item.getIp());
                    break;
                case 1 :
                    assertEquals("10.101.1.3", item.getIp());
                    break;
                case 2 :
                    assertEquals("10.101.1.1", item.getIp());
                    break;
                case 3 :
                    assertEquals("10.101.1.2", item.getIp());
                    break;
                case 4 :
                    assertEquals("10.101.1.5", item.getIp());
                    break;
                default :
                    break;
            }
            i++;
        }
    }
}