Commit d21b7f53 authored by Rohit Prasad's avatar Rohit Prasad

Add Stock class tests

parent 0a5d53d0
...@@ -7,12 +7,14 @@ class BuyTradeTest { ...@@ -7,12 +7,14 @@ class BuyTradeTest {
@Test @Test
void buyTradeTest() { void buyTradeTest() {
Customer c = new Customer("c1"); Customer c = new Customer("c1");
BuyTrade buyTrade = new BuyTrade(1, c, 100, 200); Stock stock = new Stock("Apple");
BuyTrade buyTrade = new BuyTrade(1, c, stock, 100, 200);
assertEquals(buyTrade.getCustomer().getId(), c.getId()); assertEquals(buyTrade.getCustomer().getId(), c.getId());
assertEquals(buyTrade.getPrice(), 200); assertEquals(buyTrade.getPrice(), 200);
assertEquals(buyTrade.getQty(), 100); assertEquals(buyTrade.getQty(), 100);
assertEquals(buyTrade.getTime(), 1); assertEquals(buyTrade.getTime(), 1);
assertEquals(buyTrade.getType(), "BUY"); assertEquals(buyTrade.getType(), "BUY");
assertEquals(buyTrade.getStock().getName(), "Apple");
} }
} }
...@@ -7,12 +7,14 @@ class SellTradeTest { ...@@ -7,12 +7,14 @@ class SellTradeTest {
@Test @Test
void sellTradeTest() { void sellTradeTest() {
Customer c = new Customer("c1"); Customer c = new Customer("c1");
SellTrade sellTrade = new SellTrade(1, c, 100, 200); Stock stock = new Stock("Apple");
SellTrade sellTrade = new SellTrade(1, c, stock, 100, 200);
assertEquals(sellTrade.getCustomer().getId(), c.getId()); assertEquals(sellTrade.getCustomer().getId(), c.getId());
assertEquals(sellTrade.getPrice(), 200); assertEquals(sellTrade.getPrice(), 200);
assertEquals(sellTrade.getQty(), 100); assertEquals(sellTrade.getQty(), 100);
assertEquals(sellTrade.getTime(), 1); assertEquals(sellTrade.getTime(), 1);
assertEquals(sellTrade.getType(), "SELL"); assertEquals(sellTrade.getType(), "SELL");
assertEquals(sellTrade.getStock().getName(), "Apple");
} }
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment