Convert inputtextframe tests to uispec4j

Replace selected text when using paste dialog

Fixes issue 4162

Change-Id: I0d4997c8ced84efbc1ed31276b48133ffcce909a
Reviewed-on: http://gerrit.dmdirc.com/1261
Automatic-Compile: DMDirc Local Commits <dmdirc@googlemail.com>
Reviewed-by: Chris Smith <chris@dmdirc.com>

src/com/dmdirc/addons/ui_swing/components/SwingInputField.java

@@ -439,6 +439,36 @@ public class SwingInputField extends JComponent implements InputField,
         });
     }
 
+    /**
+     * Sets the start index of the selection for this component.
+     *
+     * @param selectionStart Start index
+     */
+    public void setSelectionStart(final int selectionStart) {
+        UIUtilities.invokeLater(new Runnable() {
+
+            @Override
+            public void run() {
+                textField.setSelectionStart(selectionStart);
+            }
+        });
+    }
+
+    /**
+     * Sets the end index of the selection for this component
+     *
+     * @param selectionEnd End index
+     */
+    public void setSelectionEnd(final int selectionEnd) {
+        UIUtilities.invokeLater(new Runnable() {
+
+            @Override
+            public void run() {
+                textField.setSelectionEnd(selectionEnd);
+            }
+        });
+    }
+
     /** 
      * {@inheritDoc}
      * 

src/com/dmdirc/addons/ui_swing/components/frames/InputTextFrame.java

@@ -364,6 +364,7 @@ public abstract class InputTextFrame extends TextFrame implements InputWindow,
         //check theres something to paste
         if (clipboard != null && (clipboardLines = getSplitLine(clipboard)).length
                 > 1) {
+            getInputField().replaceSelection("");
             final int caretPosition = getInputField().getCaretPosition();
             final String inputFieldText = getInputField().getText();
             final String text = inputFieldText.substring(0, caretPosition)

test/com/dmdirc/addons/ui_swing/ClassComponentMatcher.java

@@ -0,0 +1,61 @@
+/*
+ * Copyright (c) 2006-2010 Chris Smith, Shane Mc Cormack, Gregory Holmes
+ * 
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ * 
+ * The above copyright notice and this permission notice shall be included in
+ * all copies or substantial portions of the Software.
+ * 
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+package com.dmdirc.addons.ui_swing;
+
+import java.awt.Component;
+
+import org.uispec4j.Panel;
+import org.uispec4j.finder.ComponentMatcher;
+
+/**
+ * Simple UISpec4J component matcher that matches by type.
+ */
+public class ClassComponentMatcher implements ComponentMatcher {
+
+    private final Panel comp;
+    private final Class clazz;
+
+    /**
+     * Creates a new component matcher, searching for any components matching
+     * the specified class in the specified component.
+     *
+     * @param comp Component to search in
+     * @param clazz Class to search for
+     */
+    public ClassComponentMatcher(final Panel comp, final Class clazz) {
+        this.comp = comp;
+        this.clazz = clazz;
+    }
+
+    /**
+     * {@inheritDoc}
+     * 
+     * @param component Component to match against
+     * 
+     * @return true iif the component matches
+     */
+    @Override
+    public boolean matches(final Component component) {
+        return component    .getClass() == clazz;
+    }
+}

test/com/dmdirc/addons/ui_swing/components/frames/InputTextFrameTest.java

@@ -23,43 +23,39 @@
 package com.dmdirc.addons.ui_swing.components.frames;
 
 import com.dmdirc.Main;
+import com.dmdirc.WritableFrameContainer;
+import com.dmdirc.addons.ui_swing.ClassComponentMatcher;
+import com.dmdirc.addons.ui_swing.MainFrame;
 import com.dmdirc.config.ConfigManager;
 import com.dmdirc.config.IdentityManager;
 import com.dmdirc.config.InvalidIdentityFileException;
-import com.dmdirc.harness.TestConfigManagerMap;
-import com.dmdirc.harness.TestWritableFrameContainer;
-import com.dmdirc.harness.ui.ClassFinder;
-import com.dmdirc.ui.WindowManager;
 import com.dmdirc.addons.ui_swing.SwingController;
-import com.dmdirc.addons.ui_swing.UIUtilities;
+import com.dmdirc.addons.ui_swing.SwingWindowFactory;
 import com.dmdirc.addons.ui_swing.components.TextAreaInputField;
 import com.dmdirc.plugins.PluginManager;
-import com.dmdirc.ui.interfaces.InputWindow;
+import com.dmdirc.ui.messages.IRCDocument;
 
-import java.awt.event.KeyEvent;
 
-import org.fest.swing.core.KeyPressInfo;
-import org.fest.swing.core.matcher.DialogMatcher;
-import org.fest.swing.core.matcher.JButtonMatcher;
-import org.fest.swing.fixture.DialogFixture;
-import org.fest.swing.fixture.FrameFixture;
-import org.fest.swing.fixture.JInternalFrameFixture;
 import org.junit.After;
 import org.junit.Before;
