00001 00005 package edu.mit.csail.sdg.squander.examples.javacol; 00006 00007 import static edu.mit.csail.sdg.squander.examples.javacol.Collections.*; 00008 00009 import java.util.HashMap; 00010 import java.util.Map; 00011 00012 import org.junit.Assert; 00013 import org.junit.Before; 00014 import org.junit.Test; 00015 00016 public class MapTest { 00017 00018 private static final String[] keys = new String[] { "str1", "str2", "str3" }; 00019 private static final Integer[] vals = new Integer[] { 1, 2, 3 }; 00020 00021 private Map<String, Integer> map; 00022 private Map<String, Integer> sqMap; 00023 00024 @Before 00025 public void setUp() { 00026 map = new HashMap<String, Integer>(); 00027 sqMap = new HashMap<String, Integer>(); 00028 for (int i = 0; i < keys.length; i++) { 00029 map.put(keys[i], vals[i]); 00030 sqMap.put(keys[i], vals[i]); 00031 } 00032 } 00033 00034 @Test 00035 public void testGet() { 00036 for (String key : keys) { 00037 Integer v1 = map.get(key); 00038 Integer v2 = get(sqMap, key); 00039 Assert.assertEquals(v1, v2); 00040 Assert.assertEquals(map.toString(), sqMap.toString()); 00041 } 00042 } 00043 00044 @Test 00045 public void testPut() { 00046 map = new HashMap<String, Integer>(); 00047 sqMap = new HashMap<String, Integer>(); 00048 for (int i = 0; i < keys.length; i++) { 00049 String key = keys[i]; 00050 Integer val = vals[i]; 00051 Integer v1 = map.put(key, val); 00052 Integer v2 = put(sqMap, key, val); 00053 Assert.assertEquals(v1, v2); 00054 Assert.assertEquals(map.toString(), sqMap.toString()); 00055 } 00056 for (int i = 0; i < keys.length; i++) { 00057 String key = keys[i]; 00058 Integer val = vals[i]; 00059 Integer v1 = map.put(key, val); 00060 Integer v2 = put(sqMap, key, val); 00061 Assert.assertEquals(v1, v2); 00062 Assert.assertEquals(map.toString(), sqMap.toString()); 00063 } 00064 } 00065 00066 @Test 00067 public void testRemove() { 00068 for (int i = 0; i < keys.length; i++) { 00069 String key = keys[i]; 00070 Integer v1 = map.remove(key); 00071 Integer v2 = removeKey(sqMap, key); 00072 Assert.assertEquals(v1, v2); 00073 Assert.assertEquals(map.toString(), sqMap.toString()); 00074 } 00075 } 00076 00077 }