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.eclipse.tracecompass.tmf.ui.tests.shared.WaitUtils;
  39 import org.junit.BeforeClass;
  40 import org.junit.Test;
  41 import org.junit.runner.RunWith;
  42
  43 /**
  44  * Export and Import wizard tests
  45  *
  46  * @author Matthew Khouzam
  47  *
  48  */
  49 @RunWith(SWTBotJunit4ClassRunner.class)
  50 public class TestImportExportPackageWizard {
  51
  52     // private static final int PACKAGE_SIZE = 213732;
  53     private static final String EXPORT_LOCATION = TmfTraceManager.getTemporaryDirPath() + File.separator + "test.zip";
  54     private static final String IMPORT_TRACE_PACKAGE = "Import Trace Package...";
  55     private static final String IMPORT_TRACE_PACKAGE_TITLE = "Import trace package";
  56     private static final String EXPORT_TRACE_PACKAGE = "Export Trace Package...";
  57     private static final String EXPORT_TRACE_PACKAGE_TITLE = "Export trace package";
  58     private static final String PROJECT_EXPLORER = "Project Explorer";
  59     private static final String FINISH = "Finish";
  60     private static final String COMPRESS_THE_CONTENTS_OF_THE_FILE = "Compress the contents of the file";
  61     private static final String SAVE_IN_ZIP_FORMAT = "Save in zip format";
  62     private static final String SAVE_IN_TAR_FORMAT = "Save in tar format";
  63     private static final String SELECT_ALL = "Select All";
  64     private static final String DESELECT_ALL = "Deselect All";
  65     private static final String WELCOME_NAME = "welcome";
  66     private static final String SWT_BOT_THREAD_NAME = "SWTBot Thread";
  67     private static final String PROJECT_NAME = "Test";
  68     private static final String XMLSTUB_ID = "org.eclipse.linuxtools.tmf.core.tests.xmlstub";
  69
  70     private static final String TRACE_CONTENT = "<trace>" +
  71             "<event timestamp=\"100\" name=\"event\"><field name=\"field\" value=\"1\" type=\"int\" /></event>" +
  72             "<event timestamp=\"200\" name=\"event1\"><field name=\"field\" value=\"2\" type=\"int\" /></event>" +
  73             "<event timestamp=\"201\" name=\"event\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  74             "<event timestamp=\"202\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  75             "<event timestamp=\"203\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  76             "<event timestamp=\"300\" name=\"event1\"><field name=\"field\" value=\"2\" type=\"int\" /></event>" +
  77             "<event timestamp=\"301\" name=\"event\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  78             "<event timestamp=\"302\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  79             "<event timestamp=\"333\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  80             "<event timestamp=\"500\" name=\"event1\"><field name=\"field\" value=\"2\" type=\"int\" /></event>" +
  81             "<event timestamp=\"501\" name=\"event\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  82             "<event timestamp=\"502\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  83             "<event timestamp=\"533\" name=\"event1\"><field name=\"field\" value=\"3\" type=\"int\" /></event>" +
  84             "</trace>";
  85
  86     /** The Log4j logger instance. */
  87     private static final Logger fLogger = Logger.getRootLogger();
  88     private static SWTWorkbenchBot fBot;
  89
  90     /** Test Class setup */
  91     @BeforeClass
  92     public static void init() {
  93         SWTBotPreferences.KEYBOARD_LAYOUT = "EN_US";
  94         SWTBotUtils.initialize();
  95         Thread.currentThread().setName(SWT_BOT_THREAD_NAME); // for the debugger
  96         /* set up for swtbot */
  97         SWTBotPreferences.TIMEOUT = 20000; /* 20 second timeout */
  98         fLogger.addAppender(new ConsoleAppender(new SimpleLayout()));
  99         fBot = new SWTWorkbenchBot();
 100
 101         SWTBotUtils.closeView(WELCOME_NAME, fBot);
 102
 103         SWTBotUtils.switchToTracingPerspective();
 104         /* finish waiting for eclipse to load */
 105         WaitUtils.waitForJobs();
 106
 107     }
 108
 109     /**
 110      * test opening a trace, importing
 111      *
 112      * @throws IOException
 113      *             won't happen
 114      */
 115     @Test
 116     public void test() throws IOException {
 117         File f = File.createTempFile("temp", ".xml").getCanonicalFile();
 118         try (FileWriter fw = new FileWriter(f)) {
 119             fw.write(TRACE_CONTENT);
 120         }
 121         File exportPackage = new File(EXPORT_LOCATION);
 122         if (exportPackage.exists()) {
 123             exportPackage.delete();
 124         }
 125         assertFalse("File: " + EXPORT_LOCATION + " already present, aborting test", exportPackage.exists());
 126         assertTrue("Trace :" + f.getAbsolutePath() + " does not exist, aborting test", f.exists());
 127         SWTBotUtils.createProject(PROJECT_NAME);
 128         SWTBotUtils.openTrace(PROJECT_NAME, f.getAbsolutePath(), XMLSTUB_ID);
 129         WaitUtils.waitForJobs();
 130         assertEquals("Incorrect opened trace!", f.getAbsolutePath(), (new File(TmfTraceManager.getInstance().getActiveTrace().getPath())).getAbsolutePath());
 131         SWTBotView projectExplorerBot = fBot.viewByTitle(PROJECT_EXPLORER);
 132         assertNotNull("Cannot find " + PROJECT_EXPLORER, projectExplorerBot);
 133         projectExplorerBot.show();
 134         SWTBotTreeItem treeItem = SWTBotUtils.selectTracesFolder(fBot, PROJECT_NAME);
 135
 136         treeItem.contextMenu(EXPORT_TRACE_PACKAGE).click();
 137         fBot.waitUntil(Conditions.shellIsActive(EXPORT_TRACE_PACKAGE_TITLE));
 138         SWTBot shellBot = fBot.activeShell().bot();
 139         shellBot.button(DESELECT_ALL).click();
 140         SWTBotTreeItem[] items = fBot.tree().getAllItems();
 141         for (SWTBotTreeItem item : items) {
 142             assertEquals(item.isChecked(), false);
 143         }
 144         shellBot.button(SELECT_ALL).click();
 145         for (SWTBotTreeItem item : items) {
 146             assertEquals(item.isChecked(), true);
 147         }
 148         shellBot.radio(SAVE_IN_TAR_FORMAT).click();
 149         shellBot.radio(SAVE_IN_ZIP_FORMAT).click();
 150
 151         shellBot.checkBox(COMPRESS_THE_CONTENTS_OF_THE_FILE).click();
 152         shellBot.checkBox(COMPRESS_THE_CONTENTS_OF_THE_FILE).click();
 153         shellBot.comboBox().setText(EXPORT_LOCATION);
 154         SWTBotShell shell = fBot.activeShell();
 155         shellBot.button(FINISH).click();
 156         // finished exporting
 157         WaitUtils.waitForJobs();
 158         fBot.waitUntil(Conditions.shellCloses(shell));
 159         fBot = new SWTWorkbenchBot();
 160         exportPackage = new File(EXPORT_LOCATION);
 161         assertTrue("Exported package", exportPackage.exists());
 162         // Fixme: determine why exportPackageSize is different on different machines
 163         // assertEquals("Exported package size check", PACKAGE_SIZE, exportPackage.length());
 164
 165         // import
 166         treeItem = SWTBotUtils.selectTracesFolder(fBot, PROJECT_NAME);
 167         treeItem.contextMenu(IMPORT_TRACE_PACKAGE).click();
 168         fBot.waitUntil(Conditions.shellIsActive(IMPORT_TRACE_PACKAGE_TITLE));
 169         shellBot = fBot.activeShell().bot();
 170         shellBot.comboBox().setText(EXPORT_LOCATION);
 171         shellBot.comboBox().typeText("\n");
 172
 173         shellBot.button(SELECT_ALL).click();
 174         shell = fBot.activeShell();
 175         shellBot.button(FINISH).click();
 176         fBot.button("Yes To All").click();
 177         fBot.waitUntil(Conditions.shellCloses(shell));
 178         fBot = new SWTWorkbenchBot();
 179         SWTBotUtils.openEditor(fBot, PROJECT_NAME, new Path(f.getName()));
 180         assertEquals("Test if import matches", f.getName(), TmfTraceManager.getInstance().getActiveTrace().getName());
 181         assertFalse("Test if import files don't match", f.getAbsolutePath().equals(TmfTraceManager.getInstance().getActiveTrace().getPath()));
 182         SWTBotUtils.deleteProject(PROJECT_NAME, fBot);
 183         WaitUtils.waitForJobs();
 184     }
 185
 186 }