1 files changed, 22 insertions, 0 deletions

diff --git a/test/MainTest.java b/test/MainTest.java
index 1730a6b..f90e367 100644
--- a/test/MainTest.java
+++ b/test/MainTest.java
@@ -2,8 +2,13 @@ package com.jantuomi.interpreter.test;
 
 import com.jantuomi.interpreter.main.Main;
 import com.jantuomi.interpreter.main.exception.InterpreterException;
+import org.junit.After;
+import org.junit.Before;
 import org.junit.Test;
 
+import java.io.ByteArrayOutputStream;
+import java.io.PrintStream;
+
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertTrue;
 
@@ -11,6 +16,21 @@ import static org.junit.Assert.assertTrue;
  * Created by jan on 10.6.2016.
  */
 public class MainTest {
+
+    private final ByteArrayOutputStream out = new ByteArrayOutputStream();
+    private PrintStream originalOut;
+
+    @Before
+    public void setupStreams() {
+        originalOut = System.out;
+        System.setOut(new PrintStream(out));
+    }
+
+    @After
+    public void cleanStreams() {
+        System.setOut(originalOut);
+    }
+
     @Test
     public void testParseArguments1() {
         String[] notAnArgument = {"--not-an-argument"};
@@ -26,6 +46,8 @@ public class MainTest {
     @Test
     public void testWholeProcedure() throws InterpreterException {
         String input = "- + 2 5 1";
+
         Main.run(input);
+        assertTrue(out.toString().contains("6"));
     }
 }
\ No newline at end of file