TestWritable.java
上传用户:quxuerui
上传日期:2018-01-08
资源大小:41811k
文件大小:3k
源码类别:

网格计算

开发平台:

Java

  1. /**
  2.  * Licensed to the Apache Software Foundation (ASF) under one
  3.  * or more contributor license agreements.  See the NOTICE file
  4.  * distributed with this work for additional information
  5.  * regarding copyright ownership.  The ASF licenses this file
  6.  * to you under the Apache License, Version 2.0 (the
  7.  * "License"); you may not use this file except in compliance
  8.  * with the License.  You may obtain a copy of the License at
  9.  *
  10.  *     http://www.apache.org/licenses/LICENSE-2.0
  11.  *
  12.  * Unless required by applicable law or agreed to in writing, software
  13.  * distributed under the License is distributed on an "AS IS" BASIS,
  14.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15.  * See the License for the specific language governing permissions and
  16.  * limitations under the License.
  17.  */
  18. package org.apache.hadoop.io;
  19. import java.io.*;
  20. import java.util.Random;
  21. import org.apache.hadoop.conf.Configurable;
  22. import org.apache.hadoop.conf.Configuration;
  23. import org.apache.hadoop.util.ReflectionUtils;
  24. import junit.framework.TestCase;
  25. /** Unit tests for Writable. */
  26. public class TestWritable extends TestCase {
  27.   public TestWritable(String name) { super(name); }
  28.   /** Example class used in test cases below. */
  29.   public static class SimpleWritable implements Writable {
  30.     private static final Random RANDOM = new Random();
  31.     int state = RANDOM.nextInt();
  32.     public void write(DataOutput out) throws IOException {
  33.       out.writeInt(state);
  34.     }
  35.     public void readFields(DataInput in) throws IOException {
  36.       this.state = in.readInt();
  37.     }
  38.     public static SimpleWritable read(DataInput in) throws IOException {
  39.       SimpleWritable result = new SimpleWritable();
  40.       result.readFields(in);
  41.       return result;
  42.     }
  43.     /** Required by test code, below. */
  44.     public boolean equals(Object o) {
  45.       if (!(o instanceof SimpleWritable))
  46.         return false;
  47.       SimpleWritable other = (SimpleWritable)o;
  48.       return this.state == other.state;
  49.     }
  50.   }
  51.   /** Test 1: Check that SimpleWritable. */
  52.   public void testSimpleWritable() throws Exception {
  53.     testWritable(new SimpleWritable());
  54.   }
  55.   
  56.   public void testByteWritable() throws Exception {
  57.     testWritable(new ByteWritable((byte)128));
  58.   }
  59.   public void testDoubleWritable() throws Exception {
  60.     testWritable(new DoubleWritable(1.0));
  61.   }
  62.   /** Utility method for testing writables. */
  63.   public static Writable testWritable(Writable before) 
  64.    throws Exception {
  65.    return testWritable(before, null);
  66.   }
  67.   
  68.   /** Utility method for testing writables. */
  69.   public static Writable testWritable(Writable before
  70.    , Configuration conf) throws Exception {
  71.     DataOutputBuffer dob = new DataOutputBuffer();
  72.     before.write(dob);
  73.     DataInputBuffer dib = new DataInputBuffer();
  74.     dib.reset(dob.getData(), dob.getLength());
  75.     
  76.     Writable after = (Writable)ReflectionUtils.newInstance(
  77.      before.getClass(), conf);
  78.     after.readFields(dib);
  79.     assertEquals(before, after);
  80.     return after;
  81.   }
  82. }