@Test
public void testSetRestaurantChoice() {
Assert.assertNull(ranking.getCategoryRestaurantRankingByRestaurant(rA).getRestaurantChoice());
ranking.setRestaurantChoice(rA);
Assert.assertNotNull(
ranking.getCategoryRestaurantRankingByRestaurant(rA).getRestaurantChoice());
}
@Test
public void test2Categorias() {
TreeMap<CategoryRestaurant, CategoryRestaurantRanking> list = ranking.getList();
Assert.assertEquals(crA, list.firstEntry().getKey());
list.remove(crA);
Assert.assertEquals(crB, list.firstEntry().getKey());
list.remove(crB);
Assert.assertTrue(list.isEmpty());
}
@Before
public void instanciar() {
crA = new CategoryRestaurant(1l, "A");
rA = new Restaurant(1l, "A", crA);
crB = new CategoryRestaurant(2l, "B");
rB = new Restaurant(2l, "B", crB);
ranking = new Ranking();
ranking.addRestaurant(rA, 0l);
ranking.addRestaurant(rB, 0l);
}
