This message was deleted.

With my limited gradle knowledge, it seems I can sort of config the Test task as such to make it work for me:

tasks.withType<Test>().configureEach {
    classpath += configurations.runtimeClasspath.get()
    testClassesDirs = layout.buildDirectory.files("classes/java/main")
}