1 files changed, 52 insertions, 0 deletions
diff --git a/src/test/com/jantuomi/tunkki/MainTest.java b/src/test/com/jantuomi/tunkki/MainTest.java
new file mode 100644
index 0000000..d76bac1
--- /dev/null
+++ b/src/test/com/jantuomi/tunkki/MainTest.java
@@ -0,0 +1,52 @@
+package com.jantuomi.tunkki;
+
+import com.jantuomi.tunkki.exception.BorkError;
+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;
+
+/**
+ * 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"};
+        assertFalse(Main.parseArguments(notAnArgument));
+    }
+
+    @Test
+    public void testParseArguments2() {
+        String[] fileTestArgument = {"-f", "test.file"};
+        assertTrue(Main.parseArguments(fileTestArgument));
+    }
+
+    @Test
+    public void testWholeProcedure() throws BorkError {
+        String input = "- + 2 5 1";
+
+        Main.run(input);
+        assertTrue(out.toString().contains("6"));
+    }
+}
+\ No newline at end of file