Mockito - Đặt một phương thức của đối tượng được trả về bằng phương thức đối tượng giả

Question

Giả sử tôi có một đối tượng giả, và tôi không muốn khai phá bất kỳ phương thức nào, nhưng tôi muốn tạo phương thức của đối tượng nó trở lại. Ví dụ,

when(mockObject.method1()).thenReturn(returnValue) 

là cách nó thường được thực hiện, nhưng tôi đang tìm,

when(mockObject.method1().method2()).thenReturn(returnValue) 

Là có thể? Tôi nhận được một NullPointerException nếu tôi làm điều đó. Hiện tại tôi đã khai báo phương thức đầu tiên để trả về một đối tượng giả, và sau đó sử dụng đối tượng giả đó trả về, khai báo phương thức thứ hai. Tuy nhiên, những đối tượng giả tạm thời này vô dụng đối với tôi và sau khi kết hợp nhiều phương thức với nhau, điều đó dẫn đến rất nhiều đối tượng giả tạo vô ích.

EDIT: Trên thực tế, có thể chuỗi hoạt động, nhưng đối tượng của tôi đang gây ra lỗi NPE. Mã này (dòng đầu tiên) đang gây ra một NPE:

when(graphDb.index().getNodeAutoIndexer()).thenReturn(nodeAutoIndexer); 
when(graphDb.index().getRelationshipAutoIndexer()).thenReturn(relAutoIndexer); 

Nhưng mã này hoạt động:

IndexManager indexManager = mock(IndexManager.class); 
when(graphDb.index()).thenReturn(indexManager); 
when(indexManager.getNodeAutoIndexer()).thenReturn(nodeAutoIndexer); 
when(graphDb.index().getRelationshipAutoIndexer()).thenReturn(relAutoIndexer); 

Vì vậy chaining được kết không làm việc cho getNodeAutoIndexer() trả về một đối tượng AutoIndexer trong khi nó làm việc cho getRelationshipAutoIndexer() trả về một RelationshipAutoIndexer. Cả hai giá trị trả về đều được mô phỏng như sau:

nodeAutoIndexer = (AutoIndexer<Node>) mock(AutoIndexer.class); 
relAutoIndexer = mock(RelationshipAutoIndexer.class); 

Vậy điều gì có thể gây ra sự cố?

Answer 1

Không có vấn đề gì cả.

Hãy kiểm tra những 4 dòng mã:

IndexManager indexManager = mock(IndexManager.class); 
when(graphDb.index()).thenReturn(indexManager); 
when(indexManager.getNodeAutoIndexer()).thenReturn(nodeAutoIndexer); 
when(graphDb.index().getRelationshipAutoIndexer()).thenReturn(relAutoIndexer); 

Dòng đầu tiên tạo ra một indexManager giả.

Điều thứ hai yêu cầu biểu đồ giả lập trả về indexManager (mô phỏng được tạo ở dòng đầu tiên) khi phương thức chỉ mục được gọi.

Phương thức thứ ba là chỉ số mô phỏngManager (được tạo ở dòng đầu tiên) để trả về nodeAutoIndexer khi phương thức getNodeAutoIndexer của nó được gọi.

Và dòng cuối cùng gọi graphDb.index(), trả về chỉ số giả lập (bạn đã yêu cầu nó làm điều đó ở dòng hai) và hỏi indexManager này (đó là mô hình bạn tạo ở dòng đầu tiên) để trả về relAutoIndexer khi phương thức getRelationshipAutoIndexer của nó được gọi.

Dòng cuối cùng chỉ hoạt động vì bạn đã nói với biểu đồ giảDB gì cần trả về khi phương thức chỉ mục của nó được gọi. Nếu bạn chưa làm điều này trước đây, phương thức graphDb.index() giả định sẽ trả về null và bạn sẽ có một NPE.