Add tests for equality check

src/BuyTradeTest.java

@@ -16,5 +16,23 @@ class BuyTradeTest {
 		assertEquals(buyTrade.getType(), "BUY");
 		assertEquals(buyTrade.getStock().getName(), "Apple");
 	}
+	
+	@Test
+	void equalityTest() {
+		Customer c1 = new Customer("c1");
+		Customer c2 = new Customer("c2");
+		Stock s1 = new Stock("Apple");
+		Stock s2 = new Stock("Google");
+		BuyTrade bt1 = new BuyTrade(1, c1, s1, 100, 200);
+		SellTrade st1 = new SellTrade(1, c1, s1, 100, 200);
+		
+		// checking if a two objects created with exact same arguments
+		// are indeed equal
+		assertEquals(new BuyTrade(1, c1, s1, 100, 200), bt1);
+		assertNotEquals(new BuyTrade(1, c1, s1, 100, 300), bt1);
+		
+		// checking SellTrade and BuyTrade are not equal
+		assertNotEquals(bt1, st1);
+	}
 
 }

src/SellTradeTest.java

@@ -17,4 +17,22 @@ class SellTradeTest {
 		assertEquals(sellTrade.getStock().getName(), "Apple");
 	}
 
+	@Test
+	void equalityTest() {
+		Customer c1 = new Customer("c1");
+		Customer c2 = new Customer("c2");
+		Stock s1 = new Stock("Apple");
+		Stock s2 = new Stock("Google");
+		BuyTrade bt1 = new BuyTrade(1, c1, s1, 100, 200);
+		SellTrade st1 = new SellTrade(1, c1, s1, 100, 200);
+		
+		// checking if a two objects created with exact same arguments
+		// are indeed equal
+		assertEquals(new SellTrade(1, c1, s1, 100, 200), st1);
+		assertNotEquals(new SellTrade(1, c1, s1, 100, 300), st1);
+		
+		// checking SellTrade and BuyTrade are not equal
+		assertNotEquals(bt1, st1);
+	}
+
 }