如何驗證服務層中的資料？

我有一個我無法解決的問題。我目前正在使用 Spring Boot 構建一個 Rest Api，并且我想在服務層內驗證我的用戶物體。我嘗試了不同的方法，但目前沒有任何結果。在我的測驗中，當它到達 create 方法時沒有拋出例外。

這是我當前的代碼：

用戶物體

    @Entity
    @Getter
    @Setter
    @NoArgsConstructor
    @AllArgsConstructor
    public class User {

    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;

    @NotBlank
    @Email(message = "User needs a valid Mail.")
    @Column(unique = true, nullable = false)
    private String mail;

    @NotBlank
    @Column(nullable = false)
    private String lastName;

    @NotBlank
    @Column(nullable = false)
    private String firstName;
    }

用戶服務

    @Service
    @RequiredArgsConstructor
    public class UserService implements IUserService {

    private final UserRepository userRepository;

    @Override
    public User create(@Valid User user) {
        user.setId(null);
        User createdUser = userRepository.save(user);

        return createdUser;
    }

我的測驗

    @SpringBootTest
    class UserServiceTest {

    @Mock
    UserRepository userRepository;

    @InjectMocks
    UserService userService;

    @Test
    void createUserFailsBecauseNoFirstName() {
        User testUser = new User("[email protected]", "LastName", "FirstName");
        when(userRepository.save(any(User.class))).thenReturn(testUser);
        testUser.setFirstName(null);
        Assertions.assertThrows(ConstraintViolationException.class,
                () -> userService.create(testUser));
    }

pom.xml

    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.6.2</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>
    <groupId>com.rest</groupId>
    <artifactId>test</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <properties>
        <java.version>11</java.version>
    </properties>
    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-jpa</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-validation</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>

        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <scope>runtime</scope>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>com.google.code.gson</groupId>
            <artifactId>gson</artifactId>
            <version>2.8.9</version>
        </dependency>
    </dependencies>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
                <configuration>
                    <excludes>
                        <exclude>
                            <groupId>org.projectlombok</groupId>
                            <artifactId>lombok</artifactId>
                        </exclude>
                    </excludes>
                </configuration>
            </plugin>
        </plugins>
    </build>

</project>

我嘗試了什么：

• @Validated 用于服務類（我遵循的教程）
• @Validated 另外在方法級別
• 洗掉 @Validated 和 @Validate 并改用 Validator ，但它不會被注入（按照那個教程）

uj5u.com熱心網友回復：

因為您@InjectMocks用于獲取UserService，所以它不是由 Spring 管理的 bean，因為@InjectMocks它是一個 Mockito 注釋，對 Spring 一無所知，也不知道如何從 Spring 容器中獲取 bean。所以你現在實際上是在測驗一個非spring UserServicebean，因此驗證不會發生。

更改為用于從 Spring 容器@Autowired中獲取bean，并將 Spring 容器中的 bean 替換為模擬版本。您還需要添加：UserService@MockBeanUserRepository@ValidatedUserService

@Service
@Validated
public class UserService implements IUserService {


}

@SpringBootTest
class UserServiceTest {

    @MockBean
    UserRepository userRepository;

    @Autowired
    UserService userService;

    @Test
    void createUserFailsBecauseNoFirstName() {
        User testUser = new User("[email protected]", "LastName", "FirstName");
        when(userRepository.save(any(User.class))).thenReturn(testUser);
        testUser.setFirstName(null);
        Assertions.assertThrows(ConstraintViolationException.class,
                () -> userService.create(testUser));
    }
}

uj5u.com熱心網友回復：

不是我正在尋找的解決方案，但即使在新的 Spring Boot 專案中，我也無法讓它與 Annotations 一起使用。所以我現在在我的服務中使用了一個驗證器，讓它驗證我的用戶。

服務

@Service
@RequiredArgsConstructor
public class UserService implements IUserService {

private final Validator validator;
private final UserRepository userRepository;

@Override
public User create(User user) {
    user.setId(null);

    // new Validation
    Set<ConstraintViolation<T>> violations = validator.validate(objectToValidate);
    if (!violations.isEmpty()) {
        StringBuilder sb = new StringBuilder();
        for (ConstraintViolation<T> constraintViolation : violations) {
            sb.append(constraintViolation.getMessage());
        }
        throw new ResourceValidationException("Error occurred: "   sb.toString());
    }


    User createdUser = userRepository.save(user);

    return createdUser;
}

測驗

@SpringBootTest
class UserServiceTest {

@Mock
UserRepository userRepository;

@Autowired
Validator validator;

UserService userService = new UserService(validator, userRepository);


@Test
void createUserFailsBecauseNoFirstName() {
    User testUser = new User("[email protected]", "LastName", "FirstName");
    when(userRepository.save(any(User.class))).thenReturn(testUser);
    testUser.setFirstName(null);
    Assertions.assertThrows(ConstraintViolationException.class,
            () -> userService.create(testUser));
}

因為我總是得到一個沒有驗證的驗證器 bean（所以我的測驗沒有拋出預期的錯誤）并且我不想創建一個 ValidatorFactory 我還添加了一個包含 ValidatorBean 的配置。

配置

@Configuration
public class AppConfig {

@Bean
public Validator defaultValidator() {

    return new LocalValidatorFactoryBean();
    }
}

標籤：

上一篇：如何為HttpRequestMessage類分配值？

下一篇：FOSRestBundle路由版本控制