UtilTest.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.streaming;
  19. import java.io.File;
  20. import java.io.FileOutputStream;
  21. import java.io.IOException;
  22. import java.io.PrintStream;
  23. class UtilTest {
  24.   /**
  25.    * Utility routine to recurisvely delete a directory.
  26.    * On normal return, the file does not exist.
  27.    *
  28.    * @param file File or directory to delete.
  29.    *
  30.    * @throws RuntimeException if the file, or some file within
  31.    * it, could not be deleted.
  32.    */
  33.   static void recursiveDelete(File file) {
  34.     file = file.getAbsoluteFile();
  35.     if (!file.exists()) return;
  36.     
  37.     if (file.isDirectory()) {
  38.       for (File child : file.listFiles()) {
  39. recursiveDelete(child);
  40.       }
  41.     }
  42.     if (!file.delete()) {
  43.       throw new RuntimeException("Failed to delete " + file);
  44.     }
  45.   }
  46.   
  47.   public UtilTest(String testName) {
  48.     testName_ = testName;
  49.     userDir_ = System.getProperty("user.dir");
  50.     antTestDir_ = System.getProperty("test.build.data", userDir_);
  51.     System.out.println("test.build.data-or-user.dir=" + antTestDir_);
  52.   }
  53.   void checkUserDir() {
  54.     // trunk/src/contrib/streaming --> trunk/build/contrib/streaming/test/data
  55.     if (!userDir_.equals(antTestDir_)) {
  56.       // because changes to user.dir are ignored by File static methods.
  57.       throw new IllegalStateException("user.dir != test.build.data. The junit Ant task must be forked.");
  58.     }
  59.   }
  60.   void redirectIfAntJunit() throws IOException
  61.   {
  62.     boolean fromAntJunit = System.getProperty("test.build.data") != null;
  63.     if (fromAntJunit) {
  64.       new File(antTestDir_).mkdirs();
  65.       File outFile = new File(antTestDir_, testName_+".log");
  66.       PrintStream out = new PrintStream(new FileOutputStream(outFile));
  67.       System.setOut(out);
  68.       System.setErr(out);
  69.     }
  70.   }
  71.   private String userDir_;
  72.   private String antTestDir_;
  73.   private String testName_;
  74. }