@Test
public void weightedFriendRecommendation() throws Exception {
deleteSocialGraph();
Person a = personRepository.createPerson("a");
Person b = personRepository.createPerson("b");
Person c = personRepository.createPerson("c");
Person d = personRepository.createPerson("d");
Person e = personRepository.createPerson("e");
Person f = personRepository.createPerson("f");
// A is friends with B,C and D
a.addFriend(b);
a.addFriend(c);
a.addFriend(d);
// E is only friend with B
e.addFriend(b);
// F is friend with B, C, D
f.addFriend(b);
f.addFriend(c);
f.addFriend(d);
ArrayList<Person> recommendations =
fromIterableToArrayList(a.getFriendRecommendation(2).iterator());
// Recommend F to A
assertThat(recommendations.get(0), equalTo(f));
// Recommend E to A
assertThat(recommendations.get(1), equalTo(e));
}
@Test
public void singleFriendRecommendation() throws Exception {
deleteSocialGraph();
Person a = personRepository.createPerson("a");
Person b = personRepository.createPerson("b");
Person c = personRepository.createPerson("c");
Person d = personRepository.createPerson("d");
Person e = personRepository.createPerson("e");
// A is friends with B,C and D
a.addFriend(b);
a.addFriend(c);
a.addFriend(d);
// E is also friend with B, C and D
e.addFriend(b);
e.addFriend(c);
e.addFriend(d);
Person recommendation = IteratorUtil.single(a.getFriendRecommendation(1).iterator());
assertThat(recommendation, equalTo(e));
}
