source: java2python/tudutils-t/src/test/java/tudelft/utilities/j2p/t/ReporterTest.java

package tudelft.utilities.j2p.t;

import static org.junit.Assert.assertTrue;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.logging.Level;

import org.eclipse.jdt.annotation.NonNull;
import org.junit.Test;

import tudelft.utilities.j2p.PyProgram;
import tudelft.utilities.logging.Reporter;
import tudelft.utilities.pyrunner.PythonError;
import tudelft.utilities.pyrunner.PythonVenv;

public class ReporterTest {

        private static final String TESTDIR = "src/test/java/";
        private static final String FULLNAME = "testcode/Reporter1.java";

        @Test
        public void test()
                        throws IOException, ClassNotFoundException, TranslationException {
                System.out.println("TRANSLATION:");
                final @NonNull PyProgram program = PyProgram.fromModules(
                                Paths.get(TESTDIR), Arrays.asList(new File(FULLNAME)),
                                Paths.get(TESTDIR), Collections.emptyList());
                System.out.println(program.toString());
        }

        @Test
        public void runFromZip() throws FileNotFoundException, IOException,
                        InterruptedException, PythonError, TranslationException {
                final @NonNull PyProgram program = PyProgram.fromModules(
                                Paths.get(TESTDIR), Arrays.asList(new File(FULLNAME)),
                                Paths.get(TESTDIR), Collections.emptyList());
                File zipfile = program.getZip();

                System.out.println("zip file at " + zipfile);
                Reporter reporter = new Reporter() {
                        private List<String> warnings = new LinkedList<>();

                        @Override
                        public void log(Level level, String msg) {
                                log(level, msg, null);
                        }

                        @Override
                        public void log(Level level, String msg, Throwable thrown) {
                                if (level == Level.WARNING || level == Level.SEVERE) {
                                        warnings.add(msg);
                                        System.err.println(msg);
                                        if (thrown != null)
                                                thrown.printStackTrace();
                                } else {
                                        System.out.println(level + ":" + msg);
                                }
                        }
                };
                PythonVenv venv = new PythonVenv(zipfile, reporter);
                String res = venv.call(Arrays.asList("-m", "testcode.Reporter1"),
                                10000);
                System.out.println(res);
                assertTrue(res.contains("ok"));
                assertTrue(res.contains("INFO"));
        }

}