TestCaseUtils.java

package org.libresonic.player;

import org.apache.commons.io.FileUtils;
import org.libresonic.player.dao.DaoHelper;
import org.libresonic.player.service.MediaScannerService;
import org.springframework.context.ApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class TestCaseUtils {

  private static File libresonicHomeDirForTest = null;

  /**
   * Returns the path of the LIBRESONIC_HOME directory to use for tests.
   * This will create a temporary directory.
   *
   * @return LIBRESONIC_HOME directory path.
   * @throws RuntimeException if it fails to create the temp directory.
   */
  public static String libresonicHomePathForTest() {

    if (libresonicHomeDirForTest == null) {
      try {
        libresonicHomeDirForTest = Files.createTempDirectory("libresonic_test_").toFile();
      } catch (IOException e) {
        throw new RuntimeException("Error while creating temporary LIBRESONIC_HOME directory for tests");
      }
      System.out.println("LIBRESONIC_HOME directory will be "+libresonicHomeDirForTest.getAbsolutePath());
    }
    return libresonicHomeDirForTest.getAbsolutePath();
  }


  /**
   * Cleans the LIBRESONIC_HOME directory used for tests.
   *
   * @throws IOException
     */
  public static void cleanLibresonicHomeForTest() throws IOException {

    File libresonicHomeDir = new File(libresonicHomePathForTest());
    if (libresonicHomeDir.exists() && libresonicHomeDir.isDirectory()) {
      System.out.println("Delete libresonic home (ie. "+libresonicHomeDir.getAbsolutePath()+").");
      try {
        FileUtils.deleteDirectory(libresonicHomeDir);
      } catch (IOException e) {
        System.out.println("Error while deleting libresonic home.");
        e.printStackTrace();
        throw e;
      }
    }

  }

  /**
   * Constructs a map of records count per table.
   *
   * @param daoHelper DaoHelper object
   * @return Map table name -> records count
     */
  public static Map<String, Integer> recordsInAllTables(DaoHelper daoHelper) {
    List<String> tableNames = daoHelper.getJdbcTemplate().queryForList("" +
                    "select table_name " +
                    "from information_schema.system_tables " +
                    "where table_name not like 'SYSTEM%'"
            , String.class);
    Map<String, Integer> nbRecords =
            tableNames.stream()
                    .collect(Collectors.toMap(table -> table, table -> recordsInTable(table,daoHelper)));

    return nbRecords;
  }

  /**
   * Counts records in a table.
   *
   * @param tableName
   * @param daoHelper
   * @return
     */
  public static Integer recordsInTable(String tableName, DaoHelper daoHelper) {
    return daoHelper.getJdbcTemplate().queryForObject("select count(1) from " + tableName,Integer.class);
  }


  public static ApplicationContext loadSpringApplicationContext(String baseResources) {
    String applicationContextService = baseResources + "applicationContext-service.xml";
    String applicationContextCache = baseResources + "applicationContext-cache.xml";

    String[] configLocations = new String[]{
            TestCaseUtils.class.getClass().getResource(applicationContextCache).toString(),
            TestCaseUtils.class.getClass().getResource(applicationContextService).toString()
    };
    return new ClassPathXmlApplicationContext(configLocations);
  }


  /**
   * Scans the music library   * @param mediaScannerService
   */
  public static void execScan(MediaScannerService mediaScannerService) {
    mediaScannerService.scanLibrary();

    while (mediaScannerService.isScanning()) {
      try {
        Thread.sleep(1000);
      } catch (InterruptedException e) {
        e.printStackTrace();
      }
    }

  }

}