diff options
Diffstat (limited to 'src/test/java/edu/brown/cs')
5 files changed, 140 insertions, 18 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/ProfitCalculationTest.java b/src/test/java/edu/brown/cs/student/ProfitCalculationTest.java new file mode 100644 index 0000000..99500d0 --- /dev/null +++ b/src/test/java/edu/brown/cs/student/ProfitCalculationTest.java @@ -0,0 +1,56 @@ +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; + +import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertTrue; + +public class ProfitCalculationTest { + + /** 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???"); + } + } + + @After + public void tearDown() { + db = null; + } + + @Test + 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(); + } + + +}
\ No newline at end of file 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(); + + } + + } |