package net.sourceforge.jenesis4java.util;
   
   /*
    * #%L
    * Jenesis 4 Java Code Generator
    * %%
    * Copyright (C) 2000 - 2016 jenesis4java
    * %%
    * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Lesser General Public License as
   * published by the Free Software Foundation, either version 3 of the
   * License, or (at your option) any later version.
   * 
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Lesser Public License for more details.
   * 
   * You should have received a copy of the GNU General Lesser Public
   * License along with this program.  If not, see
   * <http://www.gnu.org/licenses/lgpl-3.0.html>.
   * #L%
   */
  
  import net.sourceforge.jenesis4java.CompilationUnit;
  import net.sourceforge.jenesis4java.VirtualMachine;
  import org.junit.rules.ExternalResource;
  
  import java.io.File;
  import java.io.IOException;
  
  import static org.junit.Assert.assertTrue;
  
  public class VmInTempFolder extends ExternalResource {
  
      private VirtualMachine vm;
  
      private File tempDirectory;
  
      @Override
      public void before() throws IOException {
          vm = VirtualMachine.getVirtualMachine();
          tempDirectory = File.createTempFile("jenesisTmp", "tmp");
          tempDirectory.delete();
          tempDirectory.mkdirs();
      }
  
      @Override
      public void after() {
          recursiveDelete(tempDirectory);
      }
  
      public CompilationUnit newCompilationUnit() throws IOException {
          return vm.newCompilationUnit(tempDirectory.getCanonicalPath());
      }
  
      public VirtualMachine getVm() {
          return vm;
      }
  
      public void assertFileExists(String path) {
          File pathToFile = new File(tempDirectory, path);
          String errorMsg = "File with path '" + path //
                  + "' (complete path: '" + pathToFile + "') does not exist.";
          assertTrue(errorMsg, pathToFile.exists());
      }
  
      private void recursiveDelete(File file) {
          File[] files = file.listFiles();
          if (files != null) {
              for (File each : files) {
                  recursiveDelete(each);
              }
          }
          file.delete();
      }
  
  }