1 files changed, 55 insertions, 11 deletions
diff --git a/src/test/java/edu/brown/cs/student/HubRankTest.java b/src/test/java/edu/brown/cs/student/HubRankTest.java
index 9ba0987..cbe6112 100644
--- a/src/test/java/edu/brown/cs/student/HubRankTest.java
+++ b/src/test/java/edu/brown/cs/student/HubRankTest.java
@@ -7,9 +7,12 @@ import edu.brown.cs.student.term.hub.LinkMapper;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
 
 import java.time.Instant;
 import java.util.Map;
+import java.util.Set;
 
 public class HubRankTest {
 
@@ -30,25 +33,66 @@ public class HubRankTest {
     }
   }
 
-  /*
-  * try{
-
-    } catch(Exception e) {
-      System.out.println("Error in test");
-    }*/
-
   @After
   public void tearDown() {
     db = null;
   }
 
   @Test
-  public void testMapper(){
+  public void testEmptyDB(){
+    try{
+      db = new DatabaseQuerier("data/empty.sqlite3");
+    } catch(Exception e){
+      System.out.println("DBQuerier Test, couldn't connect to db???");
+    }
+    LinkMapper lm = new LinkMapper(db);
+    HubSearch hub = new HubSearch(lm);
+    Map<Holder, Double> hubRanks = hub.runHubSearch(start, end);
+    assertTrue(hubRanks.isEmpty());
+    tearDown();
+  }
+
+  @Test
+  public void flippedDates(){
+    setUp();
+    LinkMapper lm = new LinkMapper(db);
+    HubSearch hub = new HubSearch(lm);
+    Map<Holder, Double> hubRanks = hub.runHubSearch(end, start);
+    assertTrue(hubRanks.isEmpty());
+    tearDown();
+
+  }
+
+  @Test
+  public void testHubRankSmall(){
     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);
+    Map<Holder, Double> hubRanks = hub.runHubSearch(start, end);
+    assertEquals(6, hubRanks.size());
+    tearDown();
+  }
+
+  @Test
+  public void testCheckRightValues(){
+    setUp();
+    LinkMapper lm = new LinkMapper(db);
+    HubSearch hub = new HubSearch(lm);
+    Map<Holder, Double> hubRanks = hub.runHubSearch(start, end);
+    Holder don = new Holder(1, "Don");
+    Holder mitch = new Holder(2, "Mitch");
+    Holder nancy = new Holder(3, "Nancy");
+    Holder midge = new Holder(4, "Midge");
+    Holder bob = new Holder(5, "Bob");
+    Holder jane = new Holder(6, "Jane");
+
+    System.out.println(hubRanks);
+    assertTrue(hubRanks.get(mitch) > hubRanks.get(don));
+    assertTrue(hubRanks.get(don) > hubRanks.get(bob));
+    assertTrue(hubRanks.get(bob) > hubRanks.get(nancy));
+    assertTrue(hubRanks.get(nancy)> hubRanks.get(jane));
+    assertTrue(hubRanks.get(nancy) > hubRanks.get(midge));
+
+    tearDown();
   }
 }