source: sample/hadoop-0.17/tw/org/nchc/tuple/HashMapWritableTest.java @ 20

/*
 * Cloud9: A MapReduce Library for Hadoop
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you
 * may not use this file except in compliance with the License. You may
 * obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0 
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 * implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */

package tw.org.nchc.tuple;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;

import junit.framework.JUnit4TestAdapter;

import org.apache.hadoop.io.FloatWritable;
import org.apache.hadoop.io.IntWritable;
import org.apache.hadoop.io.LongWritable;
import org.apache.hadoop.io.Text;
import org.apache.hadoop.io.Writable;
import org.junit.Test;


public class HashMapWritableTest {

  @Test
  public void testBasic() throws IOException {
    HashMapWritable<Text, IntWritable> map = new HashMapWritable<Text, IntWritable>();

    map.put(new Text("hi"), new IntWritable(5));
    map.put(new Text("there"), new IntWritable(22));

    Text key;
    IntWritable value;

    assertEquals(map.size(), 2);
    
    key=new Text("hi");
    value=map.get(key);
    assertTrue(value!=null);
    assertEquals(value.get(), 5);
    
    value=map.remove(key);
    assertEquals(map.size(), 1);
    
    key=new Text("there");
    value=map.get(key);
    assertTrue(value!=null);
    assertEquals(value.get(), 22);
  }

  @Test
  public void testSerialize1() throws IOException {
    HashMapWritable<Text, IntWritable> origMap = new HashMapWritable<Text, IntWritable>();

    origMap.put(new Text("hi"), new IntWritable(5));
    origMap.put(new Text("there"), new IntWritable(22));
  
    ByteArrayOutputStream bytesOut = new ByteArrayOutputStream();
    DataOutputStream dataOut = new DataOutputStream(bytesOut);

    origMap.write(dataOut);

    HashMapWritable<Text, IntWritable> map = new HashMapWritable<Text, IntWritable>();

    map.readFields(new DataInputStream(new ByteArrayInputStream(
        bytesOut.toByteArray())));

    Text key;
    IntWritable value;

    assertEquals(map.size(), 2);
    
    key=new Text("hi");
    value=map.get(key);
    assertTrue(value!=null);
    assertEquals(value.get(), 5);
    
    value=map.remove(key);
    assertEquals(map.size(), 1);
    
    key=new Text("there");
    value=map.get(key);
    assertTrue(value!=null);
    assertEquals(value.get(), 22);
  }

  @Test
  public void testSerialize2() throws IOException {
    HashMapWritable<Text, LongWritable> origMap = new HashMapWritable<Text, LongWritable>();

    origMap.put(new Text("hi"), new LongWritable(52));
    origMap.put(new Text("there"), new LongWritable(77));
  
    ByteArrayOutputStream bytesOut = new ByteArrayOutputStream();
    DataOutputStream dataOut = new DataOutputStream(bytesOut);

    origMap.write(dataOut);

    HashMapWritable<Text, LongWritable> map = new HashMapWritable<Text, LongWritable>();

    map.readFields(new DataInputStream(new ByteArrayInputStream(
        bytesOut.toByteArray())));

    Text key;
    LongWritable value;

    assertEquals(map.size(), 2);
    
    key=new Text("hi");
    value=map.get(key);
    assertTrue(value!=null);
    assertEquals(value.get(), 52);
    
    value=map.remove(key);
    assertEquals(map.size(), 1);
    
    key=new Text("there");
    value=map.get(key);
    assertTrue(value!=null);
    assertEquals(value.get(), 77);
  }


  @Test
  public void testTypeSafety() throws IOException {
    HashMapWritable<Writable, Writable> origMap = new HashMapWritable<Writable, Writable>();

    origMap.put(new Text("hi"), new FloatWritable(5.3f));
    origMap.put(new Text("there"), new Text("bbb"));
  
    ByteArrayOutputStream bytesOut = new ByteArrayOutputStream();
    DataOutputStream dataOut = new DataOutputStream(bytesOut);

    origMap.write(dataOut);

    HashMapWritable<Writable, Writable> map = new HashMapWritable<Writable, Writable>();

    try {
          map.readFields(new DataInputStream(new ByteArrayInputStream(
              bytesOut.toByteArray())));
          assertTrue(false);
        } catch (Exception e) {
        }
  }



  @Test
  public void testSerializeEmpty() throws IOException {
    HashMapWritable<IntWritable, Text> map = new HashMapWritable<IntWritable, Text>();
    
    assertTrue(map.size() == 0);
    
    ByteArrayOutputStream bytesOut = new ByteArrayOutputStream();
    DataOutputStream dataOut = new DataOutputStream(bytesOut);

    map.write(dataOut);

    HashMapWritable<IntWritable, Text> newList = new HashMapWritable<IntWritable, Text>();
    newList.readFields(new DataInputStream(new ByteArrayInputStream(
        bytesOut.toByteArray())));
    assertTrue(newList.size() == 0);
  }
  
  public static junit.framework.Test suite() {
    return new JUnit4TestAdapter(HashMapWritableTest.class);
  }

}