TestPendingReplication.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.hdfs.server.namenode;
  19. import junit.framework.TestCase;
  20. import java.lang.System;
  21. import org.apache.hadoop.hdfs.protocol.Block;
  22. /**
  23.  * This class tests the internals of PendingReplicationBlocks.java
  24.  */
  25. public class TestPendingReplication extends TestCase {
  26.   public void testPendingReplication() {
  27.     int timeout = 10; // 10 seconds
  28.     PendingReplicationBlocks pendingReplications;
  29.     pendingReplications = new PendingReplicationBlocks(timeout * 1000);
  30.     //
  31.     // Add 10 blocks to pendingReplications.
  32.     //
  33.     for (int i = 0; i < 10; i++) {
  34.       Block block = new Block(i, i, 0);
  35.       pendingReplications.add(block, i);
  36.     }
  37.     assertEquals("Size of pendingReplications ",
  38.                  10, pendingReplications.size());
  39.     //
  40.     // remove one item and reinsert it
  41.     //
  42.     Block blk = new Block(8, 8, 0);
  43.     pendingReplications.remove(blk);             // removes one replica
  44.     assertEquals("pendingReplications.getNumReplicas ",
  45.                  7, pendingReplications.getNumReplicas(blk));
  46.     for (int i = 0; i < 7; i++) {
  47.       pendingReplications.remove(blk);           // removes all replicas
  48.     }
  49.     assertTrue(pendingReplications.size() == 9);
  50.     pendingReplications.add(blk, 8);
  51.     assertTrue(pendingReplications.size() == 10);
  52.     //
  53.     // verify that the number of replicas returned
  54.     // are sane.
  55.     //
  56.     for (int i = 0; i < 10; i++) {
  57.       Block block = new Block(i, i, 0);
  58.       int numReplicas = pendingReplications.getNumReplicas(block);
  59.       assertTrue(numReplicas == i);
  60.     }
  61.     //
  62.     // verify that nothing has timed out so far
  63.     //
  64.     assertTrue(pendingReplications.getTimedOutBlocks() == null);
  65.     //
  66.     // Wait for one second and then insert some more items.
  67.     //
  68.     try {
  69.       Thread.sleep(1000);
  70.     } catch (Exception e) {
  71.     }
  72.     for (int i = 10; i < 15; i++) {
  73.       Block block = new Block(i, i, 0);
  74.       pendingReplications.add(block, i);
  75.     }
  76.     assertTrue(pendingReplications.size() == 15);
  77.     //
  78.     // Wait for everything to timeout.
  79.     //
  80.     int loop = 0;
  81.     while (pendingReplications.size() > 0) {
  82.       try {
  83.         Thread.sleep(1000);
  84.       } catch (Exception e) {
  85.       }
  86.       loop++;
  87.     }
  88.     System.out.println("Had to wait for " + loop +
  89.                        " seconds for the lot to timeout");
  90.     //
  91.     // Verify that everything has timed out.
  92.     //
  93.     assertEquals("Size of pendingReplications ",
  94.                  0, pendingReplications.size());
  95.     Block[] timedOut = pendingReplications.getTimedOutBlocks();
  96.     assertTrue(timedOut != null && timedOut.length == 15);
  97.     for (int i = 0; i < timedOut.length; i++) {
  98.       assertTrue(timedOut[i].getBlockId() < 15);
  99.     }
  100.   }
  101. }