Tổng quan về Mock trong Unit Test

Unit Test là một công cụ giúp cho chúng ta có thể đảm bảo đoạn code mà chúng ta viết ra đúng chính xác những gì chúng ta muốn. Trong quá trình làm việc với Unit Test, một vấn đề mà chúng ta thường gặp đó là code của chúng ta sẽ gọi tới những phương thức của các class khác, bên ngoài class, phương thức mà chúng ta đang viết. Chúng ta có thể khởi tạo những class bên ngoài đó để gọi phương thức mà chúng ta muốn, nhưng đâu ai biết được là các phương thức của những class bên ngoài đó lại gọi những phương thức của các class bên ngoài nữa và câu hỏi đặt ra là có cần thiết phải quan tâm đến những class bên ngoài đó khi chúng ta đang test phương thức của class này bởi vì những class bên ngoài đó cũng sẽ có Unit Test riêng.

Câu trả lời theo mình là không các bạn? Unit Test là chúng ta test những đoạn code nhỏ, sẽ có Unit Test để đảm bảo logic cho những external dependencies, việc chúng ta chỉ là đảm bảo cho code hiện tại chúng ta đã test mà thôi. Trong trường hợp này, một vấn đề đặt ra là nếu phương thức chúng ta đang gọi tới phương thức của class bên ngoài thì sẽ xử lý như thế nào, làm thế nào để khi code chúng ta gọi đến những phương thức đó, kết quả trả về sẽ giúp chúng ta cover được logic trong Unit Test?

Để giải quyết vấn đề này, trong Unit Test, người ta có một khái niệm gọi là Mock, dùng để giả lập hành vi của các class bên ngoài để thực hiện hành vi mà chúng ta mong muốn. Khi đó, những đoạn code gọi các phương thức bên ngoài sẽ trả về giá trị mà chúng ta mong muốn. Bởi vì code do chúng ta control nên việc giả lập này vẫn đảm bảo quality của những đoạn code mà chúng ta đang viết Unit Test. Trong bài viết này, mình sẽ hướng dẫn các bạn cách sử dụng Mock trong Unit Test với một thư viện Mocking phổ biến trong Java đó là Mockito, các bạn nhé!

Đầu tiên, mình sẽ tạo mới một Maven project để làm ví dụ:

Tổng quan về Mock trong Unit Test

với Mockito và JUnit dependencies như sau:

Bây giờ mình sẽ tạo một ứng dụng nhỏ, cho phép người dùng truyền vào 2 số, chúng ta sẽ tính tổng 2 số đó, từ kết quả này sẽ trả về true nếu kết quả lớn hơn 10, ngược lại sẽ là false. Cụ thể, code sẽ như sau:

Chúng ta có một class để tính tổng 2 số:

Một class chính của ứng dụng trong đó nó có một phương thức để kiểm tra kết quả việc tính tổng và trả về true, false dựa vào kết quả đó:

Kết quả khi chạy như sau:

Tổng quan về Mock trong Unit Test

Bây giờ, chúng ta sẽ viết Unit Test cho những class trên các bạn nhé.


Mình sẽ tạo class CalculationTest để test cho class Calculation với nội dung như sau:

Unit Test cho class Calculation đơn giản phải không các bạn?

Đến Unit Test cho class Application, phương thức main() là phương thức để chạy ứng dụng nên chúng ta không cần test. Chỉ có phương thức check() chúng ta mới cần test mà thôi.

Bình thường, mình có thể viết Unit Test cho phương thức check() này như sau:

Thế nhưng nếu xem xét kỹ, các bạn sẽ thấy đoạn code sau trong phương thức check() của class Application:

chúng ta đã test trong class CalculationTest ở trên, nên sẽ không cần phải test nữa. Cái chúng ta chỉ cần test trong phương thức check() này chỉ là đoạn code return mà thôi.

Bây giờ, chúng ta thử apply Mock cho class Calculation xem sao nhé các bạn.

Đầu tiên, các bạn cần sử dụng Mockito để tạo ra Mock object cho class Calculation trước:

Như các bạn thấy, ở đây, mình đã dùng phương thức static mock() của class Mockito trong thư viện Mockito để tạo ra Mock object cho class Calculation. Có nhiều phương thức static overload mock() trong class Mockito, tuỳ nhu cầu các bạn có thể sử dụng nhé!

Sau khi đã có Mock object của class Calculation thì các bạn có thể set mock object này để sử dụng trong class Application như sau:

Bây giờ thì chúng ta sẽ giả lập hành vi của đối tượng Calculation trong class Application bằng cách sử dụng đối tượng Mock của nó.

Giả sử bây giờ, mình muốn khi class Application gọi đến phương thức sum() của 2 số 2 và 12 trong class Calculation, mình sẽ trả về là 14 mà không cần quan tâm đến logic của phương thức sum() thì mình sẽ mock như sau:

Các bạn có thể return lại một giá trị bất kỳ nhưng cần đảm bảo nó đúng expectation của mình nhé!

Toàn bộ code testCheck() lúc này sẽ như sau:

Kết quả:

Tổng quan về Mock trong Unit Test

Nếu bây giờ bạn sửa đoạn code mock phương thức sum() để trả về là 8:

thì kết quả sẽ như sau:

Như các bạn thấy, sử dụng Mock trong Unit Test giúp chúng ta có thể quyết định kết quả trả về của những class bên ngoài mà không cần quan tâm đến việc xử lý của chúng. Chỉ cần đảm bảo cho code mà chúng ta đang Unit Test là được rồi phải không các bạn? 🙂

Chia sẽ bài viết này ...Share on Facebook
Facebook
0Tweet about this on Twitter
Twitter
Share on LinkedIn
Linkedin

One thought on “Tổng quan về Mock trong Unit Test

Add Comment