diff options
Diffstat (limited to 'framework/src/maven/apache-maven-3.3.3/maven-compat/src/test/java/org/apache/maven/artifact/testutils/TestFileManager.java')
| -rw-r--r-- | framework/src/maven/apache-maven-3.3.3/maven-compat/src/test/java/org/apache/maven/artifact/testutils/TestFileManager.java | 224 |
1 files changed, 224 insertions, 0 deletions
diff --git a/framework/src/maven/apache-maven-3.3.3/maven-compat/src/test/java/org/apache/maven/artifact/testutils/TestFileManager.java b/framework/src/maven/apache-maven-3.3.3/maven-compat/src/test/java/org/apache/maven/artifact/testutils/TestFileManager.java new file mode 100644 index 00000000..a9f1ab7d --- /dev/null +++ b/framework/src/maven/apache-maven-3.3.3/maven-compat/src/test/java/org/apache/maven/artifact/testutils/TestFileManager.java @@ -0,0 +1,224 @@ +package org.apache.maven.artifact.testutils; + +/* + * 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. + */ + +/* + * 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. + */ + +import java.io.File; +import java.io.IOException; +import java.util.ArrayList; +import java.util.Iterator; +import java.util.List; + +import junit.framework.Assert; + +import org.codehaus.plexus.util.FileUtils; + +public class TestFileManager +{ + + public static final String TEMP_DIR_PATH = System.getProperty( "java.io.tmpdir" ); + + private List<File> filesToDelete = new ArrayList<File>(); + + private final String baseFilename; + + private final String fileSuffix; + + private StackTraceElement callerInfo; + + private Thread cleanupWarning; + + private boolean warnAboutCleanup = false; + + public TestFileManager( String baseFilename, String fileSuffix ) + { + this.baseFilename = baseFilename; + this.fileSuffix = fileSuffix; + + initializeCleanupMonitoring(); + } + + private void initializeCleanupMonitoring() + { + callerInfo = new NullPointerException().getStackTrace()[2]; + + Runnable warning = new Runnable() + { + + public void run() + { + maybeWarnAboutCleanUp(); + } + + }; + + cleanupWarning = new Thread( warning ); + + Runtime.getRuntime().addShutdownHook( cleanupWarning ); + } + + private void maybeWarnAboutCleanUp() + { + if ( warnAboutCleanup ) + { + System.out.println( "[WARNING] TestFileManager from: " + callerInfo.getClassName() + " not cleaned up!" ); + } + } + + public void markForDeletion( File toDelete ) + { + filesToDelete.add( toDelete ); + warnAboutCleanup = true; + } + + public synchronized File createTempDir() + { + try + { + Thread.sleep( 20 ); + } + catch ( InterruptedException e ) + { + // ignore + } + + File dir = new File( TEMP_DIR_PATH, baseFilename + System.currentTimeMillis() ); + + dir.mkdirs(); + markForDeletion( dir ); + + return dir; + } + + public synchronized File createTempFile() + throws IOException + { + File tempFile = File.createTempFile( baseFilename, fileSuffix ); + tempFile.deleteOnExit(); + markForDeletion( tempFile ); + + return tempFile; + } + + public void cleanUp() + throws IOException + { + for ( Iterator it = filesToDelete.iterator(); it.hasNext(); ) + { + File file = (File) it.next(); + + if ( file.exists() ) + { + if ( file.isDirectory() ) + { + FileUtils.deleteDirectory( file ); + } + else + { + file.delete(); + } + } + + it.remove(); + } + + warnAboutCleanup = false; + } + + public void assertFileExistence( File dir, String filename, boolean shouldExist ) + { + File file = new File( dir, filename ); + + if ( shouldExist ) + { + Assert.assertTrue( file.exists() ); + } + else + { + Assert.assertFalse( file.exists() ); + } + } + + public void assertFileContents( File dir, String filename, String contentsTest, String encoding ) + throws IOException + { + assertFileExistence( dir, filename, true ); + + File file = new File( dir, filename ); + + String contents = FileUtils.fileRead( file, encoding ); + + Assert.assertEquals( contentsTest, contents ); + } + + public File createFile( File dir, String filename, String contents, String encoding ) + throws IOException + { + File file = new File( dir, filename ); + + file.getParentFile().mkdirs(); + + FileUtils.fileWrite( file.getPath(), encoding, contents ); + + markForDeletion( file ); + + return file; + } + + public String getFileContents( File file, String encoding ) + throws IOException + { + return FileUtils.fileRead( file, encoding ); + } + + protected void finalize() + throws Throwable + { + maybeWarnAboutCleanUp(); + + super.finalize(); + } + + public File createFile( String filename, String content, String encoding ) + throws IOException + { + File dir = createTempDir(); + return createFile( dir, filename, content, encoding ); + } + +} |