1 package org.simantics.scl.compiler.tests;
3 import java.io.IOException;
4 import java.io.InputStream;
5 import java.io.StringWriter;
6 import java.nio.charset.Charset;
7 import java.util.ArrayList;
8 import java.util.Arrays;
9 import java.util.regex.Pattern;
11 import org.junit.Assert;
12 import org.simantics.scl.compiler.errors.Failable;
13 import org.simantics.scl.compiler.errors.Failure;
14 import org.simantics.scl.compiler.module.ImportDeclaration;
15 import org.simantics.scl.compiler.module.Module;
16 import org.simantics.scl.compiler.module.options.ModuleCompilationOptions;
17 import org.simantics.scl.compiler.module.repository.ModuleRepository;
18 import org.simantics.scl.compiler.module.repository.UpdateListener;
19 import org.simantics.scl.compiler.source.ModuleSource;
20 import org.simantics.scl.compiler.source.StringModuleSource;
21 import org.simantics.scl.compiler.source.repository.MapModuleSourceRepository;
22 import org.simantics.scl.compiler.top.ValueNotFound;
23 import org.simantics.scl.runtime.SCLContext;
24 import org.simantics.scl.runtime.reporting.SCLReportingHandler;
25 import org.simantics.scl.runtime.reporting.WriterSCLReportingHandler;
27 public class TestBase {
29 public static final ModuleRepository PRELUDE_MODULE_REPOSITORY = InitialRepository.getInitialRepository();
30 private static final Pattern TEST_SEPARATOR = Pattern.compile("^--+ *$", Pattern.MULTILINE);
31 private static final Charset UTF8 = Charset.forName("UTF-8");
35 public TestBase(String path) {
39 protected void test() {
40 String testModuleName = Thread.currentThread().getStackTrace()[2].getMethodName();
41 String testPath = path + "/" + testModuleName + ".scl";
44 String[] testParts = readTestParts(testPath);
47 ArrayList<String> auxModuleNameList = new ArrayList<String>();
48 while(j < testParts.length) {
49 String part = testParts[j];
50 if(part.startsWith("// module "))
51 auxModuleNameList.add(part.substring(10).split("\\n", 2)[0].trim());
57 String[] moduleNames = new String[mainId+1];
58 String[] moduleTexts = new String[mainId+1];
59 for(int i=0;i<mainId;++i) {
60 moduleNames[i] = auxModuleNameList.get(i);
61 moduleTexts[i] = testParts[i];
63 moduleNames[mainId] = testModuleName;
65 for(;j<testParts.length;j+=2) {
66 moduleTexts[mainId] = testParts[j];
67 String expectedOutput = j+1<testParts.length ? testParts[j+1] : "";
68 String actualOutput = test(moduleNames, moduleTexts);
70 canonicalizeOutput(expectedOutput),
71 canonicalizeOutput(actualOutput));
73 } catch(IOException e) {
74 throw new RuntimeException(e);
75 } catch (ValueNotFound e) {
76 throw new RuntimeException(e);
80 private static String canonicalizeOutput(String text) {
81 return text.trim().replace("\r\n", "\n");
84 protected String test(String[] moduleNames, String[] moduleTexts) throws ValueNotFound {
85 if(moduleNames.length != moduleTexts.length)
86 throw new IllegalArgumentException();
87 /*for(int i=0;i<moduleNames.length;++i) {
88 System.out.println("-- " + moduleNames[i] + " --");
89 System.out.println(moduleTexts[i]);
92 ModuleSource[] moduleSources = new ModuleSource[moduleNames.length];
93 for(int i=0;i<moduleNames.length;++i)
94 moduleSources[i] = new StringModuleSource(
95 moduleNames[i], getClass().getClassLoader(), moduleTexts[i]) {
97 protected ImportDeclaration[] getBuiltinImports(UpdateListener listener) {
98 return ImportDeclaration.ONLY_BUILTINS;
101 ModuleRepository testRepository = new ModuleRepository(
102 PRELUDE_MODULE_REPOSITORY,
103 new MapModuleSourceRepository(moduleSources));
104 testRepository.setAdvisor(moduleName -> ModuleCompilationOptions.SILENT);
105 int lastId = moduleNames.length-1;
106 Failable<Module> result = testRepository.getModule(moduleNames[lastId]);
107 if(!result.didSucceed())
108 return ((Failure)result).toString(moduleTexts[lastId]);
110 SCLContext context = SCLContext.getCurrent();
111 StringWriter writer = new StringWriter();
112 Object oldReportingHandler = context.put(SCLReportingHandler.REPORTING_HANDLER, new WriterSCLReportingHandler(writer));
114 Object main = testRepository.getRuntimeModule(moduleNames[lastId]).getResult().getValue("main");
115 writer.write(String.valueOf(main));
116 return writer.toString();
118 context.put(SCLReportingHandler.REPORTING_HANDLER, oldReportingHandler);
123 private String[] readTestParts(String testPath) throws IOException {
124 InputStream stream = getClass().getResourceAsStream(testPath);
126 byte[] buffer = new byte[1024];
129 int c = stream.read(buffer, pos, buffer.length-pos);
133 if(pos < buffer.length)
135 buffer = Arrays.copyOf(buffer, pos*2);
137 String text = new String(buffer, 0, pos, UTF8);
138 String[] result = TEST_SEPARATOR.split(text);
139 for(int i=1;i<result.length;++i) {
140 if(result[i].startsWith("\r\n"))
141 result[i] = result[i].substring(2);
142 if(result[i].startsWith("\n") || result[i].startsWith("\r"))
143 result[i] = result[i].substring(1);