4
4
.
.
2
2
.
.
1
1
1
1
v
v
e
e
r
r
i
i
f
f
y
y
(
(
)
)
-
-
A
A
r
r
g
g
u
u
m
m
e
e
n
n
t
t
C
C
a
a
p
p
t
t
o
o
r
r
I
I
n
n
f
f
o
o
[
[
G
G
]
]
ArgumentCaptor is used in combination with verify() to capture arguments with which Mocked Method was called.
Application Schema [Result]
Spring Boot Starters
GROUP
DEPENDENCY
DESCRIPTION
Web
Spring Web
Enables: Controller Annotations, Tomcat Server
Syntax
ArgumentCaptor<Integer> integerArgumentCaptor = ArgumentCaptor.forClass(Integer.class);
verify(personRepositoryMock, times(2)).getPersonById(integerArgumentCaptor.capture());
MyController
http://localhost:8080/GetPerson
getPerson()
PersonRepository
Person
Browser
Tomcat
MyControllerTest
@Mock
PersonRepository
P
P
r
r
o
o
c
c
e
e
d
d
u
u
r
r
e
e
Create Project: springboot_mockito_verify_argumentcaptor (add Spring Boot Starters from the table)
Create Package: entities (inside main package)
– Create Class: Person.java (inside entities package)
Create Package: respositories (inside main package)
– Create Class: PersonRepository.java (inside entities package)
Create Package: controllers (inside main package)
– Create Class: MyController.java (inside controllers package)
Create Package: controllers (inside test package)
– Create Test Class: MyControllerTest.java (inside entities package)
Person.java
package com.ivoronline.springboot_mockito_verify_argumentcaptor.entities;
public class Person {
//PROPERTIES
public Integer id;
public String name;
public Integer age;
//CONSTRUCTOR
public Person(Integer id, String name, Integer age) {
this.id = id;
this.name = name;
this.age = age;
}
}
PersonRepository.java
package com.ivoronline.springboot_mockito_verify_argumentcaptor.respositories;
import com.ivoronline.springboot_mockito_verify_argumentcaptor.entities.Person;
import org.springframework.stereotype.Component;
import java.util.HashMap;
import java.util.Map;
@Component
public class PersonRepository {
//PROPERTY
Map<Integer, Person> persons = new HashMap();
//CONSTRUCTOR
public PersonRepository() {
persons.put(1, new Person(1, "John", 20));
persons.put(2, new Person(2, "Bill", 30));
persons.put(3, new Person(3, "Jack", 40));
}
//GET PERSON BY ID
public Person getPersonById(Integer id) {
return persons.get(id);
}
}
MyController.java
package com.ivoronline.springboot_mockito_verify_argumentcaptor.controllers;
import com.ivoronline.springboot_mockito_verify_argumentcaptor.entities.Person;
import com.ivoronline.springboot_mockito_verify_argumentcaptor.respositories.PersonRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
@Controller
public class MyController {
@Autowired PersonRepository personRepository;
@ResponseBody
@RequestMapping("/GetPerson")
public String getPerson(@RequestParam Integer id) {
//GET PERSON
Person person = personRepository.getPersonById(id);
String name = person.name;
Integer age = person.age;
//RETURN SOMETHING
return name + " is " + age + " years old";
}
}
MyControllerTest.java
package com.ivoronline.springboot_mockito_argumentcaptor.controllers;
import com.ivoronline.springboot_mockito_argumentcaptor.entities.Person;
import com.ivoronline.springboot_mockito_argumentcaptor.respositories.PersonRepository;
import org.junit.jupiter.api.Test;
import org.mockito.ArgumentCaptor;
import org.mockito.InjectMocks;
import org.springframework.boot.test.context.SpringBootTest;
import org.mockito.Mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
@SpringBootTest
class MyControllerTest {
//MOCK DEPENDENCY CLASS
@Mock PersonRepository personRepositoryMock;
//INJECT MOCKS (where @autowired is used)
@InjectMocks MyController myController;
//ENDPOINT
@Test
void getPerson() {
//CREATE CAPTOR THAT CAPTURES INTEGER ARGUMENTS
//ArgumentCaptor<Integer> integerArgumentCaptor = ArgumentCaptor.forClass(Integer.class);
//MOCK REPOSITORY METHOD
when(personRepositoryMock.getPersonById(1)).thenReturn(new Person(1, "Susan", 50));
when(personRepositoryMock.getPersonById(2)).thenReturn(new Person(1, "Susan", 50));
//CALL REPOSITORY METHOD
myController.getPerson(1);
myController.getPerson(2);
//CAPTURE INTEGER ARGUMENT (from last call)
ArgumentCaptor<Integer> integerArgumentCaptor = ArgumentCaptor.forClass(Integer.class);
verify(personRepositoryMock, times(2)).getPersonById(integerArgumentCaptor.capture());
//GET LAST CAPTURED ARGUMENT
Integer argument = integerArgumentCaptor.getValue();
//DISPLAY CAPTURED ARGUMENT
System.out.println(argument);
}
}