diff --git a/test/environment/RuntimeEnvironmentTester.java b/test/environment/RuntimeEnvironmentTester.java index fa87a7c..1dbbe1a 100644 --- a/test/environment/RuntimeEnvironmentTester.java +++ b/test/environment/RuntimeEnvironmentTester.java @@ -77,4 +77,39 @@ public class RuntimeEnvironmentTester { assertEquals(errorManager, environment.getErrorManager()); } + @Test + public void assignOutputDecorator() { + environment.setOutputDecorator(s -> "[" + s + "]"); + + assertEquals("[test]", environment.decorateOutput("test")); + } + + @Test + public void assignValueOutputDecorator() { + environment.setValueOutputDecorator(s -> "(" + s + ")"); + + assertEquals("(test)", environment.decorateValueOutput("test")); + } + + @Test + public void assignWarningOutputDecorator() { + environment.setWarningOutputDecorator(s -> "|" + s + "|"); + + assertEquals("|test|", environment.decorateWarningOutput("test")); + } + + @Test + public void assignErrorOutputDecorator() { + environment.setErrorOutputDecorator(s -> "{" + s + "}"); + + assertEquals("{test}", environment.decorateErrorOutput("test")); + } + + @Test + public void assignCriticalOutputDecorator() { + environment.setCriticalOutputDecorator(s -> "/" + s + "/"); + + assertEquals("/test/", environment.decorateCriticalOutput("test")); + } + }