TestUtils.java

package com.formulasearchengine.mathosphere;

import java.io.*;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Scanner;

public class TestUtils {
  private static Connection cn = null;

  public static String getTestQueryString() throws IOException {
    return getFileContents("de/tuberlin/dima/schubotz/fse/fQuery.xml");
  }

  public static String getTestFile10String() throws IOException {
    return getFileContents("de/tuberlin/dima/schubotz/fse/test10.xml");
  }

  public static String getTestFile10StringPath() {
    return "de/tuberlin/dima/schubotz/fse/test10.xml";
  }

  public static String getFQueryStringPath() {
    return "de/tuberlin/dima/schubotz/fse/fQuery.xml";
  }

  public static String getTestResultForTest11() throws IOException {
    return getFileContents("expectedMatch.xml");
  }
    /*
    public static String getTestFile1() throws IOException {
        String[] split = getTestFile10().split(MainProgram.DOCUMENT_SEPARATOR,2);
        return split[0];
    }
    */


  public static String getFileContents(String fName) throws IOException {
    final InputStream is = getClassLoader().getResourceAsStream(fName);
    try {
      final Scanner s = new Scanner(is, "UTF-8");
      //Stupid scanner tricks to read the entire file as one token
      s.useDelimiter("\\A");
      return s.hasNext() ? s.next() : "";
    } finally {
      is.close();
    }
  }


  public static int countLines(String filename) throws IOException {
    InputStream is = new BufferedInputStream(new FileInputStream(filename));
    try {
      byte[] c = new byte[1024];
      int count = 0;
      int readChars = 0;
      boolean readingLine = false;
      while ((readChars = is.read(c)) != -1) {
        readingLine = true;
        for (int i = 0; i < readChars; i++) {
          if (c[i] == '\n') {
            readingLine = false;
            count++;
          }
        }
      }
      return readingLine ? ++count : count;
    } catch (FileNotFoundException e) {
      System.out.println("File not found or given directory is incorrect.");
      e.printStackTrace();
      return 0;
    } finally {
      is.close();
    }
  }


  public static ClassLoader getClassLoader() {
    return TestUtils.class.getClassLoader();
  }


  private static String StringAugmentTree(String t) {
    return t.replace("apply[times", "[.$*$")
        .replace("apply[plus", "[.$+$")
        .replace("apply[minus", "[.$-$")
        .replace("apply[eq", "[.$=$")
        .replace("apply[divide", "[.$/$")
        .replace("apply[leq", "[.$\\leq$")
        .replaceAll("ci\\[(\\w)\\]", "\\$$1\\$")
        .replaceAll("cn\\[(\\d+)\\]", "\\$$1\\$")
        .replaceAll("apply\\[(\\w+)", "[.@ $1")
        .replaceAll("(\\w+\\d?)\\[(\\w+)", "[.$1 $2")
        .replace("]", " ] ");
  }

  public static Connection getConnection() throws SQLException, IOException {
    if (cn == null) {
      cn = DriverManager.getConnection("jdbc:mysql://localhost:3306/mathosphere",
          "mathosphere", getFileContents("testpassword"));
    }
    return cn;
  }
}