Git-Mirrors/haveno
1
0
Fork 0
mirror of https://github.com/haveno-dex/haveno.git synced 2025-04-20 07:46:05 -04:00
Code Issues Packages Projects Releases Wiki Activity

Only keep peer keys when reusing network stress test directory

Browse Source
This commit is contained in:
Ivan Vilata-i-Balaguer 2016-04-27 14:10:20 +02:00
parent 6c5c3371e6
commit f100ef6384
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
network/src/test/java/io/bitsquare/p2p/network/NetworkStressTest.java
View File

@ -173,20 +173,30 @@ public class NetworkStressTest {
return stressTestDirPath;
}
/**
* Delete the test data directory recursively, unless <code>STRESS_TEST_DIR</code> is defined,
* in which case peer node keys are kept.
*
* @throws IOException
*/
private void deleteTestDataDirectory() throws IOException {
// Based on <https://stackoverflow.com/a/27917071/6239236> by Tomasz Dzięcielewski.
if (System.getenv(TEST_DIR_ENVVAR) != null)
return; // do not remove if given explicitly
boolean keep = System.getenv(TEST_DIR_ENVVAR) != null; // do not remove if given explicitly
Files.walkFileTree(testDataDir, new SimpleFileVisitor<Path>() {
@Override
public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
Files.delete(file);
final String fileName = file.getFileName().toString();
if (!(keep && (fileName.equals("enc.key") || fileName.equals("sig.key"))))
Files.delete(file);
return FileVisitResult.CONTINUE;
}
@Override
public FileVisitResult postVisitDirectory(Path dir, IOException exc) throws IOException {
Files.delete(dir);
// ``dir`` is always a directory, I/O errors may still trigger ``NullPointerException``.
//noinspection ConstantConditions
if (!(keep && dir.toFile().listFiles().length > 0))
Files.delete(dir);
return FileVisitResult.CONTINUE;
}
});