1 files changed, 44 insertions, 9 deletions
diff --git a/src/test/java/edu/brown/cs/student/SuspicionRankerTest.java b/src/test/java/edu/brown/cs/student/SuspicionRankerTest.java
index d641507..3f8ea3d 100644
--- a/src/test/java/edu/brown/cs/student/SuspicionRankerTest.java
+++ b/src/test/java/edu/brown/cs/student/SuspicionRankerTest.java
@@ -10,24 +10,33 @@ import org.junit.Before;
 import org.junit.Test;
 
 import java.time.Instant;
+import java.util.Comparator;
 import java.util.List;
 import java.util.Map;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
 
 public class SuspicionRankerTest {
 
+  //  ./cs32-test tests/student/hub/*.test
+
+  //friday april 9th
+  //private Instant start = Instant.ofEpochMilli(1618004180000L);
+
+  //saturday april 10th
+  //private Instant end = Instant.ofEpochMilli(1618019436000L);
+
   //12 am on 3/11 in UTC
-  //private Instant start = Instant.parse("2021-03-11T05:00:00.00Z");
-  private Instant start = Instant.ofEpochMilli(161800418000L);
+  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 Instant end = Instant.ofEpochMilli(1618019436000L);
+  private Instant end = Instant.parse("2021-03-28T05:00:00.00Z");
 
   private DatabaseQuerier db;
 
   @Before
   public void setUp() {
     try{
-      db = new DatabaseQuerier("data/trades.sqlite3");
+      db = new DatabaseQuerier("data/lil_mock.sqlite3");
     } catch(Exception e){
       System.out.println("DBQuerier Test, couldn't connect to db???");
     }
@@ -39,14 +48,40 @@ public class SuspicionRankerTest {
   }
 
   @Test
-  public void testMapper(){
+  public void testSuspicionRanksOnMockData(){
     setUp();
     SuspicionRanker r = new SuspicionRanker(db);
     List<Holder> him = r.getSuspicionScoreList(start, end);
-    //System.out.println(him);
-    for(Holder guy: him){
-      System.out.println(guy.getName() + " " + guy.getSuspicionScore());
+    assertEquals(6, him.size());
+    for(Holder p: him){
+      System.out.println(p.toTestString());
+    }
+    tearDown();
+  }
+
+  @Test
+  public void testSusRankEmptyDB(){
+    try{
+      db = new DatabaseQuerier("data/testing/empty.sqlite3");
+    } catch(Exception e){
+      System.out.println("DBQuerier Test, couldn't connect to db???");
     }
+    SuspicionRanker r = new SuspicionRanker(db);
+    List<Holder> vals = r.getSuspicionScoreList(start, end);
+    assertTrue(vals.isEmpty());
     tearDown();
+
   }
+
+  @Test
+  public void testBadDates(){
+    setUp();
+    SuspicionRanker r = new SuspicionRanker(db);
+    List<Holder> vals = r.getSuspicionScoreList(end, start);
+    assertTrue(vals.isEmpty());
+    tearDown();
+
+  }
+
+
 }