在Spring中以测试模式运行时如何禁用方法?

时间:2018-12-11 09:32:06

标签: spring rabbitmq amqp

我有一个看起来像这样的方法:

@PostMapping(path = ["/signup"],
        consumes = [(MediaType.APPLICATION_JSON_UTF8_VALUE)])
fun signUp(@RequestBody dto: RegistrationDto)
        : ResponseEntity<Void> {

    val userId : String = dto.userInfo!!.username!!
    val password : String = dto.password!!

    val registered = if(!dto.secretPassword.isNullOrBlank() && dto.secretPassword.equals(adminCode)) {
        authService.createUser(userId, password, setOf("ADMIN"))
    } else {
        authService.createUser(userId, password, setOf("USER"))
    }

    if (!registered) {
        return ResponseEntity.status(400).build()
    }

    val userDetails = userDetailsService.loadUserByUsername(userId)
    val token = UsernamePasswordAuthenticationToken(userDetails, password, userDetails.authorities)

    authenticationManager.authenticate(token)

    if (token.isAuthenticated) {
        SecurityContextHolder.getContext().authentication = token
    }

    /**
     * AMQP
     */
    amqpService.send(dto.userInfo!!, "USER-REGISTRATION")

    return ResponseEntity.status(204).build()
}

如果您发现我有一种方法amqpService.send(dto.userInfo!!, "USER-REGISTRATION,当我在“开发”模式下运行时如何禁用该方法?

我想在测试模式下运行时禁用RabbiqMQ,以便不调用此方法?

感谢

1 个答案:

答案 0 :(得分:0)

在测试模式下,您可以在模拟上替换amqpService:

@Configuration
public class AmqpConfig {

   @Profile("test")
   @Bean 
   public AmqpService amqpService(){
     return mock(AmqpService.class);
   }   
}

或者您可以在测试类中立即使用@MockBean来在测试中的模拟中替换此bean。

因此,您运行模拟方法而不是真实对象。