diff --git a/mRemoteNGTests/Tree/SelectedConnectionDeletionConfirmerTests.cs b/mRemoteNGTests/Tree/SelectedConnectionDeletionConfirmerTests.cs index d7e9108c5..a881f3c57 100644 --- a/mRemoteNGTests/Tree/SelectedConnectionDeletionConfirmerTests.cs +++ b/mRemoteNGTests/Tree/SelectedConnectionDeletionConfirmerTests.cs @@ -1,7 +1,7 @@ -using System.Windows.Forms; -using mRemoteNG.Connection; +using mRemoteNG.Connection; using mRemoteNG.Tree; using NUnit.Framework; +using System.Windows.Forms; namespace mRemoteNGTests.Tree @@ -30,12 +30,12 @@ namespace mRemoteNGTests.Tree Assert.That(deletionConfirmer.Confirm(_testConnectionInfo), Is.False); } - private DialogResult MockClickYes(string promptMessage, string title, MessageBoxButtons buttons, MessageBoxIcon icon) + private DialogResult MockClickYes(string promptMessage) { return DialogResult.Yes; } - private DialogResult MockClickNo(string promptMessage, string title, MessageBoxButtons buttons, MessageBoxIcon icon) + private DialogResult MockClickNo(string promptMessage) { return DialogResult.No; } diff --git a/mRemoteV1/Tree/SelectedConnectionDeletionConfirmer.cs b/mRemoteV1/Tree/SelectedConnectionDeletionConfirmer.cs index aeba09090..2f76679fe 100644 --- a/mRemoteV1/Tree/SelectedConnectionDeletionConfirmer.cs +++ b/mRemoteV1/Tree/SelectedConnectionDeletionConfirmer.cs @@ -1,17 +1,16 @@ -using System; -using System.Windows.Forms; -using mRemoteNG.Connection; +using mRemoteNG.Connection; using mRemoteNG.Container; -using mRemoteNG.UI.TaskDialog; +using System; +using System.Windows.Forms; namespace mRemoteNG.Tree { - public class SelectedConnectionDeletionConfirmer : IConfirm + public class SelectedConnectionDeletionConfirmer : IConfirm { - private readonly Func _confirmationFunc; + private readonly Func _confirmationFunc; - public SelectedConnectionDeletionConfirmer(Func confirmationFunc) + public SelectedConnectionDeletionConfirmer(Func confirmationFunc) { _confirmationFunc = confirmationFunc; } @@ -49,7 +48,7 @@ namespace mRemoteNG.Tree private bool PromptUser(string promptMessage) { - var msgBoxResponse = CTaskDialog.MessageBox(Application.ProductName, promptMessage, "", ETaskDialogButtons.YesNo, ESysIcons.Question); + var msgBoxResponse = _confirmationFunc(promptMessage); return msgBoxResponse == DialogResult.Yes; } } diff --git a/mRemoteV1/UI/Window/ConnectionTreeWindow.cs b/mRemoteV1/UI/Window/ConnectionTreeWindow.cs index e5976d136..cc8b6be6b 100644 --- a/mRemoteV1/UI/Window/ConnectionTreeWindow.cs +++ b/mRemoteV1/UI/Window/ConnectionTreeWindow.cs @@ -5,6 +5,7 @@ using mRemoteNG.Themes; using mRemoteNG.Tree; using mRemoteNG.Tree.Root; using mRemoteNG.UI.Controls; +using mRemoteNG.UI.TaskDialog; using System; using System.Collections.Generic; using System.ComponentModel; @@ -104,7 +105,8 @@ namespace mRemoteNG.UI.Window #region ConnectionTree private void SetConnectionTreeEventHandlers() { - olvConnections.NodeDeletionConfirmer = new SelectedConnectionDeletionConfirmer(MessageBox.Show); + olvConnections.NodeDeletionConfirmer = new SelectedConnectionDeletionConfirmer(prompt => + CTaskDialog.MessageBox(Application.ProductName, prompt, "", ETaskDialogButtons.YesNo, ESysIcons.Question)); olvConnections.KeyDown += tvConnections_KeyDown; olvConnections.KeyPress += tvConnections_KeyPress; SetTreePostSetupActions();