TestMultiKeyUntyped.java

/*
 * *************************************************************************************
 *  Copyright (C) 2008 EsperTech, Inc. All rights reserved.                            *
 *  http://esper.codehaus.org                                                          *
 *  http://www.espertech.com                                                           *
 *  ---------------------------------------------------------------------------------- *
 *  The software in this package is published under the terms of the GPL license       *
 *  a copy of which has been included with this distribution in the license.txt file.  *
 * *************************************************************************************
 */

package com.espertech.esper.collection;

import junit.framework.TestCase;

import java.util.Set;
import java.util.HashSet;

import com.espertech.esper.client.EventBean;
import com.espertech.esper.support.event.SupportEventBeanFactory;

public class TestMultiKeyUntyped extends TestCase
{
    MultiKeyUntyped keys1 = new MultiKeyUntyped("a", "b");
    MultiKeyUntyped keys2 = new MultiKeyUntyped("a", "b");
    MultiKeyUntyped keys3 = new MultiKeyUntyped("a", null);
    MultiKeyUntyped keys4 = new MultiKeyUntyped(null, "b");
    MultiKeyUntyped keys5 = new MultiKeyUntyped(null, null);
    MultiKeyUntyped keys6 = new MultiKeyUntyped("a");
    MultiKeyUntyped keys7 = new MultiKeyUntyped("a", "b", "c");
    MultiKeyUntyped keys8 = new MultiKeyUntyped("a", "b", null);
    MultiKeyUntyped keys9 = new MultiKeyUntyped("a", "b", "c", "d");
    MultiKeyUntyped keys10 = new MultiKeyUntyped("a", "b", "c", "d");

    public void testHashCode()
    {
        assertTrue(keys1.hashCode() == ("a".hashCode()*31 ^ "b".hashCode()));
        assertTrue(keys3.hashCode() == "a".hashCode());
        assertTrue(keys4.hashCode() == "b".hashCode());
        assertTrue(keys5.hashCode() == 0);

        assertTrue(keys8.hashCode() == keys1.hashCode());
        assertTrue(keys1.hashCode() == keys2.hashCode());
        assertTrue(keys1.hashCode() != keys3.hashCode());
        assertTrue(keys1.hashCode() != keys4.hashCode());
        assertTrue(keys1.hashCode() != keys5.hashCode());

        assertTrue(keys7.hashCode() != keys8.hashCode());
        assertTrue(keys9.hashCode() == keys10.hashCode());
    }

    public void testEquals()
    {
        assertEquals(keys2, keys1);
        assertEquals(keys1, keys2);

        assertFalse(keys1.equals(keys3));
        assertFalse(keys3.equals(keys1));
        assertFalse(keys1.equals(keys4));
        assertFalse(keys2.equals(keys5));
        assertFalse(keys3.equals(keys4));
        assertFalse(keys4.equals(keys5));

        assertTrue(keys1.equals(keys1));
        assertTrue(keys2.equals(keys2));
        assertTrue(keys3.equals(keys3));
        assertTrue(keys4.equals(keys4));
        assertTrue(keys5.equals(keys5));

        assertFalse(keys1.equals(keys7));
        assertFalse(keys1.equals(keys8));
        assertFalse(keys1.equals(keys9));
        assertFalse(keys1.equals(keys10));
        assertTrue(keys9.equals(keys10));
    }

    public void testGet()
    {
        assertEquals(1, keys6.size());
        assertEquals(2, keys1.size());
        assertEquals(3, keys8.size());
        assertEquals(4, keys9.size());

        assertEquals("a", keys1.get(0));
        assertEquals("b", keys1.get(1));
        assertTrue(null == keys4.get(0));
        assertTrue("d" == keys10.get(3));
    }

    public void testWithSet()
    {
        EventBean[][] testEvents = new EventBean[][] {
            SupportEventBeanFactory.makeEvents(new String[] {"a", "b"}),
            SupportEventBeanFactory.makeEvents(new String[] {"a"}),
            SupportEventBeanFactory.makeEvents(new String[] {"a", "b", "c"}),
            SupportEventBeanFactory.makeEvents(new String[] {"a", "b"}),
        };

        Set<MultiKeyUntyped> mapSet = new HashSet<MultiKeyUntyped>();

        // Test contains
        mapSet.add(new MultiKeyUntyped(testEvents[0]));
        assertTrue(mapSet.contains(new MultiKeyUntyped(testEvents[0])));
        assertFalse(mapSet.contains(new MultiKeyUntyped(testEvents[1])));
        assertFalse(mapSet.contains(new MultiKeyUntyped(testEvents[2])));
        assertFalse(mapSet.contains(new MultiKeyUntyped(testEvents[3])));

        // Test unique
        mapSet.add(new MultiKeyUntyped(testEvents[0]));
        assertEquals(1, mapSet.size());

        mapSet.add(new MultiKeyUntyped(testEvents[1]));
        mapSet.add(new MultiKeyUntyped(testEvents[2]));
        mapSet.add(new MultiKeyUntyped(testEvents[3]));
        assertEquals(4, mapSet.size());

        mapSet.remove(new MultiKeyUntyped(testEvents[0]));
        assertEquals(3, mapSet.size());
        assertFalse(mapSet.contains(new MultiKeyUntyped(testEvents[0])));

        mapSet.remove(new MultiKeyUntyped(testEvents[1]));
        mapSet.remove(new MultiKeyUntyped(testEvents[2]));
        mapSet.remove(new MultiKeyUntyped(testEvents[3]));
        assertEquals(0, mapSet.size());
    }
}