@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());
}
