tmf/org.eclipse.tracecompass.tmf.ui.swtbot.tests/src/org/eclipse/tracecompass/tmf/ui/swtbot/tests/wizards/TestImportExportPackageWizard.java

   1 /*******************************************************************************
   2  * Copyright (c) 2014, 2015 Ericsson
   3  *
   4  * All rights reserved. This program and the accompanying materials are
   5  * made available under the terms of the Eclipse Public License v1.0 which
   6  * accompanies this distribution, and is available at
   7  * http://www.eclipse.org/legal/epl-v10.html
   8  *
   9  * Contributors:
  10  *   Matthew Khouzam - Initial API and implementation
  11  *******************************************************************************/
  12
  13 package org.eclipse.tracecompass.tmf.ui.swtbot.tests.wizards;
  14
  15 import static org.junit.Assert.assertEquals;
  16 import static org.junit.Assert.assertFalse;
  17 import static org.junit.Assert.assertNotNull;
  18 import static org.junit.Assert.assertTrue;
  19
  20 import java.io.File;
  21 import java.io.FileWriter;
  22 import java.io.IOException;
  23
  24 import org.apache.log4j.ConsoleAppender;
  25 import org.apache.log4j.Logger;
  26 import org.apache.log4j.SimpleLayout;
  27 import org.eclipse.core.runtime.Path;
  28 import org.eclipse.swtbot.eclipse.finder.SWTWorkbenchBot;
  29 import org.eclipse.swtbot.eclipse.finder.widgets.SWTBotView;
  30 import org.eclipse.swtbot.swt.finder.SWTBot;
  31 import org.eclipse.swtbot.swt.finder.junit.SWTBotJunit4ClassRunner;
  32 import org.eclipse.swtbot.swt.finder.utils.SWTBotPreferences;
  33 import org.eclipse.swtbot.swt.finder.waits.Conditions;
  34 import org.eclipse.swtbot.swt.finder.widgets.SWTBotShell;
  35 import org.eclipse.swtbot.swt.finder.widgets.SWTBotTreeItem;
  36 import org.eclipse.tracecompass.tmf.core.trace.TmfTraceManager;
  37 import org.eclipse.tracecompass.tmf.ui.swtbot.tests.shared.SWTBotUtils;
  38 import org.junit.BeforeClass;
  39 import org.junit.Test;
  40 import org.junit.runner.RunWith;
  41
  42 /**
  43  * Export and Import wizard tests
  44  *
  45  * @author Matthew Khouzam
  46  *
  47  */
  48 @RunWith(SWTBotJunit4ClassRunner.class)
  49 public class TestImportExportPackageWizard {
  50
  51     // private static final int PACKAGE_SIZE = 213732;
  52     private static final String EXPORT_LOCATION = TmfTraceManager.getTemporaryDirPath() + File.separator + "test.zip";
  53     private static final String IMPORT_TRACE_PACKAGE = "Import Trace Package...";
  54     private static final String IMPORT_TRACE_PACKAGE_TITLE = "Import trace package";
  55     private static final String EXPORT_TRACE_PACKAGE = "Export Trace Package...";
  56     private static final String EXPORT_TRACE_PACKAGE_TITLE = "Export trace package";
  57     private static final String PROJECT_EXPLORER = "Project Explorer";
  58     private static final String FINISH = "Finish";
  59     private static final String COMPRESS_THE_CONTENTS_OF_THE_FILE = "Compress the contents of the file";
  60     private static final String SAVE_IN_ZIP_FORMAT = "Save in zip format";
  61     private static final String SAVE_IN_TAR_FORMAT = "Save in tar format";
  62     private static final String SELECT_ALL = "Select All";
  63     private static final String DESELECT_ALL = "Deselect All";
  64     private static final String WELCOME_NAME = "welcome";
  65     private static final String SWT_BOT_THREAD_NAME = "SWTBot Thread";
  66     private static final String PROJECT_NAME = "Test";
  67     private static final String XMLSTUB_ID = "org.eclipse.linuxtools.tmf.core.tests.xmlstub";
  68
  69     /*
  70      * FIXME: bug https://bugs.eclipse.org/bugs/show_bug.cgi?id=477508, we
  71      * shouldn't need so many events.
  72      */
  73     private static final String TRACE_CONTENT = "<trace>" +
  74             "<event timestamp=\"100\" name=\"event\"><field name=\"field\" value=\"1\" type=\"int\" /></event>" +
  75             "<event timestamp=\"200\" name=\"event1\"><field name=\"field\" value=\"2\" type=\"int\" /></event>" +
  76             "<event timestamp=\"201\" name=\"event\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  77             "<event timestamp=\"202\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  78             "<event timestamp=\"203\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  79             "<event timestamp=\"300\" name=\"event1\"><field name=\"field\" value=\"2\" type=\"int\" /></event>" +
  80             "<event timestamp=\"301\" name=\"event\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  81             "<event timestamp=\"302\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  82             "<event timestamp=\"333\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  83             "<event timestamp=\"500\" name=\"event1\"><field name=\"field\" value=\"2\" type=\"int\" /></event>" +
  84             "<event timestamp=\"501\" name=\"event\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  85             "<event timestamp=\"502\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  86             "<event timestamp=\"533\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  87             "</trace>";
  88
  89     /** The Log4j logger instance. */
  90     private static final Logger fLogger = Logger.getRootLogger();
  91     private static SWTWorkbenchBot fBot;
  92
  93     /** Test Class setup */
  94     @BeforeClass
  95     public static void init() {
  96         SWTBotPreferences.KEYBOARD_LAYOUT = "EN_US";
  97         SWTBotUtils.initialize();
  98         Thread.currentThread().setName(SWT_BOT_THREAD_NAME); // for the debugger
  99         /* set up for swtbot */
 100         SWTBotPreferences.TIMEOUT = 20000; /* 20 second timeout */
 101         fLogger.addAppender(new ConsoleAppender(new SimpleLayout()));
 102         fBot = new SWTWorkbenchBot();
 103
 104         SWTBotUtils.closeView(WELCOME_NAME, fBot);
 105
 106         SWTBotUtils.switchToTracingPerspective();
 107         /* finish waiting for eclipse to load */
 108         SWTBotUtils.waitForJobs();
 109
 110     }
 111
 112     /**
 113      * test opening a trace, importing
 114      *
 115      * @throws IOException
 116      *             won't happen
 117      */
 118     @Test
 119     public void test() throws IOException {
 120         File f = File.createTempFile("temp", ".xml").getCanonicalFile();
 121         try (FileWriter fw = new FileWriter(f)) {
 122             fw.write(TRACE_CONTENT);
 123         }
 124         File exportPackage = new File(EXPORT_LOCATION);
 125         if (exportPackage.exists()) {
 126             exportPackage.delete();
 127         }
 128         assertFalse("File: " + EXPORT_LOCATION + " already present, aborting test", exportPackage.exists());
 129         assertTrue("Trace :" + f.getAbsolutePath() + " does not exist, aborting test", f.exists());
 130         SWTBotUtils.createProject(PROJECT_NAME);
 131         SWTBotUtils.openTrace(PROJECT_NAME, f.getAbsolutePath(), XMLSTUB_ID);
 132         SWTBotUtils.waitForJobs();
 133         assertEquals("Incorrect opened trace!", f.getAbsolutePath(), (new File(TmfTraceManager.getInstance().getActiveTrace().getPath())).getAbsolutePath());
 134         SWTBotView projectExplorerBot = fBot.viewByTitle(PROJECT_EXPLORER);
 135         assertNotNull("Cannot find " + PROJECT_EXPLORER, projectExplorerBot);
 136         projectExplorerBot.show();
 137         SWTBotTreeItem treeItem = SWTBotUtils.selectTracesFolder(fBot, PROJECT_NAME);
 138
 139         treeItem.contextMenu(EXPORT_TRACE_PACKAGE).click();
 140         fBot.waitUntil(Conditions.shellIsActive(EXPORT_TRACE_PACKAGE_TITLE));
 141         SWTBot shellBot = fBot.activeShell().bot();
 142         shellBot.button(DESELECT_ALL).click();
 143         SWTBotTreeItem[] items = fBot.tree().getAllItems();
 144         for (SWTBotTreeItem item : items) {
 145             assertEquals(item.isChecked(), false);
 146         }
 147         shellBot.button(SELECT_ALL).click();
 148         for (SWTBotTreeItem item : items) {
 149             assertEquals(item.isChecked(), true);
 150         }
 151         shellBot.radio(SAVE_IN_TAR_FORMAT).click();
 152         shellBot.radio(SAVE_IN_ZIP_FORMAT).click();
 153
 154         shellBot.checkBox(COMPRESS_THE_CONTENTS_OF_THE_FILE).click();
 155         shellBot.checkBox(COMPRESS_THE_CONTENTS_OF_THE_FILE).click();
 156         shellBot.comboBox().setText(EXPORT_LOCATION);
 157         SWTBotShell shell = fBot.activeShell();
 158         shellBot.button(FINISH).click();
 159         // finished exporting
 160         SWTBotUtils.waitForJobs();
 161         fBot.waitUntil(Conditions.shellCloses(shell));
 162         fBot = new SWTWorkbenchBot();
 163         exportPackage = new File(EXPORT_LOCATION);
 164         assertTrue("Exported package", exportPackage.exists());
 165         // Fixme: determine why exportPackageSize is different on different machines
 166         // assertEquals("Exported package size check", PACKAGE_SIZE, exportPackage.length());
 167
 168         // import
 169         treeItem = SWTBotUtils.selectTracesFolder(fBot, PROJECT_NAME);
 170         treeItem.contextMenu(IMPORT_TRACE_PACKAGE).click();
 171         fBot.waitUntil(Conditions.shellIsActive(IMPORT_TRACE_PACKAGE_TITLE));
 172         shellBot = fBot.activeShell().bot();
 173         shellBot.comboBox().setText(EXPORT_LOCATION);
 174         shellBot.comboBox().typeText("\n");
 175
 176         shellBot.button(SELECT_ALL).click();
 177         shell = fBot.activeShell();
 178         shellBot.button(FINISH).click();
 179         fBot.button("Yes To All").click();
 180         fBot.waitUntil(Conditions.shellCloses(shell));
 181         fBot = new SWTWorkbenchBot();
 182         SWTBotUtils.openEditor(fBot, PROJECT_NAME, new Path(f.getName()));
 183         assertEquals("Test if import matches", f.getName(), TmfTraceManager.getInstance().getActiveTrace().getName());
 184         assertFalse("Test if import files don't match", f.getAbsolutePath().equals(TmfTraceManager.getInstance().getActiveTrace().getPath()));
 185         SWTBotUtils.deleteProject(PROJECT_NAME, fBot);
 186         SWTBotUtils.waitForJobs();
 187     }
 188
 189 }