-import org.junit.BeforeClass;
 import org.junit.Test;
-
-public class InputTextFrameTest {
-
-    static FrameFixture mainframe;
-    static JInternalFrameFixture window;
-    static TestConfigManagerMap cmmap;
-    static TestWritableFrameContainer<InputWindow> owner;
-    static SwingController controller;
-
-    @BeforeClass
-    public static void setUpClass() throws InvalidIdentityFileException {
-        IdentityManager.load();
+import org.uispec4j.Trigger;
+import org.uispec4j.UISpecTestCase;
+import org.uispec4j.Window;
+import org.uispec4j.interception.WindowInterceptor;
+import static org.mockito.Mockito.*;
+
+public class InputTextFrameTest extends UISpecTestCase {
+
+    static InputTextFrame tf;
+    static String text;
+
+    static {
+        try {
+            IdentityManager.load();
+        } catch (InvalidIdentityFileException ex) {
+            //Ignore
+        }
         IdentityManager.getAddonIdentity().setOption("test", "windowMenuItems",
                 "1");
         IdentityManager.getAddonIdentity().setOption("test",
@@ -69,142 +65,119 @@ public class InputTextFrameTest {
                 "textpanebackground", "");
         IdentityManager.getAddonIdentity().setOption("test", "desktopbackground",
                 "");
-        controller = new SwingController();
-        controller.setDomain("test");
-        controller.onLoad();
-
         Main.ensureExists(PluginManager.getPluginManager(), "tabcompletion");
 
-        UIUtilities.initUISettings();
-
-        mainframe = new FrameFixture(controller.getMainWindow());
-    }
-
-    @Before
-    public void setUp() {
-        cmmap = new TestConfigManagerMap();
-        cmmap.settings.put("ui.pasteProtectionLimit", "1");
-
-        owner = new TestWritableFrameContainer<InputWindow>(512, cmmap,
-                InputWindow.class);
-        setupWindow(cmmap);
-    }
-
-    @After
-    public void tearDown() {
-        UIUtilities.invokeAndWait(new Runnable() {
-
-            @Override
-            public void run() {
-                window.close();
-                controller.getWindowFactory().getSwingWindow(owner).close();
-                WindowManager.removeWindow(owner);
-            }
-        });
+        final IRCDocument document = mock(IRCDocument.class);
+        @SuppressWarnings("unchecked")
+        final WritableFrameContainer<CustomInputFrame> container = mock(
+                WritableFrameContainer.class);
+        final ConfigManager config = mock(ConfigManager.class);
+        when(container.getDocument()).thenReturn(document);
+        when(container.getConfigManager()).thenReturn(config);
+        when(config.getOption(anyString(), anyString())).thenReturn("mirc");
+        final SwingController controller = mock(SwingController.class);
+        when(controller.getDomain()).thenReturn("test");
+        final SwingWindowFactory wf = new SwingWindowFactory(controller);
+        when(controller.getWindowFactory()).thenReturn(wf);
+        final MainFrame mainFrame = new MainFrame(controller);
+        when(controller.getMainFrame()).thenReturn(mainFrame);
+
+        tf = new CustomInputFrame(controller, container);
+        text = "line1\nline2";
     }
 
     @Test
     public void testPasteDialogContents() throws InterruptedException {
-        UIUtilities.invokeAndWait(new Runnable() {
-
-            @Override
-            public void run() {
-                ((InputTextFrame) window.target).doPaste("line1\nline2");
-
-                final DialogFixture dlg = mainframe.dialog(DialogMatcher.
-                        withTitle(
-                        "DMDirc: Multi-line paste").andShowing());
-
-                dlg.requireVisible().button(JButtonMatcher.withText("Edit")).
-                        click();
-                dlg.textBox(new ClassFinder<TextAreaInputField>(
-                        TextAreaInputField.class,
-                        null)).requireText("line1\nline2");
-                dlg.target.dispose();
-            }
-        });
+        final Window dialog = getDialog();
+        dialog.titleEquals("DMDirc: Multi-line paste").check();
+        dialog.getButton("Edit").click();
+        assertTrue(dialog.getTextBox(new ClassComponentMatcher(dialog,
+                TextAreaInputField.class)).getText().equals(text));
     }
 
     @Test
     public void testPasteDialogWithTextBefore() throws InterruptedException {
-        UIUtilities.invokeAndWait(new Runnable() {
-
-            @Override
-            public void run() {
-                window.textBox().enterText("testing:");
-                ((InputTextFrame) window.target).doPaste("line1\nline2");
-
-                final DialogFixture dlg = mainframe.dialog(DialogMatcher.
-                        withTitle("DMDirc: Multi-line paste").andShowing());
-
-                dlg.requireVisible().button(JButtonMatcher.withText("Edit")).
-                        click();
-                dlg.textBox(new ClassFinder<TextAreaInputField>(
-                        TextAreaInputField.class, null)).requireText(
-                        "testing:line1\nline2");
-                dlg.target.dispose();
-            }
-        });
+        tf.getInputField().setText("testing:");
+        final Window dialog = getDialog();
+        dialog.titleEquals("DMDirc: Multi-line paste").check();
+        dialog.getButton("Edit").click();
+        assertTrue(dialog.getTextBox(new ClassComponentMatcher(dialog,
+                TextAreaInputField.class)).getText().equals("testing:" + text));
     }
 
     @Test
     public void testPasteDialogWithTextAfter() throws InterruptedException {
-        UIUtilities.invokeAndWait(new Runnable() {
-
-            @Override
-            public void run() {
-                window.textBox().enterText("<- testing").pressAndReleaseKey(
-                        KeyPressInfo.keyCode(KeyEvent.VK_HOME));
-                ((InputTextFrame) window.target).doPaste("line1\nline2");
-
-                final DialogFixture dlg = mainframe.dialog(DialogMatcher.
-                        withTitle(
-                        "DMDirc: Multi-line paste").andShowing());
-
-                dlg.requireVisible().button(JButtonMatcher.withText("Edit")).
-                        click();
-                dlg.textBox(new ClassFinder<TextAreaInputField>(
-                        TextAreaInputField.class,
-                        null)).requireText("line1\nline2<- testing");
-                dlg.target.dispose();
-            }
-        });
+        tf.getInputField().setText(":testing");
+        tf.getInputField().setCaretPosition(0);
+        final Window dialog = getDialog();
+        dialog.titleEquals("DMDirc: Multi-line paste").check();
+        dialog.getButton("Edit").click();
+        assertTrue(dialog.getTextBox(new ClassComponentMatcher(dialog,
+                TextAreaInputField.class)).getText().equals(text + ":testing"));
     }
 
     @Test
     public void testPasteDialogWithTextAround() throws InterruptedException {
-        UIUtilities.invokeAndWait(new Runnable() {
-
-            @Override
-            public void run() {
-                window.textBox().enterText("testing:<- testing").selectText(8, 8);
-                ((InputTextFrame) window.target).doPaste("line1\nline2");
-
-                final DialogFixture dlg = mainframe.dialog(DialogMatcher.
-                        withTitle(
-                        "DMDirc: Multi-line paste").andShowing());
-
-                dlg.requireVisible().button(JButtonMatcher.withText("Edit")).
-                        click();
-                dlg.textBox(new ClassFinder<TextAreaInputField>(
-                        TextAreaInputField.class,
-                        null)).requireText("testing:line1\nline2<- testing");
-                dlg.target.dispose();
-            }
-        });
+        tf.getInputField().setText("testing::testing");
+        tf.getInputField().setCaretPosition(8);
+        final Window dialog = getDialog();
+        dialog.titleEquals("DMDirc: Multi-line paste").check();
+        dialog.getButton("Edit").click();
+        assertEquals("testing:" + text + ":testing",
+                dialog.getTextBox(new ClassComponentMatcher(dialog,
+                TextAreaInputField.class)).getText());
     }
 
-    protected void setupWindow(final ConfigManager configManager) {
-        final CustomInputFrame titf = new CustomInputFrame(controller, owner);
-
-        owner.setTitle("testing123");
-        owner.addWindow(titf);
-
-        WindowManager.addWindow(owner);
+    @Test
+    public void testPasteDialogWithSelection() {
+        tf.getInputField().setText("testing:SELECTED:testing");
+        tf.getInputField().setSelectionStart(8);
+        tf.getInputField().setSelectionEnd(16);
+        final Window dialog = getDialog();
+        dialog.titleEquals("DMDirc: Multi-line paste").check();
+        dialog.getButton("Edit").click();
+        assertEquals("testing:" + text + ":testing",
+                dialog.getTextBox(new ClassComponentMatcher(dialog,
+                TextAreaInputField.class)).getText());
+    }
 
-        titf.open();
+    /**
+     * Creates a new paste dialog with the specified text for the specified
+     * frame.
+     *
+     * @param frame Parent frame
+     * @param text Text to "paste"
+     *
+     * @return Wrapped Dialog
+     */
+    private Window getDialog() {
+        return WindowInterceptor.run(new PasteDialogTrigger(tf, text));
+    }
 
-        window = new JInternalFrameFixture(mainframe.robot, titf);
-        window.robot.settings().eventPostingDelay(250);
+    /**
+     * Creates a new paste dialog for the specified frame with the specified text.
+     */
+    private class PasteDialogTrigger implements Trigger {
+
+        private InputTextFrame frame;
+        private String text;
+
+        /**
+         * Creates a new paste dialog for the specified frame with the specified text.
+         *
+         * @param frame Parent frame
+         * @param text Text to "paste"
+         */
+        public PasteDialogTrigger(final InputTextFrame frame, final String text) {
+            this.frame = frame;
+            this.text = text;
+        }
+
+        /** {@inheritDoc} */
+        @Override
+        public void run() throws Exception {
+            frame.doPaste(text);
+        }
     }
+
 }