JUNIT: Sử dụng Mockito với jUnit trong Spring Unit Testing

@RunWith(MockitoJUnitRunner.class)
public class CategoryManagerTest {
    //Khai báo đối tượng thực hiện Unit test
    //@InjectMocks: yêu cầu MockitoJUnitRunner tạo đối tượng cho biến, 
   // gán các đối tượng mock cho các thuộc tính bên trong đối tượng này.
    @InjectMocks 
    private CategoryManagerImpl categoryManager;
 
    @Mock //Khai báo đối tượng sẽ được mock.
    private CategoryDao categoryDao;
    private Map<String, Category> categoryMap = new HashMap<String, Category>();
    private Map<String, Category> createCategoryMap(int length) {
        Map<String, Category> quesMap = new HashMap<String, Category>();
        for (int i = 0; i < length; i++) {
            Category question = createCategory("Category_" + i, i);
            quesMap.put("Category_" + i, question);
        }
        return quesMap;
    }
    private Category createCategory(String id, int i) {
        Category category = new Category(id, "Category_" + i);
        return category;
    }
    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
        //Tạo đối tượng Map dùng để giả lập data trong database.
        categoryMap = createCategoryMap(10);
        //Thực hiện mock method .getCategories() và giả lập xử lý lấy
        //dữ liệu từ Database thông qua categoryMap.
        when(categoryDao.getCategories()).thenAnswer(new Answer<List<Category>>() {
            public List<Category> answer(InvocationOnMock invocation)
                    throws Throwable {
                List<Category> newList = new ArrayList<Category>();
                for (int i = 0; i < categoryMap.size(); i++) {
                    newList.add(categoryMap.get("Category_" + i));
                }
                return newList;
            }
        });
    }
    @Test
    public void getCategoriesTest() {
        List<Category> list = categoryManager.getCategories();
        Assert.assertTrue(list.size() == 10);
    }
    @Test
    public void getProductsTest() {
        //Thực hiện mock method .getProducts() và giả lập xử lý trả về kết
        //quả khi lấy dữ liệu từ Database. Kết quả trả về là list<Product>.
        when(categoryManager.getProducts(Mockito.anyString()))
                            .thenAnswer(new Answer<List<Product>>() {
            public List<Product> answer(InvocationOnMock invocation)
                    throws Throwable {
                List<Product> listProduct = new ArrayList<Product>();
                for (int i = 0; i < 5; i++) {
                    Product item = new Product("Product_" + i, "Product_" + i, 
                                                (String)invocation.getArguments()[0]);
                    listProduct.add(item);
                }
                return listProduct;
            }
        });
        //Thực hiện test method categoryManager.getProducts(idTest)
        String idTest = "Category_1";
        List<Product> listProduct = categoryManager.getProducts(idTest);
        //Test kết quả sau khi chạy method cần test.
        //assertTrue là method dùng để  kiểm tra thông tin sau khi chạy 
        //testcase có đúng hay không.
        Assert.assertTrue(listProduct.size() == 5);
    }
 
    @Test
    public void getProductsTest_NotFound() {
        //Thực hiện mock method .getProducts() và giả lập xử lý trả về một                
        //Exception.    
        when(categoryManager.getProducts(Mockito.anyString()))
                            .thenAnswer(new Answer<List<Product>>() {
            public List<Product> answer(InvocationOnMock invocation)
                    throws Throwable {
                throw new Exception("NotFound");
            }
        });
        try{
            //Chạy testcase
                String idTest = "Category_999";
                List<Product> listProduct = categoryManager.getProducts(idTest);
                Assert.assertNull(listProduct);
        }catch(Exception e){
            //Kiểm tra xem thông điệp lỗi có đúng không.
            Assert.assertEquals("NotFound",e.getMessage());
        }
    }
}