@@ -21,10 +21,8 @@ import org.assertj.assertions.generator.description.converter.ClassToClassDescri
2121import org.assertj.assertions.generator.util.ClassUtil
2222import org.assertj.generator.gradle.internal.tasks.AssertionsGeneratorReport
2323import org.assertj.generator.gradle.tasks.config.AssertJGeneratorExtension
24- import org.gradle.api.file.ConfigurableFileCollection
25- import org.gradle.api.file.DirectoryProperty
26- import org.gradle.api.file.FileTree
27- import org.gradle.api.file.FileVisitDetails
24+ import org.assertj.generator.gradle.tasks.config.SerializedTemplate
25+ import org.gradle.api.file.*
2826import org.gradle.api.logging.Logging
2927import org.gradle.api.model.ObjectFactory
3028import org.gradle.api.provider.ListProperty
@@ -49,10 +47,10 @@ class AssertJGenerationTask extends SourceTask {
4947
5048 @InputFiles
5149 @Classpath
52- final ListProperty< File > templateFiles
50+ final FileCollection templateFiles
5351
5452 @Input
55- final ListProperty<String > templateStrings
53+ final ListProperty<SerializedTemplate > templates
5654
5755 @OutputDirectory
5856 final DirectoryProperty outputDir
@@ -72,7 +70,7 @@ class AssertJGenerationTask extends SourceTask {
7270
7371 this . outputDir = assertJOptions. outputDir
7472 this . templateFiles = assertJOptions. templates. templateFiles
75- this . templateStrings = assertJOptions. templates. templateStrings
73+ this . templates = assertJOptions. templates. templates
7674 }
7775
7876 @TaskAction
@@ -92,7 +90,7 @@ class AssertJGenerationTask extends SourceTask {
9290 } else if (sourceFiles. contains(change. file)) {
9391 // source file changed
9492 classesToGenerate + = change. file
95- } else if (templateFiles. get() . contains(change. file)) {
93+ } else if (templateFiles. contains(change. file)) {
9694 fullRegenRequired = true
9795 }
9896 }
@@ -139,8 +137,10 @@ class AssertJGenerationTask extends SourceTask {
139137
140138 Path absOutputDir = project. rootDir. toPath(). resolve(this . outputDir. getAsFile(). get(). toPath())
141139 report. setDirectoryPathWhereAssertionFilesAreGenerated(absOutputDir. toFile())
142- assertJOptions. templates. getTemplates(report). each {
143- generator. register(it)
140+
141+ def templates = assertJOptions. templates. templates. get(). collect { it. maybeLoadTemplate() }. findAll()
142+ for (template in templates) {
143+ generator. register(template)
144144 }
145145
146146 try {
0 commit comments