1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
/*
* Copyright (c) 2009, 2021, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
/*
* @test
* @bug 6829503
* @summary 1) Test Console and DeleteOnExitHook can be initialized
* while shutdown is in progress
* 2) Test if files that are added by the application shutdown
* hook are deleted on exit during shutdown
* @library /test/lib
* @run testng/othervm ShutdownHooks
*/
import jdk.test.lib.process.ProcessTools;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.Test;
import java.io.*;
import java.nio.file.Files;
import static jdk.test.lib.Asserts.assertFalse;
public class ShutdownHooks {
private static final String TEST_FILE_NAME = "fileToBeDeleted";
private static final File TEST_FILE = new File(TEST_FILE_NAME);
private static final String TEST_CLASSES = System.getProperty(
"test.classes", ".");
@BeforeTest
public static void setUp() throws Exception {
// Make sure file does not exist before test
Files.deleteIfExists(TEST_FILE.toPath());
}
@Test
public void testShutdownHooks() throws Exception {
// Run in a new process in order to evaluate shutdown hook results
String[] testCommand = new String[] {"-classpath", TEST_CLASSES,
ShutdownHooksProcess.class.getName()};
ProcessTools.executeTestJava(testCommand).shouldHaveExitValue(0);
String errorMsg = "File exists despite shutdown hook has been run";
assertFalse(Files.exists(TEST_FILE.toPath()), errorMsg);
}
// This class main will be the entry point for test subprocesses
static class ShutdownHooksProcess {
public static void main(String[] args) throws Exception {
Runtime.getRuntime().addShutdownHook(new Cleaner());
System.out.println("Writing to "+ TEST_FILE);
try (PrintWriter pw = new PrintWriter(TEST_FILE)) {
pw.println("Shutdown begins");
}
}
static class Cleaner extends Thread {
public void run() {
// register the Console's shutdown hook while the application
// shutdown hook is running
Console cons = System.console();
// register the DeleteOnExitHook while the application
// shutdown hook is running
TEST_FILE.deleteOnExit();
try (PrintWriter pw = new PrintWriter(TEST_FILE)) {
pw.println("File is being deleted");
} catch (FileNotFoundException e) {
throw new RuntimeException(e);
}
}
}
}
}
|
