У меня есть @Query в моем репозитории, в который я возвращаю данные в мой контроллер на основе SQL запроса, как бы я это высмеял?
Вот мой репозиторий
package movieweb.movies.repository;
import movieweb.movies.models.UserMovies;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.CrudRepository;
import org.springframework.stereotype.Repository;
import java.util.List;
@Repository
public interface UserMoviesRepository extends CrudRepository<UserMovies, Integer> {
@Query(value = "select * from movies, user_movies where movies.movie_id = user_movies.movie_id and uname = ?1", nativeQuery = true)
List<UserMovies> findByUname(String uname);
}
Вот мой контроллер
package movieweb.movies.controllers;
import movieweb.movies.models.Movies;
import movieweb.movies.models.UserMovies;
import movieweb.movies.repository.UserMoviesRepository;
import org.apache.catalina.User;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.jpa.repository.Query;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import java.util.List;
@RestController
public class UserMoviesController {
@Autowired
private UserMoviesRepository umRepository;
@CrossOrigin
@PostMapping(path = "/newUserMovie")
public ResponseEntity<UserMovies> addNewUserMovie(@RequestBody UserMovies data){
umRepository.save(data);
return new ResponseEntity<UserMovies>(data, HttpStatus.CREATED);
}
@CrossOrigin
@GetMapping(path="/getUserMovies")
public List<UserMovies> getUsersMovies(){
return (List<UserMovies>) umRepository.findAll();
}
@CrossOrigin
@GetMapping(path = "/getUserMovies/{uname}")
public List<UserMovies> getUserMovies(@PathVariable String uname){
return umRepository.findByUname(uname);
}
@CrossOrigin
@DeleteMapping(path ="/deleteUserMovies/{id}")
ResponseEntity deleteUserMovie(@PathVariable Integer id) {
umRepository.deleteById(id);
return new ResponseEntity(HttpStatus.ACCEPTED);
}
@CrossOrigin
@PutMapping("/usermovie/update/{id}")
public ResponseEntity<UserMovies> updateMovie(@RequestBody UserMovies updateMovie, @PathVariable Integer id) {
return umRepository.findById(id)
.map(userMovies -> {
userMovies.setMovieId(updateMovie.getMovieId());
userMovies.setUname(updateMovie.getUname());
umRepository.save(userMovies);
return new ResponseEntity<UserMovies>(userMovies, HttpStatus.OK);
})
.orElseGet(() -> {
updateMovie.setMovieId(id);
umRepository.save(updateMovie);
return new ResponseEntity<UserMovies>(updateMovie, HttpStatus.OK);
});
}
}
А вот что я пытаюсь сделать в своем тесте
@Test
void getUserMoviesPerUser() throws Exception{
ArrayList<UserMovies> userMovies = new ArrayList<>();
userMovies.add(new UserMovies(1, "jamie", 1));
userMovies.add(new UserMovies(2, "joe", 1));
userMovies.add(new UserMovies(3, "jamie", 2));
userMovies.add(new UserMovies(4, "joe", 2));
when(userMoviesRepository.findByUname("jamie")).thenReturn(userMovies);
mockMvc.perform(get("/getUserMovies/{uname}", "jamie"))
.andDo(print())
.andExpect(jsonPath("$", hasSize(2)))
.andExpect(jsonPath("$[0].userMovieID", is(1)))
.andExpect(jsonPath("$[0].uname", is("jamie")))
.andExpect(jsonPath("$[0].movieId", is(1)))
.andExpect(jsonPath("$[1].userMovieID", is(3)))
.andExpect(jsonPath("$[1].uname", is("jamie")))
.andExpect(jsonPath("$[1].movieId", is(1)));
Mockito.verify(userMoviesRepository, times(1)).findAll();
}
Проблема, с которой я сталкиваюсь, заключается в том, что когда я запускаю тест, описанный выше, я не возвращая только 2 записи, как я ожидал, я возвращаюсь 4. Это имеет смысл в том факте, что я не могу зависеть от SQL для запуска тестов. Есть ли лучший способ написать мой контроллер для получения данных, основанных на movie_id?