buy-and-sell-n-times/src/test/java/SolutionTest.java

import java.io.*;

import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.CsvFileSource;

import static java.nio.charset.StandardCharsets.UTF_8;
import static org.junit.jupiter.api.Assertions.*;

class SolutionTest {
	
	static long totalDuration;
	
	@BeforeAll
	static void beforeAll() {
		totalDuration = 0;
	}
	
	@AfterAll
	static void afterAll() {
		System.out.println( String.format( "All solutions took %01d.%03d secs",  totalDuration / 1000,  totalDuration % 1000 ));
	}
	
	@ParameterizedTest
	@CsvFileSource( resources = "testdata.csv", numLinesToSkip = 1 )
	void main( final String sellCount, final String input, final String expected ) throws IOException {
		// keep original streams
		InputStream oldIn = System.in;
		PrintStream oldOut = System.out;
		PrintStream oldErr = System.err;
		ByteArrayOutputStream bos =  new ByteArrayOutputStream();
		
		try {
			redirectStreams(
					new ByteArrayInputStream( (sellCount + " " + input).getBytes( UTF_8 )),
					new PrintStream( bos, true, UTF_8 ),
					new PrintStream( new ByteArrayOutputStream(), true, UTF_8 )
			);
			
			// start time tracking
			long start = System.currentTimeMillis();
			
			Solution.main( new String[0] );
			
			// stop time tracking
			long duration = System.currentTimeMillis() - start;
			totalDuration += duration;
			
			// restore streams
			redirectStreams( oldIn, oldOut, oldErr );
			
			System.out.println( String.format( "Solution took %01d.%03d secs",  duration / 1000,  duration % 1000 ));
			
			try (BufferedReader chk = new BufferedReader( new InputStreamReader( new ByteArrayInputStream( bos.toByteArray() ), UTF_8 ))) {
				String[] expectedLines = expected.split( "\\s*[|]\\s*" );
				
				int lineCount = 0;
				String line;
				for ( ; (line = chk.readLine()) != null; lineCount++ ) {
					assertEquals( expectedLines[ lineCount ], line );
				}
				assertEquals( expectedLines.length, lineCount );
			}
		}
		finally {
			// restore streams
			redirectStreams( oldIn, oldOut, oldErr );
		}
	}
	
	static void redirectStreams( final InputStream input, final PrintStream output, final PrintStream error ) {
		System.setIn( input );
		System.setOut( output );
		System.setErr( error );
	}
}