harness/src/test/java/org/apache/geode/perftest/yardstick/analysis/YardstickThroughputSensorParserTest.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more contributor license
 * agreements. See the NOTICE file distributed with this work for additional information regarding
 * copyright ownership. The ASF licenses this file to You under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance with the License. You may obtain a
 * copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */

package org.apache.geode.perftest.yardstick.analysis;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.nio.file.Path;

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.io.TempDir;

public class YardstickThroughputSensorParserTest {

  @TempDir
  Path temporaryFolder;

  @Test
  public void parsesInputFile() throws IOException {
    final File testFolder = temporaryFolder.resolve("testFolder").toFile();
    assertTrue(testFolder.mkdir());
    final File testFile = new File(testFolder, YardstickThroughputSensorParser.sensorOutputFile);
    assertTrue(testFile.createNewFile());
    BufferedWriter output = new BufferedWriter(new FileWriter(testFile));
    output.write("1542151468,42906.00,371126.31");
    output.newLine();
    output.write("1542151469,45134.00,353812.31");
    output.newLine();
    output.write("1542151470,47091.00,339720.36");
    output.newLine();
    output.write("1542151471,47567.00,335120.03");
    output.newLine();
    output.close();

    YardstickThroughputSensorParser parser = new YardstickThroughputSensorParser();
    parser.parseResults(testFolder);
    assertEquals(45674.5, parser.getAverageThroughput(), .01);
  }

  @Test
  public void doesNotThrowExceptionOnMissingFile() throws IOException {
    final File testFolder = temporaryFolder.resolve("testFolder").toFile();
    assertTrue(testFolder.mkdirs());
    YardstickThroughputSensorParser parser = new YardstickThroughputSensorParser();
    parser.parseResults(testFolder);
    assertEquals(3, parser.getProbeResults().size());
  }

  @Test
  public void throwsExceptionOnBadInput() throws IOException {
    final File testFolder = temporaryFolder.resolve("testFolder").toFile();
    assertTrue(testFolder.mkdirs());
    final File testFile = new File(testFolder, YardstickThroughputSensorParser.sensorOutputFile);
    assertTrue(testFile.createNewFile());
    BufferedWriter output = new BufferedWriter(new FileWriter(testFile));
    output.write("1542151468,42906.00,371126.31");
    output.newLine();
    output.write("1542151469,45134.00,353812.31");
    output.newLine();
    output.write("This is not a proper comment");
    output.newLine();
    output.write("1542151471,47567.00,335120.03");
    output.newLine();
    output.close();

    YardstickThroughputSensorParser parser = new YardstickThroughputSensorParser();
    assertThrows(IOException.class, () -> parser.parseResults(testFolder));
  }

  @Test
  public void ignoreYardstickCommentAndMetadataLines() throws IOException {
    final File testFolder = temporaryFolder.resolve("testFolder").toFile();
    assertTrue(testFolder.mkdirs());
    final File testFile = new File(testFolder, YardstickThroughputSensorParser.sensorOutputFile);
    assertTrue(testFile.createNewFile());
    BufferedWriter output = new BufferedWriter(new FileWriter(testFile));
    output.write("--This is a comment");
    output.newLine();
    output.write("@@Classname");
    output.newLine();
    output.write("**Metadata");
    output.newLine();
    output.write("1542151468,42906.00,371126.31");
    output.newLine();
    output.close();

    YardstickThroughputSensorParser parser = new YardstickThroughputSensorParser();
    parser.parseResults(testFolder);
    assertEquals(42906f, parser.getAverageThroughput(), .01);
  }
}