Spring Boot-如何通过一个实现BeforeAllCallback的自定义扩展类来设置或重写application.yml中定义的属性？

Question

我有一个扩展类，它实现BeforeAllCallback来设置类似这样的随机端口。

import org.junit.jupiter.api.extension.ExtensionContext
import java.net.ServerSocket

class CustomExtension : org.junit.jupiter.api.extension.BeforeAllCallback {
    var port: Int? = null

    override fun beforeAll(context: ExtensionContext?) {
        port = getRandomPort()
    }
    
    private fun getRandomPort(): Int {
        ServerSocket(0).use { 
            serverSocket -> return serverSocket.localPort 
        }
    }

}

另外，我在src/test/resources/application.yml中定义了一个属性，如下所示。

app:
  random-port: 8765

import org.springframework.boot.context.properties.ConfigurationProperties
import org.springframework.boot.context.properties.ConstructorBinding

@ConstructorBinding
@ConfigurationProperties(prefix = "app")
data class PortProperty(
    val randomPort: Int
)

import org.junit.jupiter.api.extension.ExtendWith
import org.springframework.boot.test.context.SpringBootTest

@SpringBootTest
@ExtendWith(CustomExtension::class)
class RandomPortTest {
}

该用例实际上是使用测试类的AWS DynamoDBLocal库启动本地dynamo db。

Answer 1

可以在BeforeAll方法中设置app.random-port系统属性。由于系统属性比属性文件具有更高的优先级，application.yml的端口值将被重写：

kotlin prettyprint-override">override fun beforeAll(context: ExtensionContext?) {
    port = getRandomPort()
    System.setProperty("app.random-port", port?.toString())
}