org.eclipse.linuxtools.tmf.ui.tests/src/org/eclipse/linuxtools/tmf/ui/tests/trace/CustomTxtIndexTest.java

   1 /*******************************************************************************
   2  * Copyright (c) 2009, 2014 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  *   Francois Chouinard - Initial API and implementation
  11  *   Francois Chouinard - Adapted for TMF Trace Model 1.0
  12  *   Alexandre Montplaisir - Port to JUnit4
  13  *   Marc-Andre Laperle - Adapted to CustomTxtTrace
  14  *******************************************************************************/
  15
  16 package org.eclipse.linuxtools.tmf.ui.tests.trace;
  17
  18 import java.io.BufferedWriter;
  19 import java.io.File;
  20 import java.io.FileWriter;
  21 import java.text.SimpleDateFormat;
  22 import java.util.Date;
  23
  24 import org.eclipse.linuxtools.tmf.core.exceptions.TmfTraceException;
  25 import org.eclipse.linuxtools.tmf.core.parsers.custom.CustomTxtTrace;
  26 import org.eclipse.linuxtools.tmf.core.parsers.custom.CustomTxtTraceDefinition;
  27 import org.eclipse.linuxtools.tmf.core.trace.TmfTraceManager;
  28 import org.eclipse.linuxtools.tmf.core.trace.indexer.ITmfTraceIndexer;
  29
  30 /**
  31  * Test suite for indexing using a CustomTxtTrace.
  32  *
  33  * @author Marc-Andre Laperle
  34  */
  35 public class CustomTxtIndexTest extends AbstractCustomTraceIndexTest {
  36
  37     private static final String TRACE_DIRECTORY = TmfTraceManager.getTemporaryDirPath() + File.separator + "dummyTxtTrace";
  38     private static final String TRACE_PATH = TRACE_DIRECTORY + File.separator + "test.txt";
  39     private static final String DEFINITION_PATH = "tracesets" + File.separator + "txt" + File.separator + "testTxtDefinition.xml";
  40
  41     private static CustomTxtTraceDefinition createDefinition() {
  42         CustomTxtTraceDefinition[] definitions = CustomTxtTraceDefinition.loadAll(new File(DEFINITION_PATH).toString());
  43         return definitions[0];
  44     }
  45
  46     @Override
  47     protected String getTraceDirectory() {
  48         return TRACE_DIRECTORY;
  49     }
  50
  51     @Override
  52     protected TestTrace createTrace() throws Exception {
  53         CustomTxtTraceDefinition definition = createDefinition();
  54         final File file = new File(TRACE_PATH);
  55         try (BufferedWriter writer = new BufferedWriter(new FileWriter(file));) {
  56             for (int i = 0; i < NB_EVENTS; ++i) {
  57                 SimpleDateFormat f = new SimpleDateFormat(TIMESTAMP_FORMAT);
  58                 String eventStr = f.format(new Date(i)) + " hello world\n";
  59                 writer.write(eventStr);
  60             }
  61         }
  62
  63         return new TestTxtTrace(file.toString(), definition, BLOCK_SIZE);
  64     }
  65
  66     private class TestTxtTrace extends CustomTxtTrace implements TestTrace {
  67         public TestTxtTrace(String path, CustomTxtTraceDefinition createDefinition, int blockSize) throws TmfTraceException {
  68             super(null, createDefinition, path, blockSize);
  69         }
  70
  71         @Override
  72         protected ITmfTraceIndexer createIndexer(int interval) {
  73             return new TestIndexer(this, interval);
  74         }
  75
  76         @Override
  77         public TestIndexer getIndexer() {
  78             return (TestIndexer) super.getIndexer();
  79         }
  80     }
  81 }