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.indexer.ITmfTraceIndexer;
  28
  29 /**
  30  * Test suite for indexing using a CustomTxtTrace.
  31  *
  32  * @author Marc-Andre Laperle
  33  */
  34 public class CustomTxtIndexTest extends AbstractCustomTraceIndexTest {
  35
  36     private static final String TRACE_DIRECTORY = System.getProperty("java.io.tmpdir") + File.separator + "dummyTxtTrace";
  37     private static final String TRACE_PATH = TRACE_DIRECTORY + File.separator + "test.txt";
  38     private static final String DEFINITION_PATH = "tracesets" + File.separator + "txt" + File.separator + "testTxtDefinition.xml";
  39
  40     private static CustomTxtTraceDefinition createDefinition() {
  41         CustomTxtTraceDefinition[] definitions = CustomTxtTraceDefinition.loadAll(new File(DEFINITION_PATH).toString());
  42         return definitions[0];
  43     }
  44
  45     @Override
  46     protected String getTraceDirectory() {
  47         return TRACE_DIRECTORY;
  48     }
  49
  50     @Override
  51     protected TestTrace createTrace() throws Exception {
  52         CustomTxtTraceDefinition definition = createDefinition();
  53         final File file = new File(TRACE_PATH);
  54         try (BufferedWriter writer = new BufferedWriter(new FileWriter(file));) {
  55             for (int i = 0; i < NB_EVENTS; ++i) {
  56                 SimpleDateFormat f = new SimpleDateFormat(TIMESTAMP_FORMAT);
  57                 String eventStr = f.format(new Date(i)) + " hello world\n";
  58                 writer.write(eventStr);
  59             }
  60         }
  61
  62         return new TestTxtTrace(file.toString(), definition, BLOCK_SIZE);
  63     }
  64
  65     private class TestTxtTrace extends CustomTxtTrace implements TestTrace {
  66         public TestTxtTrace(String path, CustomTxtTraceDefinition createDefinition, int blockSize) throws TmfTraceException {
  67             super(null, createDefinition, path, blockSize);
  68         }
  69
  70         @Override
  71         protected ITmfTraceIndexer createIndexer(int interval) {
  72             return new TestIndexer(this, interval);
  73         }
  74
  75         @Override
  76         public TestIndexer getIndexer() {
  77             return (TestIndexer) super.getIndexer();
  78         }
  79     }
  80 }