@Test public void shouldReturnTotalValue() { Portfolio portfolio = new Portfolio(); Stock stock = new Stock("Infi", 5); portfolio.addStock(stock); // Create StockMarket stockMarket = EasyMock.createMock(StockMarket.class); // Add Expectation EasyMock.expect(stockMarket.getPrice("Infi")).andReturn(10); EasyMock.replay(stockMarket); portfolio.setStockMarket(stockMarket); int totalValue = portfolio.getTotalValue(); Assert.assertEquals("Unexpected portfolio total value", 50, totalValue); // Verify EasyMock.verify(stockMarket); }
@Test public void shouldAddStockToPortfolio() { Portfolio portfolio = new Portfolio(); Stock stock = new Stock("Infi", 5); portfolio.addStock(stock); Assert.assertNotNull(portfolio.getStocks()); Assert.assertEquals("Unexpected stock list size", 1, portfolio.getStocks().size()); }