package test2903_gmail_com; import static org.junit.Assert.assertEquals; import java.io.ByteArrayOutputStream; import java.io.PrintStream; import org.junit.After; import org.junit.Before; import org.junit.Test; public class JUnitLuckyFiveSecondTest { private final ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); private final PrintStream oPrintStream = System.out; @Before public void setUpStream() { System.setOut(new PrintStream(outputStream)); } @After public void restoreStream() { System.setOut(oPrintStream); } @Test public void isLuckyFiveSecond() { test2903_gmail_com.LuckyFiveSecond.main(null); assertEquals("Is Loop not same", "12345", outputStream.toString()); test2903_gmail_com.LuckyFiveSecond.loop(6); } }