Small changes to testing for parsing.

1 files changed, 15 insertions, 13 deletions

diff --git a/src/test/java/edu/brown/cs/student/TransactionTest.java b/src/test/java/edu/brown/cs/student/TransactionTest.java
index f9a00f7..e8b95bc 100644
--- a/src/test/java/edu/brown/cs/student/TransactionTest.java
+++ b/src/test/java/edu/brown/cs/student/TransactionTest.java
@@ -31,20 +31,8 @@ public class TransactionTest {
   @Test
   public void singleTrade(){
     setUp();
-    Document doc = _xmlParser.parse("data/xml_single_trade_test.xml");
-    assertNotEquals(doc, null);
-
-    // One trades in transaction
-    Transaction transaction = new Transaction(doc);
-    assertEquals(transaction.getTrades().size(), 1);
-
-    // TODO: add more qualities on trade to test...
-    Trade firstTrade = transaction.getTrades().get(0);
-    assertEquals(firstTrade.getNumShares(), 8236);
-
     tearDown();
   }
-
   @Test
   public void multipleTrades(){
     setUp();
@@ -90,7 +78,21 @@ public class TransactionTest {
   @Test
   public void noTrades(){
     setUp();
-    // TODO: add case, but won't realistically come up
+    // TODO: add case from example
     tearDown();
   }
+
+  @Test
+  public void bad0Trade(){
+    setUp();
+    Document doc = _xmlParser.parse("data/xml_single_trade_test.xml");
+    assertNotEquals(doc, null);
+
+    // One trades in transaction
+    Transaction transaction = new Transaction(doc);
+    assertEquals(transaction.getTrades().size(), 0);
+
+    tearDown();
+  }
+
 }