diff options
Diffstat (limited to 'src/test/java/edu/brown/cs')
4 files changed, 49 insertions, 13 deletions
diff --git a/src/test/java/edu/brown/cs/student/DBQuerierTest.java b/src/test/java/edu/brown/cs/student/DBQuerierTest.java index b3ea140..4bcb193 100644 --- a/src/test/java/edu/brown/cs/student/DBQuerierTest.java +++ b/src/test/java/edu/brown/cs/student/DBQuerierTest.java @@ -54,7 +54,7 @@ public class DBQuerierTest { @Test public void testEmptyDatabase(){ 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???"); } 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(); + } } diff --git a/src/test/java/edu/brown/cs/student/LinkMapperTest.java b/src/test/java/edu/brown/cs/student/LinkMapperTest.java index 3d4bedc..643f63a 100644 --- a/src/test/java/edu/brown/cs/student/LinkMapperTest.java +++ b/src/test/java/edu/brown/cs/student/LinkMapperTest.java @@ -76,7 +76,7 @@ public class LinkMapperTest { @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???"); } diff --git a/src/test/java/edu/brown/cs/student/SuspicionRankerTest.java b/src/test/java/edu/brown/cs/student/SuspicionRankerTest.java index d641507..266eb30 100644 --- a/src/test/java/edu/brown/cs/student/SuspicionRankerTest.java +++ b/src/test/java/edu/brown/cs/student/SuspicionRankerTest.java @@ -10,8 +10,11 @@ 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 { @@ -39,14 +42,16 @@ 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(); } + + } |
