1 files changed, 38 insertions, 7 deletions
diff --git a/src/test/java/edu/brown/cs/student/HubRankTest.java b/src/test/java/edu/brown/cs/student/HubRankTest.java
index cbe6112..07fd282 100644
--- a/src/test/java/edu/brown/cs/student/HubRankTest.java
+++ b/src/test/java/edu/brown/cs/student/HubRankTest.java
@@ -24,6 +24,13 @@ public class HubRankTest {
 
   private DatabaseQuerier db;
 
+  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");
+
   @Before
   public void setUp() {
     try{
@@ -41,7 +48,7 @@ public class HubRankTest {
   @Test
   public void testEmptyDB(){
     try{
-      db = new DatabaseQuerier("data/empty.sqlite3");
+      db = new DatabaseQuerier("data/testing/empty.sqlite3");
     } catch(Exception e){
       System.out.println("DBQuerier Test, couldn't connect to db???");
     }
@@ -79,12 +86,6 @@ public class HubRankTest {
     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));
@@ -95,4 +96,34 @@ public class HubRankTest {
 
     tearDown();
   }
+
+  @Test
+  public void testFollowersSetsConstructedProperly(){
+    setUp();
+    LinkMapper lm = new LinkMapper(db);
+    HubSearch hub = new HubSearch(lm);
+    Map<Holder, Double> hubRanks = hub.runHubSearch(start, end);
+
+    for(Holder person: hubRanks.keySet()){
+      System.out.println(person.getName() + " " + person.getFollowers());
+      if(person.equals(don)) {
+        assertTrue(person.getFollowers().contains(mitch));
+        assertTrue(person.getFollowers().contains(nancy));
+        assertTrue(person.getFollowers().contains(midge));
+      } else if(person.equals(bob)){
+        assertTrue(person.getFollowers().contains(mitch));
+        assertTrue(person.getFollowers().contains(midge));
+      } else if(person.equals(nancy)){
+        assertTrue(person.getFollowers().contains(midge));
+      } else if(person.equals(mitch)){
+        assertTrue(person.getFollowers().contains(nancy));
+        assertTrue(person.getFollowers().contains(midge));
+        assertTrue(person.getFollowers().contains(bob));
+      } else{
+        assertTrue(person.getFollowers().isEmpty());
+      }
+    }
+
+    tearDown();
+  }
 }