Added functionality that adds to the DB directly. Also, can handle multiple trades within a transaction now.

2 files changed, 20 insertions, 5 deletions

diff --git a/src/test/java/edu/brown/cs/student/TradeTest.java b/src/test/java/edu/brown/cs/student/TransactionTest.java
index 90d656d..a5e2987 100644
--- a/src/test/java/edu/brown/cs/student/TradeTest.java
+++ b/src/test/java/edu/brown/cs/student/TransactionTest.java
@@ -12,7 +12,7 @@ import org.w3c.dom.NodeList;
 
 import static org.junit.Assert.*;
 
-public class TradeTest {
+public class TransactionTest {
   private XmlParser _xmlParser;
   private Document _doc;
 
@@ -29,7 +29,7 @@ public class TradeTest {
   @Test
   public void personDataParse(){
     setUp();
-    Document doc = _xmlParser.parse("data/xml_trade_test.xml");
+    Document doc = _xmlParser.parse("data/xml_single_trade_test.xml");
     assertNotNull(doc);
 
     // Id of person
@@ -48,9 +48,24 @@ public class TradeTest {
   }
 
   @Test
+  public void securityDataParse(){
+    setUp();
+    Document doc = _xmlParser.parse("data/xml_single_trade_test.xml");
+    assertNotNull(doc);
+
+    // Ticker of security
+    NodeList idNode = doc.getElementsByTagName("issuerTradingSymbol");
+    assertEquals(idNode.getLength(), 1);
+    String id = idNode.item(0).getTextContent();
+    assertEquals(id, "GATO");
+
+    tearDown();
+  }
+
+  @Test
   public void tradeDataParse(){
     setUp();
-    Document doc = _xmlParser.parse("data/xml_trade_test.xml");
+    Document doc = _xmlParser.parse("data/xml_single_trade_test.xml");
     assertNotEquals(doc, null);
 
     // Data of trade in an array of values
diff --git a/src/test/java/edu/brown/cs/student/XmlParserTest.java b/src/test/java/edu/brown/cs/student/XmlParserTest.java
index d3bc4ff..5f5b7a8 100644
--- a/src/test/java/edu/brown/cs/student/XmlParserTest.java
+++ b/src/test/java/edu/brown/cs/student/XmlParserTest.java
@@ -34,7 +34,7 @@ public class XmlParserTest {
   @Test
   public void parsesLocal(){
     setUp();
-    Document doc = _localXmlParser.parse("data/xml_trade_test.xml");
+    Document doc = _localXmlParser.parse("data/xml_single_trade_test.xml");
     assertNotNull(doc);
 
     // Id of person
@@ -64,7 +64,7 @@ public class XmlParserTest {
   @Test
   public void urlSameAsLocal(){
     setUp();
-    Document local = _localXmlParser.parse("data/xml_trade_test.xml");
+    Document local = _localXmlParser.parse("data/xml_single_trade_test.xml");
     Document url =
         _urlXmlParser.parse("https://www.sec.gov/Archives/edgar/data/1517006/000110465921046242/tm2112036-4_4seq1.xml");