1212import org .springframework .mock .web .MockMultipartFile ;
1313import org .springframework .test .context .junit4 .SpringRunner ;
1414import org .springframework .test .web .servlet .MockMvc ;
15+ import ru .rustavil .fuel_consumption .domain .Driver ;
16+ import ru .rustavil .fuel_consumption .domain .FuelConsumption ;
17+ import ru .rustavil .fuel_consumption .domain .FuelPurchase ;
1518import ru .rustavil .fuel_consumption .domain .FuelType ;
1619import ru .rustavil .fuel_consumption .domain .exceptions .ResourceNotFoundException ;
1720import ru .rustavil .fuel_consumption .rest .dto .FuelConsumptionRequestDto ;
2225import java .time .LocalDate ;
2326import java .time .format .DateTimeFormatter ;
2427import java .util .Collections ;
28+ import java .util .UUID ;
2529
2630import static org .assertj .core .api .Assertions .assertThat ;
2731import static org .mockito .ArgumentMatchers .any ;
28- import static org .mockito .Mockito .doNothing ;
29- import static org .mockito .Mockito .doThrow ;
32+ import static org .mockito .Mockito .*;
3033import static org .springframework .test .web .servlet .request .MockMvcRequestBuilders .multipart ;
3134import static org .springframework .test .web .servlet .request .MockMvcRequestBuilders .post ;
3235import static org .springframework .test .web .servlet .result .MockMvcResultMatchers .jsonPath ;
@@ -53,7 +56,14 @@ public void setUp() throws Exception {
5356
5457 @ Test
5558 public void whenRegisteredValidFuelConsumptionThenExpected () throws Exception {
56- doNothing ().when (register ).registerPurchase (any (FuelPurchaseRequestDto .class ));
59+ when (register .registerPurchase (any (FuelPurchaseRequestDto .class ))).
60+ thenReturn (
61+ new FuelPurchase (new FuelConsumption (
62+ UUID .randomUUID (),
63+ Driver .builder ().identifier (11111L ).build (),
64+ FuelType .TYPE_95 ,
65+ 100.0 ,
66+ BigDecimal .valueOf (1000.0 ))));
5767
5868 FuelConsumptionRequestDto fuelConsumptionRequestDto = new FuelConsumptionRequestDto (
5969 LocalDate .now (), 11111L , FuelType .TYPE_95 , 100.0 , BigDecimal .valueOf (200.0 )
@@ -125,7 +135,14 @@ public void whenRegisteredValidFuelConsumptionAndHappenedInternalErrorThenExpect
125135
126136 @ Test
127137 public void whenRegisteredValidFuelConsumptionBulkFileThenExpected () throws Exception {
128- doNothing ().when (register ).registerPurchase (any (FuelPurchaseRequestDto .class ));
138+ when (register .registerPurchase (any (FuelPurchaseRequestDto .class ))).
139+ thenReturn (
140+ new FuelPurchase (new FuelConsumption (
141+ UUID .randomUUID (),
142+ Driver .builder ().identifier (11111L ).build (),
143+ FuelType .TYPE_95 ,
144+ 100.0 ,
145+ BigDecimal .valueOf (1000.0 ))));
129146
130147 FuelConsumptionRequestDto fuelConsumptionRequestDto = new FuelConsumptionRequestDto (
131148 LocalDate .now (), 11111L , FuelType .TYPE_95 , 100.0 , BigDecimal .valueOf (200.0 )
0 commit comments