source: java2python/src/test/java/tudelft/utilities/j2p/ActionsTestTest.java@ 383

package tudelft.utilities.j2p;

import static org.junit.Assert.assertEquals;

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

import org.junit.Test;

import tudelft.utilities.j2p.t.PyTest;
import tudelft.utilities.logging.Reporter;
import tudelft.utilities.pyrunner.PythonError;
import tudelft.utilities.pyrunner.PythonVenv;

/**
 * A more complex package involving multiple files.
 */
public class ActionsTestTest extends PyTest {

        private static final String TESTDIR = "src/test/java/";
        private static final Path TESTPATH = Path.of(TESTDIR);
        private static final String FULLNAME = "testcode/actions/ActionsTest.java";

        @Test
        public void test() throws IOException, ClassNotFoundException {
                System.out.println(getProgram());
        }

        private PyProgram getProgram() throws FileNotFoundException {
                // get all java files under TESTDIR/testcode/actions
                File dir = TESTPATH.resolve("testcode/actions").toFile();
                File[] javafiles = dir.listFiles(new FilenameFilter() {
                        @Override
                        public boolean accept(File dir, String name) {
                                return name.endsWith(".java");
                        }
                });

                List<PyModule> modules = new LinkedList<>();
                for (File file : javafiles) {
                        // re-relativize
                        file = TESTPATH.relativize(file.toPath()).toFile();
                        modules.add(PyModule.fromJavaFile(TESTPATH, file));
                }

                return new PyProgram(modules);
        }

        @Test
        public void runFromZip() throws FileNotFoundException, IOException,
                        InterruptedException, PythonError {
                File zipfile = zip(translate(TESTDIR + FULLNAME), new File(FULLNAME));
                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.actions.ActionsTest"), 10000);
                System.out.println(res);
                assertEquals("Person[Kees,25]\n"
                                + "{\"PersonJson\":{\"name\":\"Kees\",\"age\":25}}\n"
                                + "True\n", res.replace(" ", ""));
                // as in Java but true->True and pyson inserts some extra whitespace
        }

}