package org.webslinger.util;

 import java.util.ArrayList;

 import java.util.Arrays;

 import java.util.Collection;

 import java.util.HashMap;

 import java.util.HashSet;

 import java.util.Iterator;

 import java.util.LinkedHashMap;

 import java.util.List;

 import java.util.Map;

 import junit.framework.TestCase;

 import org.webslinger.collections.ArrayUtil;

 public abstract class GenericTestCase extends TestCase {

     protected GenericTestCase(String name) {

         super(name);

     }

     public static <T> void assertEquals(List<T> wanted, Object got) {

         assertEquals(null, wanted, got);

     }

     public static <T> void assertEquals(String msg, List<T> wanted, Object got) {

         msg = msg == null ? "" : msg + ' ';

         assertNotNull(msg + "expected a value", got);

         if (got.getClass().isArray()) got = ArrayUtil.asList(got);

         if (!(got instanceof Collection)) fail(msg + "expected a collection, got a " + got.getClass());

         Iterator<T> leftIt = wanted.iterator();

         Iterator rightIt = ((Collection) got).iterator();

         int i = 0;

         while (leftIt.hasNext() && rightIt.hasNext()) {

             T left = leftIt.next();

             Object right = rightIt.next();

             assertEquals(msg + "item " + i, left, right);

         }

         assertFalse(msg + "not enough items", leftIt.hasNext());

         assertFalse(msg + "too many items", rightIt.hasNext());

     }

     public static <T> void assertEquals(Map<T, ?> wanted, Object got) {

         assertEquals(null, wanted, got);

     }

     public static <T> void assertEquals(String msg, Map<T, ?> wanted, Object got) {

         msg = msg == null ? "" : msg + ' ';

         assertNotNull(msg + "expected a value", got);

         if (!(got instanceof Map)) fail(msg + "expected a map");

         Map<?, ?> gotMap = (Map) got;

         HashSet<T> leftKeys = new HashSet<T>(wanted.keySet());

         HashSet<Object> rightKeys = new HashSet<Object>(gotMap.keySet());

         for(T key: leftKeys) {

             assertTrue(msg + "got key(" + key + ")", rightKeys.remove(key));

             assertEquals(msg + "key(" + key + ") value", wanted.get(key), gotMap.get(key));

         }

         assertTrue(msg + "extra entries", rightKeys.isEmpty());

     }

     public static void assertEquals(String msg, String wanted, String got) {

         TestCase.assertEquals(msg, wanted, got);

     }

     public static void assertEquals(Object wanted, Object got) {

         assertEquals(null, wanted, got);

     }

     public static void assertEquals(String msg, Object wanted, Object got) {

         if (wanted instanceof List) {

             assertEquals(msg, (List<?>) wanted, got);

         } else if (wanted instanceof Map) {

             assertEquals(msg, (Map<?, ?>) wanted, got);

         } else if (wanted == null) {

             TestCase.assertEquals(msg, wanted, got);

         } else if (wanted.getClass().isArray()) {

             assertEquals(msg, ArrayUtil.asList(wanted), got);

         } else {

             TestCase.assertEquals(msg, wanted, got);

         }

     }

     protected static <T> List<T> list(T value) {

         ArrayList<T> list = new ArrayList<T>(1);

         list.add(value);

         return list;

     }

     protected static <T> List<T> list(T[] list) {

         return Arrays.asList(list);

     }

     protected static Map<?, ?> map(Object[] list) {

         assertEquals("list has even number of elements", 0, list.length % 2);

         Map<Object, Object> map = new LinkedHashMap<Object, Object>();

         for (int i = 0; i < list.length; i += 2) {

             map.put(list[i], list[i + 1]);

         }

         return map;

     }

 }