{"id":326884,"date":"2023-09-09T18:37:18","date_gmt":"2023-09-09T18:37:18","guid":{"rendered":"https:\/\/www.baeldung.com\/?p=165449"},"modified":"2023-09-09T18:37:18","modified_gmt":"2023-09-09T18:37:18","slug":"mock-same-method-with-different-parameters","status":"publish","type":"post","link":"https:\/\/itteacheritfreelance.hk\/wordpress\/index.php\/2023\/09\/09\/mock-same-method-with-different-parameters\/","title":{"rendered":"Mock Same Method with Different Parameters"},"content":{"rendered":"

<\/p>\n

\"\"<\/p>\n

1. Overview<\/h2>\n
<\/div>\n

When mocking<\/a> a method in Java, it can be useful to receive different responses based on the parameters passed in. In this article, we’ll look at different ways to achieve that goal depending on the complexity of our requirements.<\/p>\n

2. Setup<\/h2>\n
<\/div>\n

First, let’s create an example service which we want to mock<\/strong>:<\/p>\n

class ExampleService {\r\n    int getValue(int arg){\r\n        return 1;\r\n    }\r\n}<\/code><\/pre>\n
We’ve got a very simple service with a single method. The method has a single int<\/em> as an argument and returns an int<\/em>. Note that the argument and the return value have no relation, so by default, it’ll always return 1.<\/p>\n
3. Limitations of Consecutive Stubbing<\/h2>\n
<\/div>\n
Let’s look at consecutive stubbing and what we can and can’t do with it. We can use consecutive stubbing to get different arguments from our mock in order regardless of the input we provide.<\/strong> This obviously lacks control over matching particular inputs to desired outputs but is useful in many cases. To do this we pass the method we want to stub to\u00a0when()<\/em><\/a>. We then chain a call to thenReturn()<\/em> providing the responses in the order we want them:<\/p>\n
@Test\r\nvoid givenAMethod_whenUsingConsecutiveStubbing_thenExpectResultsInOrder(){\r\n    when(exampleService.getValue(anyInt())).thenReturn(9, 18, 27);\r\n    assertEquals(9, exampleService.getValue(1));\r\n    assertEquals(18, exampleService.getValue(1));\r\n    assertEquals(27, exampleService.getValue(1));\r\n    assertEquals(27, exampleService.getValue(1));\r\n}<\/code><\/pre>\n
We can see from the assertions that despite always passing in 1 as the parameter, we received back the expected values in order<\/strong>. Once all the values have been returned, all future calls will return the final value, as seen in the fourth call in our test.<\/p>\n
4. Stubbing Calls for Different Parameters<\/h2>\n
<\/div>\n
We can extend our use of when()<\/em> and thenReturn()<\/em> to return different values for different parameters:\u00a0<\/strong><\/p>\n
@Test\r\nvoid givenAMethod_whenStubbingForMultipleArguments_thenExpectDifferentResults() {\r\n    when(exampleService.getValue(10)).thenReturn(100);\r\n    when(exampleService.getValue(20)).thenReturn(200);\r\n    when(exampleService.getValue(30)).thenReturn(300);\r\n    assertEquals(100, exampleService.getValue(10));\r\n    assertEquals(200, exampleService.getValue(20));\r\n    assertEquals(300, exampleService.getValue(30));\r\n}<\/code><\/pre>\n
The argument for when()<\/em> is the method we want to stub, along with the value we want to specify a response for. By chaining the call to when()<\/em> with\u00a0thenReturn(),<\/em> we’ve instructed the mock to return the requested value when the correct argument is received. We’re free to apply as many of these as we want to our mock to handle a range of inputs. We’ll receive the requested return value every time the expected input value is provided.<\/p>\n
5. Using thenAnswer()<\/em><\/h2>\n
<\/div>\n
A more complex option, offering maximum control, is to use\u00a0thenAnswer()<\/em><\/a>.<\/strong> This allows us to take the arguments, perform any computation on them we want, and then return a value that’ll be outputted when interacting with the mock:<\/p>\n
@Test\r\nvoid givenAMethod_whenUsingThenAnswer_thenExpectDifferentResults() {\r\n    when(exampleService.getValue(anyInt())).thenAnswer(invocation -> {\r\n        int argument = (int) invocation.getArguments()[0];\r\n        int result;\r\n        switch (argument) {\r\n        case 25:\r\n            result = 125;\r\n            break;\r\n        case 50:\r\n            result = 150;\r\n            break;\r\n        case 75:\r\n            result = 175;\r\n            break;\r\n        default:\r\n            result = 0;\r\n        }\r\n        return result;\r\n    });\r\n    assertEquals(125, exampleService.getValue(25));\r\n    assertEquals(150, exampleService.getValue(50));\r\n    assertEquals(175, exampleService.getValue(75));\r\n}<\/code><\/pre>\n
Above, we’ve grabbed the arguments using getArguments()<\/em> on the provided invocation object. We’ve assumed a single int<\/em> argument here, but we could’ve catered for several different types. We also could’ve checked that there was at least one argument and the cast to an int<\/em> was successful. To demonstrate the capabilities, we’ve used a switch<\/em><\/a> statement to return different values based on the input. At the bottom, we can see from the assertions that our mocked service returns the results of the switch<\/em> statement.<\/p>\n
This option allows us to handle an unlimited amount of inputs with a single call to\u00a0when()<\/em>. The sacrifice is the readability and maintainability of the tests.<\/p>\n
6. Conclusion<\/h2>\n
<\/div>\n
In this tutorial, we’ve seen three ways of configuring a mocked method to return different values. We looked at consecutive stubbing and saw that it’s useful for returning known values in order for any input but is very limited beyond that. Using when()<\/em> combined with thenReturn()<\/em> for each potential input offers a simple solution with improved control.\u00a0 Alternatively, we can use thenAnswer()<\/em> for maximum control over the relationship between the given input and the expected output. All three are useful depending on the requirements under test.<\/p>\n
As always, the full code for the examples is available over on GitHub<\/a>.<\/p>\n
\"\"<\/p>\n
<\/a> <\/a> <\/a> <\/a> <\/a> <\/a> <\/a> <\/div>\n\n
\\\"IT\u96fb\u8166\u88dc\u7fd2
\u7acb\u523b\u8a3b\u518a\u53ca\u5831\u540d\u96fb\u8166\u88dc\u7fd2\u8ab2\u7a0b\u5427!<\/figcaption><\/figure>\r\n<\/br>Find A Teacher Form:\r\n<\/br>https:\/\/docs.google.com\/forms\/d\/1vREBnX5n262umf4wU5U2pyTwvk9O-JrAgblA-wH9GFQ\/viewform?edit_requested=true#responses\r\n<\/br><\/br>Email:\r\n<\/br>public1989two@gmail.com






\r\nwww.itsec.hk<\/a>
\r\nwww.itsec.vip<\/a>
\r\nwww.itseceu.uk<\/a>
<\/p>","protected":false},"excerpt":{"rendered":"
\"\"<\/p>\n
Learn how to mock a method in Java to receive different responses based on the parameters passed in.<\/p>\n
<\/a>\u00a0<\/a>\u00a0<\/a>\u00a0<\/a>\u00a0<\/a>\u00a0<\/a>\u00a0<\/a>\u00a0<\/div>\n<\/div>","protected":false},"author":1977,"featured_media":0,"comment_status":"open","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"slim_seo":{"title":"Mock Same Method with Different Parameters - ITTeacherITFreelance.hk","description":"Learn how to mock a method in Java to receive different responses based on the parameters passed in. \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0 \u00a0"},"footnotes":""},"categories":[6,1307],"tags":[],"_links":{"self":[{"href":"https:\/\/itteacheritfreelance.hk\/wordpress\/index.php\/wp-json\/wp\/v2\/posts\/326884"}],"collection":[{"href":"https:\/\/itteacheritfreelance.hk\/wordpress\/index.php\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/itteacheritfreelance.hk\/wordpress\/index.php\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/itteacheritfreelance.hk\/wordpress\/index.php\/wp-json\/wp\/v2\/users\/1977"}],"replies":[{"embeddable":true,"href":"https:\/\/itteacheritfreelance.hk\/wordpress\/index.php\/wp-json\/wp\/v2\/comments?post=326884"}],"version-history":[{"count":1,"href":"https:\/\/itteacheritfreelance.hk\/wordpress\/index.php\/wp-json\/wp\/v2\/posts\/326884\/revisions"}],"predecessor-version":[{"id":326885,"href":"https:\/\/itteacheritfreelance.hk\/wordpress\/index.php\/wp-json\/wp\/v2\/posts\/326884\/revisions\/326885"}],"wp:attachment":[{"href":"https:\/\/itteacheritfreelance.hk\/wordpress\/index.php\/wp-json\/wp\/v2\/media?parent=326884"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/itteacheritfreelance.hk\/wordpress\/index.php\/wp-json\/wp\/v2\/categories?post=326884"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/itteacheritfreelance.hk\/wordpress\/index.php\/wp-json\/wp\/v2\/tags?post=326884"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}