2 files changed, 62 insertions, 1 deletions
diff --git a/src/test/java/edu/brown/cs/student/HubRankTest.java b/src/test/java/edu/brown/cs/student/HubRankTest.java
new file mode 100644
index 0000000..9ba0987
--- /dev/null
+++ b/src/test/java/edu/brown/cs/student/HubRankTest.java
@@ -0,0 +1,54 @@
+package edu.brown.cs.student;
+
+import edu.brown.cs.student.term.DatabaseQuerier;
+import edu.brown.cs.student.term.hub.Holder;
+import edu.brown.cs.student.term.hub.HubSearch;
+import edu.brown.cs.student.term.hub.LinkMapper;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.time.Instant;
+import java.util.Map;
+
+public class HubRankTest {
+
+  /** these should span the entire mock dataset */
+  //12 am on 3/11 in UTC
+  private Instant start = Instant.parse("2021-03-11T05:00:00.00Z");
+  //12 am on 3/28 in UTC
+  private Instant end = Instant.parse("2021-03-28T05:00:00.00Z");
+
+  private DatabaseQuerier db;
+
+  @Before
+  public void setUp() {
+    try{
+      db = new DatabaseQuerier("data/lil_mock.sqlite3");
+    } catch(Exception e){
+      System.out.println("DBQuerier Test, couldn't connect to db???");
+    }
+  }
+
+  /*
+  * try{
+
+    } catch(Exception e) {
+      System.out.println("Error in test");
+    }*/
+
+  @After
+  public void tearDown() {
+    db = null;
+  }
+
+  @Test
+  public void testMapper(){
+    setUp();
+    LinkMapper lm = new LinkMapper(db);
+    lm.makeFollowerLinks(start, end);
+    HubSearch hub = new HubSearch(lm);
+    Map<Holder, Double> him = hub.runHubSearch(start, end);
+    System.out.println(him);
+  }
+}
diff --git a/src/test/java/edu/brown/cs/student/LinkMapperTest.java b/src/test/java/edu/brown/cs/student/LinkMapperTest.java
index 9b46d5e..2683b90 100644
--- a/src/test/java/edu/brown/cs/student/LinkMapperTest.java
+++ b/src/test/java/edu/brown/cs/student/LinkMapperTest.java
@@ -1,12 +1,15 @@
 package edu.brown.cs.student;
 
 import edu.brown.cs.student.term.DatabaseQuerier;
+import edu.brown.cs.student.term.hub.Holder;
+import edu.brown.cs.student.term.hub.HubSearch;
 import edu.brown.cs.student.term.hub.LinkMapper;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 
 import java.time.Instant;
+import java.util.Map;
 
 public class LinkMapperTest {
 
@@ -21,7 +24,7 @@ public class LinkMapperTest {
   @Before
   public void setUp() {
     try{
-      db = new DatabaseQuerier("data/mock_trades.sqlite3");
+      db = new DatabaseQuerier("data/lil_mock.sqlite3");
     } catch(Exception e){
       System.out.println("DBQuerier Test, couldn't connect to db???");
     }
@@ -44,5 +47,9 @@ public class LinkMapperTest {
     setUp();
     LinkMapper lm = new LinkMapper(db);
     lm.makeFollowerLinks(start, end);
+    HubSearch hub = new HubSearch(lm);
+    Map<Holder, Double> him = hub.runHubSearch(start, end);
+    System.out.println(him);
   }
+
 }