test(plugins/DbCopy): solve directory creation failure due to limited…

… permissions
halibobo1205 · May 20, 2024 · e6f14ed · e6f14ed
1 parent e7c8d40
commit e6f14ed
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/plugins/src/test/java/org/tron/plugins/DbCopyTest.java b/plugins/src/test/java/org/tron/plugins/DbCopyTest.java
@@ -11,7 +11,7 @@ public class DbCopyTest extends DbTest {
  @Test
  public void testRun() {
  String[] args = new String[] { "db", "cp", INPUT_DIRECTORY,
- tmpDir + UUID.randomUUID()};
+ genarateTmpDir()};
  Assert.assertEquals(0, cli.execute(args));
  }
 
@@ -32,7 +32,7 @@ public void testNotExist() {
  @Test
  public void testEmpty() throws IOException {
  String[] args = new String[] {"db", "cp", temporaryFolder.newFolder().toString(),
- tmpDir + UUID.randomUUID()};
+ genarateTmpDir()};
  Assert.assertEquals(0, cli.execute(args));
  }
 
@@ -46,7 +46,7 @@ public void testDestIsExist() throws IOException {
  @Test
  public void testSrcIsFile() throws IOException {
  String[] args = new String[] {"db", "cp", temporaryFolder.newFile().toString(),
- tmpDir + UUID.randomUUID()};
+ genarateTmpDir()};
  Assert.assertEquals(403, cli.execute(args));
  }
 

diff --git a/plugins/src/test/java/org/tron/plugins/DbTest.java b/plugins/src/test/java/org/tron/plugins/DbTest.java
@@ -86,4 +86,14 @@ private static void initDB(File file) throws IOException {
  }
  }
  }
+
+ /**
+ * Generate a not-exist temporary directory path.
+ * @return temporary path
+ */
+ public String genarateTmpDir() {
+ File dir = Paths.get(tmpDir, UUID.randomUUID().toString()).toFile();
+ dir.deleteOnExit();
+ return dir.getPath();
+ }
 }