1 files changed, 167 insertions, 0 deletions
diff --git a/tmf/org.eclipse.tracecompass.tmf.ui.swtbot.tests/src/org/eclipse/tracecompass/tmf/ui/swtbot/tests/viewers/events/SDViewTest.java b/tmf/org.eclipse.tracecompass.tmf.ui.swtbot.tests/src/org/eclipse/tracecompass/tmf/ui/swtbot/tests/viewers/events/SDViewTest.java
new file mode 100644
index 0000000000..6e8396858c
--- /dev/null
+++ b/tmf/org.eclipse.tracecompass.tmf.ui.swtbot.tests/src/org/eclipse/tracecompass/tmf/ui/swtbot/tests/viewers/events/SDViewTest.java
@@ -0,0 +1,167 @@
+/*******************************************************************************
+ * Copyright (c) 2015 Ericsson
+ *
+ * All rights reserved. This program and the accompanying materials are
+ * made available under the terms of the Eclipse Public License v1.0 which
+ * accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *   Matthew Khouzam - Initial API and implementation
+ *******************************************************************************/
+package org.eclipse.tracecompass.tmf.ui.swtbot.tests.viewers.events;
+
+import static org.junit.Assert.assertArrayEquals;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.apache.log4j.ConsoleAppender;
+import org.apache.log4j.Logger;
+import org.apache.log4j.SimpleLayout;
+import org.eclipse.core.runtime.Platform;
+import org.eclipse.swtbot.eclipse.finder.SWTWorkbenchBot;
+import org.eclipse.swtbot.eclipse.finder.widgets.SWTBotView;
+import org.eclipse.swtbot.swt.finder.utils.SWTBotPreferences;
+import org.eclipse.swtbot.swt.finder.widgets.SWTBotToolbarButton;
+import org.eclipse.swtbot.swt.finder.widgets.SWTBotTreeItem;
+import org.eclipse.tracecompass.tmf.core.io.BufferedRandomAccessFile;
+import org.eclipse.tracecompass.tmf.ui.swtbot.tests.shared.SWTBotUtils;
+import org.eclipse.tracecompass.tmf.ui.views.uml2sd.SDView;
+import org.eclipse.tracecompass.tmf.ui.views.uml2sd.core.Frame;
+import org.junit.After;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+/**
+ * Test for Sequence Diagram view in trace compass
+ */
+public class SDViewTest {
+
+    private static final String UML2DVIEW_ID = "org.eclipse.linuxtools.tmf.ui.tmfUml2SDSyncView";
+
+    private static final String XMLSTUB_ID = "org.eclipse.linuxtools.tmf.core.tests.xmlstub";
+
+    private static final String TRACE_START = "<trace>";
+    private static final String EVENT_BEGIN = "<event timestamp=\"";
+    private static final String EVENT_MIDDLE1 = " \" name=\"";
+    private static final String EVENT_MIDDLE2 = "\">";
+    private static final String FIELD_SENDER = "<field name=\"sender\" value=\"";
+    private static final String FIELD_RECEIVER = "<field name=\"receiver\" value=\"";
+    private static final String FIELD_SIGNAL = "<field name=\"signal\" value=\"";
+    private static final String FIELD_END = "\" type=\"string\" />";
+    private static final String EVENT_END = "</event>";
+    private static final String TRACE_END = "</trace>";
+
+    private static final String PROJECT_NAME = "TestForFiltering";
+
+    /** The Log4j logger instance. */
+    private static final Logger fLogger = Logger.getRootLogger();
+    private static SWTWorkbenchBot fBot;
+
+    private static String makeEvent(int ts, String eventName, String send, String recv, String signal) {
+        return EVENT_BEGIN + Integer.toString(ts) + EVENT_MIDDLE1 + eventName + EVENT_MIDDLE2 + FIELD_SENDER + send + FIELD_END + FIELD_RECEIVER + recv + FIELD_END + FIELD_SIGNAL + signal + FIELD_END + EVENT_END + "\n";
+    }
+
+    private static File fFileLocation;
+
+    /**
+     * Initialization, creates a temp trace
+     *
+     * @throws IOException
+     *             should not happen
+     */
+    @BeforeClass
+    public static void init() throws IOException {
+        SWTBotUtils.failIfUIThread();
+        Thread.currentThread().setName("SWTBot Thread"); // for the debugger
+        /* set up for swtbot */
+        SWTBotPreferences.TIMEOUT = 20000; /* 20 second timeout */
+        fLogger.removeAllAppenders();
+        fLogger.addAppender(new ConsoleAppender(new SimpleLayout()));
+        fBot = new SWTWorkbenchBot();
+
+        SWTBotUtils.closeView("welcome", fBot);
+
+        SWTBotUtils.switchToTracingPerspective();
+        /* finish waiting for eclipse to load */
+        SWTBotUtils.waitForJobs();
+        fFileLocation = File.createTempFile("sample", ".xml");
+        String eventNames[] = { "test:SEND", "test:RECEIVE" };
+        String targets[] = { "peer1", "peer2" };
+
+        try (BufferedRandomAccessFile braf = new BufferedRandomAccessFile(fFileLocation, "rw")) {
+            braf.writeBytes(TRACE_START);
+            for (int i = 0; i < 100; i++) {
+                braf.writeBytes(makeEvent(i * i * 100, eventNames[i % 2], targets[i % 2], targets[(i + 1) % 2], Integer.toString(i % 2 + 1000)));
+            }
+            braf.writeBytes(TRACE_END);
+        }
+    }
+
+    /**
+     * Open a trace in an editor
+     */
+    @Before
+    public void beforeTest() {
+        SWTBotUtils.createProject(PROJECT_NAME);
+        SWTBotTreeItem treeItem = SWTBotUtils.selectTracesFolder(fBot, PROJECT_NAME);
+        assertNotNull(treeItem);
+        SWTBotUtils.openTrace(PROJECT_NAME, fFileLocation.getAbsolutePath(), XMLSTUB_ID);
+        SWTBotUtils.openView(UML2DVIEW_ID);
+    }
+
+    /**
+     * Delete the file
+     */
+    @AfterClass
+    public static void cleanUp() {
+        fFileLocation.delete();
+        fLogger.removeAllAppenders();
+    }
+
+    /**
+     * Close the editor
+     */
+    @After
+    public void tearDown() {
+        fBot.closeAllEditors();
+        SWTBotUtils.deleteProject(PROJECT_NAME, fBot);
+    }
+
+    /**
+     * Test Sequence diagram view, counting the columns
+     */
+    @Test
+    public void testSDView() {
+        SWTBotView viewBot = fBot.viewById(UML2DVIEW_ID);
+
+        assertNotNull(viewBot);
+        viewBot.setFocus();
+        SWTBotUtils.waitForJobs();
+        List<SWTBotToolbarButton> viewButtons = viewBot.getToolbarButtons();
+        List<String> titles = new ArrayList<>();
+        for (SWTBotToolbarButton buttonBot : viewButtons) {
+            titles.add(buttonBot.getToolTipText());
+        }
+
+        final char commandKeyChar = (char) 0x2318;
+        final String findShortcut = (Platform.getOS().equals(Platform.OS_MACOSX) ? commandKeyChar : "Ctrl+") + "F";
+        String[] expected = { "Reset zoom factor", "Select",
+                "Zoom in the diagram", "Zoom out the diagram",
+                "Go to next page", "Go to previous page",
+                "Go to first page", "Go to last page",
+                "Find... (" + findShortcut + ")"
+        };
+        assertArrayEquals("Buttons", expected, titles.toArray(new String[0]));
+        SDView view = (SDView) viewBot.getViewReference().getPart(false);
+        Frame frame = view.getFrame();
+        assertEquals(2, frame.lifeLinesCount());
+    }
+}