slider-core/src/test/groovy/org/apache/slider/other/TestLocalDirStatus.groovy

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.slider.other

import groovy.transform.CompileStatic
import org.apache.slider.test.SliderTestUtils
import org.apache.slider.tools.TestUtility
import org.junit.Test

/**
 * This test exists to diagnose local FS permissions
 */
@CompileStatic
class TestLocalDirStatus extends SliderTestUtils {


  public static final int SIZE = 0x200000
  
  @Test
  public void testTempDir() throws Throwable {
    File tmpf = null
    try {
      tmpf = File.createTempFile("testl", ".bin")
      createAndReadFile(tmpf, SIZE)
      tmpf.delete()
      assert !tmpf.exists()
    } finally {
      tmpf?.delete()
    }
  }
  
  @Test
  public void testTargetDir() throws Throwable {
    File target = target()
    File tmpf = null
    try {
      tmpf = File.createTempFile("testl", ".bin", target)
      createAndReadFile(tmpf, SIZE)
      tmpf.delete()
      assert !tmpf.exists()
    } finally {
      tmpf?.delete()
    }
  }

  public File target() {
    File target = new File("target").absoluteFile
    assert target.exists()
    return target
  }

  @Test
  public void testRenameInTargetDir() throws Throwable {
    def target = target()
    File tmpf = null, dst= null
    try {
      tmpf = File.createTempFile("testl", ".bin", target)
      dst = File.createTempFile("test-dest", ".bin", target)
      createRenameAndReadFile(tmpf, dst, SIZE)
      assert !tmpf.exists()
      dst.delete()
    } finally {
      tmpf?.delete()
      dst?.delete()
    } 
  }
  
  @Test
  public void testRenameInTmpDir() throws Throwable {
    def target = target()
    File tmpf = null, dst= null
    try {
      tmpf = File.createTempFile("testl", ".bin")
      dst = File.createTempFile("test-dest", ".bin")
      createRenameAndReadFile(tmpf, dst, SIZE)
      assert !tmpf.exists()
      dst.delete()
    } finally {
      tmpf?.delete()
      dst?.delete()
    } 
  }
  
  protected void createAndReadFile(File path, int len) {
    byte[] dataset = TestUtility.dataset(len, 32, 128)
    writeFile(path, dataset)
    assert path.exists()
    assert path.length() == len
    def persisted = readFile(path)
    TestUtility.compareByteArrays(dataset, persisted, len)
  }  
  
  protected void createRenameAndReadFile(File src, File dst , int len) {
    byte[] dataset = TestUtility.dataset(len, 32, 128)
    writeFile(src, dataset)
    assert src.exists()
    assert src.length() == len
    dst.delete()
    assert !dst.exists()
    assert src.renameTo(dst)
    assert dst.length() == len
    def persisted = readFile(dst)
    TestUtility.compareByteArrays(dataset, persisted, len)
  }
  
  protected void writeFile(File path, byte[] dataset) {
    def out = new FileOutputStream(path)
    try {
      out.write(dataset)
      out.flush()
    } finally {
      out.close()
    }
  }  
  
  protected byte[] readFile(File path) {
    assert path.absoluteFile.exists()
    assert path.absoluteFile.isFile()
    int len = (int)path.length()
    byte[] dataset = new byte[len]
    def ins = new FileInputStream(path)
    try {
      ins.read(dataset)
    } finally {
      ins.close()
    }
    return dataset
  }
  
  
}