Mirrors_Framework/ghidra
1
0
Fork 0
mirror of https://github.com/NationalSecurityAgency/ghidra.git synced 2026-06-02 07:00:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'GP-0_ghidragon_fixed_close_tool_test'

Browse Source
This commit is contained in:
ghidragon
2023-06-09 15:05:33 -04:00
parent 5574923e23 dea7046785
commit 25597c2e46
1 changed files with 37 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Ghidra/Features/Base/src/test.slow/java/ghidra/framework/project/tool/CloseToolTest.java
+37 -5
View File
@@ -22,6 +22,7 @@ import java.awt.Window;
import org.junit.*;
import docking.action.DockingActionIf;
import docking.widgets.OptionDialog;
import ghidra.app.context.ProgramActionContext;
import ghidra.app.plugin.core.progmgr.ProgramManagerPlugin;
import ghidra.framework.cmd.BackgroundCommand;
@@ -114,7 +115,7 @@ public class CloseToolTest extends AbstractGhidraHeadedIntegrationTest {
}
@Test
public void testCannotCloseToolWithBackgroundTaskRunning() throws Exception {
public void testCloseToolWithBackgroundTaskRunningAnswerNo() throws Exception {
// open a program
ProgramDB program = new ProgramBuilder("notepad", ProgramBuilder._TOY).getProgram();
@@ -130,15 +131,15 @@ public class CloseToolTest extends AbstractGhidraHeadedIntegrationTest {
closeTool(tool);
// check for warning dialog
Window window = waitForWindow("Tool Busy");
assertNotNull("Did not get tool busy dialog", window);
closeWindow(window);
OptionDialog dialog = waitForDialogComponent(OptionDialog.class);
assertNotNull("Did not get option dialog to terminate tasks", dialog);
pressButtonByText(dialog, "No");
// try to close the program
closeProgram(tool, program);
// check for warning dialog
window = waitForWindow("Close notepad Failed");
Window window = waitForWindow("Close notepad Failed");
assertNotNull("Did not get \"close failed\" dialog", window);
closeWindow(window);
@@ -151,6 +152,37 @@ public class CloseToolTest extends AbstractGhidraHeadedIntegrationTest {
assertNull("Tool did not close after task", tool.getToolFrame());
}
@Test
public void testCloseToolWithBackgroundTaskRunningAnswerYes() throws Exception {
// open a program
ProgramDB program = new ProgramBuilder("notepad", ProgramBuilder._TOY).getProgram();
// launch a tool with the program
PluginTool tool = env.launchDefaultTool(program);
// start a background task that will run until we tell it to finish.
ControllableBackgroundCommand cmd = new ControllableBackgroundCommand();
tool.executeBackgroundCommand(cmd, program);
waitForCommandToStart(cmd);
// try to close the tool
closeTool(tool);
// check for warning dialog
OptionDialog dialog = waitForDialogComponent(OptionDialog.class);
assertNotNull("Did not get option dialog to close tool with running task", dialog);
pressButtonByText(dialog, "Yes");
// check for another warning about closing the program with running tasks
dialog = waitForDialogComponent(OptionDialog.class);
assertNotNull("Did not get option dialog to close program with running task", dialog);
pressButtonByText(dialog, "Close!");
waitForSwing();
assertNull("Tool did not close after task", tool.getToolFrame());
stopBackgroundCommand(tool, cmd);
}
private void closeWindow(final Window window) {
runSwing(() -> window.dispose());
}