TestDU.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.fs;
  19. import junit.framework.TestCase;
  20. import java.io.File;
  21. import java.io.IOException;
  22. import java.io.RandomAccessFile;
  23. import java.util.Random;
  24. /** This test makes sure that "DU" does not get to run on each call to getUsed */ 
  25. public class TestDU extends TestCase {
  26.   final static private File DU_DIR = new File(
  27.       System.getProperty("test.build.data","/tmp"), "dutmp");
  28.   public void setUp() throws IOException {
  29.       FileUtil.fullyDelete(DU_DIR);
  30.       assertTrue(DU_DIR.mkdirs());
  31.   }
  32.   public void tearDown() throws IOException {
  33.       FileUtil.fullyDelete(DU_DIR);
  34.   }
  35.     
  36.   private void createFile(File newFile, int size) throws IOException {
  37.     // write random data so that filesystems with compression enabled (e.g., ZFS)
  38.     // can't compress the file
  39.     Random random = new Random();
  40.     byte[] data = new byte[size];
  41.     random.nextBytes(data);
  42.     newFile.createNewFile();
  43.     RandomAccessFile file = new RandomAccessFile(newFile, "rws");
  44.     file.write(data);
  45.       
  46.     file.getFD().sync();
  47.     file.close();
  48.   }
  49.   /**
  50.    * Verify that du returns expected used space for a file.
  51.    * We assume here that if a file system crates a file of size 
  52.    * that is a multiple of the block size in this file system,
  53.    * then the used size for the file will be exactly that size.
  54.    * This is true for most file systems.
  55.    * 
  56.    * @throws IOException
  57.    * @throws InterruptedException
  58.    */
  59.   public void testDU() throws IOException, InterruptedException {
  60.     int writtenSize = 32*1024;   // writing 32K
  61.     File file = new File(DU_DIR, "data");
  62.     createFile(file, writtenSize);
  63.     Thread.sleep(5000); // let the metadata updater catch up
  64.     
  65.     DU du = new DU(file, 10000);
  66.     du.start();
  67.     long duSize = du.getUsed();
  68.     du.shutdown();
  69.     assertEquals(writtenSize, duSize);
  70.     
  71.     //test with 0 interval, will not launch thread 
  72.     du = new DU(file, 0);
  73.     du.start();
  74.     duSize = du.getUsed();
  75.     du.shutdown();
  76.     
  77.     assertEquals(writtenSize, duSize);  
  78.     
  79.     //test without launching thread 
  80.     du = new DU(file, 10000);
  81.     duSize = du.getUsed();
  82.     
  83.     assertEquals(writtenSize, duSize);     
  84.   }
  85. }