spring-boot-realworld-examp.../src/test/java/io/spring/api/TestWithCurrentUser.java
2017-08-15 10:47:18 +08:00

40 lines
1.2 KiB
Java

package io.spring.api;
import io.spring.application.JwtService;
import io.spring.application.user.UserData;
import io.spring.application.user.UserReadService;
import io.spring.core.user.User;
import io.spring.core.user.UserRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.mock.mockito.MockBean;
import java.util.Optional;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.when;
class TestWithCurrentUser {
@MockBean
protected UserRepository userRepository;
@MockBean
protected UserReadService userReadService;
protected User user;
protected UserData userData;
protected String token;
@Autowired
private JwtService jwtService;
protected void userFixture(String email, String username, String defaultAvatar) {
user = new User(email, username, "123", "", defaultAvatar);
when(userRepository.findByUsername(eq(username))).thenReturn(Optional.of(user));
userData = new UserData(user.getId(), email, username, "", defaultAvatar);
when(userReadService.findByUsername(eq(username))).thenReturn(userData);
token = jwtService.toToken(userData);
}
}