Merge pull request #5824 from halibobo1205/test/toolkits

test(plugins/DbCopy): solve directory creation failure due to limited permissions
tronprotocol · May 31, 2024 · 3a3a37e · 3a3a37e
2 parents afd15a7 + e6f14ed
commit 3a3a37e
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();
+  }
 }