diff --git a/src/test/java/millions/CSVStockFileParserTest.java b/src/test/java/millions/CSVStockFileParserTest.java index 2b53311..8cb459b 100644 --- a/src/test/java/millions/CSVStockFileParserTest.java +++ b/src/test/java/millions/CSVStockFileParserTest.java @@ -11,6 +11,8 @@ import java.nio.file.Files; import java.nio.file.Path; +import static org.junit.jupiter.api.Assertions.assertEquals; + public class CSVStockFileParserTest { @TempDir static Path tempDir; @@ -32,9 +34,8 @@ public static void setUpTestFile() throws Exception { @Test public void parseStockFileTest(){ StockFileReader stockFileReader = new StockFileReader(sharedFile); - stockFileReader.readFile().forEach(System.out::println); CSVStockFileParser parser = new CSVStockFileParser(stockFileReader.readFile()); - parser.parse().forEach(s -> System.out.println(s.toString())); + assertEquals(3, parser.parse().size()); } } diff --git a/src/test/java/millions/StockFileReaderTest.java b/src/test/java/millions/StockFileReaderTest.java new file mode 100644 index 0000000..c40e4ca --- /dev/null +++ b/src/test/java/millions/StockFileReaderTest.java @@ -0,0 +1,35 @@ +package millions; + +import millions.controller.fileIO.StockFileReader; +import org.junit.jupiter.api.BeforeAll; +import org.junit.jupiter.api.Test; +import org.junit.jupiter.api.io.TempDir; + +import java.nio.file.Files; +import java.nio.file.Path; + +import static org.junit.jupiter.api.Assertions.assertEquals; + +public class StockFileReaderTest { + @TempDir + static Path tempDir; + + static Path sharedFile; + @BeforeAll + public static void setUpTestFile() throws Exception { + sharedFile = Files.createFile(tempDir.resolve("file.csv")); + String string = "# Top 500 US Stocks by Market Cap\n"; + string += "# Ticker,Name,Price\n"; + string += "\n"; + string += "NVDA,Nvidia,191.27\n"; + string += "AAPL,Apple Inc.,276.43\n"; + string += "MSFT,Microsoft,404.68\n"; + Files.writeString(sharedFile, string); + } + + @Test + public void testReadStockFile() throws Exception { + StockFileReader stockFileReader = new StockFileReader(sharedFile); + assertEquals(6, stockFileReader.readFile().size()); + } +}