TestIFileStreams.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.mapred;
  19. import org.apache.hadoop.fs.ChecksumException;
  20. import org.apache.hadoop.io.DataInputBuffer;
  21. import org.apache.hadoop.io.DataOutputBuffer;
  22. import junit.framework.TestCase;
  23. public class TestIFileStreams extends TestCase {
  24.   public void testIFileStream() throws Exception {
  25.     final int DLEN = 100;
  26.     DataOutputBuffer dob = new DataOutputBuffer(DLEN + 4);
  27.     IFileOutputStream ifos = new IFileOutputStream(dob);
  28.     for (int i = 0; i < DLEN; ++i) {
  29.       ifos.write(i);
  30.     }
  31.     ifos.close();
  32.     DataInputBuffer dib = new DataInputBuffer();
  33.     dib.reset(dob.getData(), DLEN + 4);
  34.     IFileInputStream ifis = new IFileInputStream(dib, 104);
  35.     for (int i = 0; i < DLEN; ++i) {
  36.       assertEquals(i, ifis.read());
  37.     }
  38.     ifis.close();
  39.   }
  40.   public void testBadIFileStream() throws Exception {
  41.     final int DLEN = 100;
  42.     DataOutputBuffer dob = new DataOutputBuffer(DLEN + 4);
  43.     IFileOutputStream ifos = new IFileOutputStream(dob);
  44.     for (int i = 0; i < DLEN; ++i) {
  45.       ifos.write(i);
  46.     }
  47.     ifos.close();
  48.     DataInputBuffer dib = new DataInputBuffer();
  49.     final byte[] b = dob.getData();
  50.     ++b[17];
  51.     dib.reset(b, DLEN + 4);
  52.     IFileInputStream ifis = new IFileInputStream(dib, 104);
  53.     int i = 0;
  54.     try {
  55.       while (i < DLEN) {
  56.         if (17 == i) {
  57.           assertEquals(18, ifis.read());
  58.         } else {
  59.           assertEquals(i, ifis.read());
  60.         }
  61.         ++i;
  62.       }
  63.       ifis.close();
  64.     } catch (ChecksumException e) {
  65.       assertEquals("Unexpected bad checksum", DLEN - 1, i);
  66.       return;
  67.     }
  68.     fail("Did not detect bad data in checksum");
  69.   }
  70.   public void testBadLength() throws Exception {
  71.     final int DLEN = 100;
  72.     DataOutputBuffer dob = new DataOutputBuffer(DLEN + 4);
  73.     IFileOutputStream ifos = new IFileOutputStream(dob);
  74.     for (int i = 0; i < DLEN; ++i) {
  75.       ifos.write(i);
  76.     }
  77.     ifos.close();
  78.     DataInputBuffer dib = new DataInputBuffer();
  79.     dib.reset(dob.getData(), DLEN + 4);
  80.     IFileInputStream ifis = new IFileInputStream(dib, 100);
  81.     int i = 0;
  82.     try {
  83.       while (i < DLEN - 8) {
  84.         assertEquals(i++, ifis.read());
  85.       }
  86.       ifis.close();
  87.     } catch (ChecksumException e) {
  88.       assertEquals("Checksum before close", i, DLEN - 8);
  89.       return;
  90.     }
  91.     fail("Did not detect bad data in checksum");
  92.   }
  93. }