/*
 * Hivemall: Hive scalable Machine Learning Library
 *
 * Copyright (C) 2013
 *   National Institute of Advanced Industrial Science and Technology (AIST)
 *   Registration Number: H25PRO-1520
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */
package hivemall.utils.collections;

import hivemall.utils.lang.mutable.MutableInt;

import java.util.Map;

import org.junit.Test;
import org.junit.Assert;

public class OpenHashMapTest {

    @Test
    public void testPutAndGet() {
        Map<Object, Object> map = new OpenHashMap<Object, Object>(16384);
        final int numEntries = 1000000;
        for(int i = 0; i < numEntries; i++) {
            map.put(Integer.toString(i), i);
        }
        Assert.assertEquals(numEntries, map.size());
        for(int i = 0; i < numEntries; i++) {
            Object v = map.get(Integer.toString(i));
            Assert.assertEquals(i, v);
        }
    }

    @Test
    public void testIterator() {
        OpenHashMap<String, Integer> map = new OpenHashMap<String, Integer>(1000);
        IMapIterator<String, Integer> itor = map.entries();
        Assert.assertFalse(itor.hasNext());

        final int numEntries = 1000000;
        for(int i = 0; i < numEntries; i++) {
            map.put(Integer.toString(i), i);
        }

        itor = map.entries();
        Assert.assertTrue(itor.hasNext());
        while(itor.hasNext()) {
            Assert.assertFalse(itor.next() == -1);
            String k = itor.getKey();
            Integer v = itor.getValue();
            Assert.assertEquals(Integer.valueOf(k), v);
        }
        Assert.assertEquals(-1, itor.next());
    }

    @Test
    public void testIteratorGetProbe() {
        OpenHashMap<String, MutableInt> map = new OpenHashMap<String, MutableInt>(100);
        IMapIterator<String, MutableInt> itor = map.entries();
        Assert.assertFalse(itor.hasNext());

        final int numEntries = 1000000;
        for(int i = 0; i < numEntries; i++) {
            map.put(Integer.toString(i), new MutableInt(i));
        }

        final MutableInt probe = new MutableInt();
        itor = map.entries();
        Assert.assertTrue(itor.hasNext());
        while(itor.hasNext()) {
            Assert.assertFalse(itor.next() == -1);
            String k = itor.getKey();
            itor.getValue(probe);
            Assert.assertEquals(Integer.valueOf(k).intValue(), probe.intValue());
        }
        Assert.assertEquals(-1, itor.next());
    }
}

@@ -984,19 +984,21 @@
 g.junit.
-Tes
+Asser
 t;%0Aimpor
@@ -1009,21 +1009,19 @@
 g.junit.
-Asser
+Tes
 t;%0A%0Apubl
@@ -1504,24 +1504,212 @@
 ;%0A        %7D%0A
+        map.put(Integer.toString(1), Integer.MAX_VALUE);%0A        Assert.assertEquals(Integer.MAX_VALUE, map.get(Integer.toString(1)));%0A        Assert.assertEquals(numEntries, map.size());%0A
     %7D%0A%0A    @

Add more checks in testPutAndGet()