Fix for NSFE when mvn test run from /diktat-rules (#1366)

### What's done: - added arguments for methods in Generation.kt
saveourtool · Jun 16, 2022 · fc84620 · fc84620
1 parent 48360cd
commit fc84620
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 7 deletions.
diff --git a/diktat-rules/pom.xml b/diktat-rules/pom.xml
@@ -151,6 +151,10 @@
                 </executions>
                 <configuration>
                     <mainClass>org.cqfn.diktat.ruleset.generation.GenerationKt</mainClass>
+                    <arguments>
+                        <argument>${project.build.sourceDirectory}</argument>
+                        <argument>${project.basedir}/src/test/resources</argument>
+                    </arguments>
                 </configuration>
             </plugin>
         </plugins>

diff --git a/diktat-rules/src/main/kotlin/org/cqfn/diktat/ruleset/generation/Generation.kt b/diktat-rules/src/main/kotlin/org/cqfn/diktat/ruleset/generation/Generation.kt
@@ -31,12 +31,15 @@ private val autoGenerationComment =
         | This document contains all enum properties from Warnings.kt as Strings.
     """.trimMargin()
 
-fun main() {
-    generateWarningNames()
-    validateYear()
+fun main(args: Array<String>) {
+    require(args.size == 2) {
+        "Only to arguments are expected"
+    }
+    generateWarningNames(args[0])
+    validateYear(args[1])
 }
 
-private fun generateWarningNames() {
+private fun generateWarningNames(sourceDirectory: String) {
     val enumValNames = Warnings.values().map { it.name }
 
     val propertyList = enumValNames.map {
@@ -59,11 +62,11 @@ private fun generateWarningNames() {
         .addFileComment(autoGenerationComment)
         .build()
 
-    kotlinFile.writeTo(Paths.get("diktat-rules/src/main/kotlin"))  // fixme: need to add it to pom
+    kotlinFile.writeTo(Paths.get(sourceDirectory))
 }
 
-private fun validateYear() {
-    val folder = Paths.get("diktat-rules/src/test/resources/test/paragraph2/header")
+private fun validateYear(testResourcesDirectory: String) {
+    val folder = Paths.get(testResourcesDirectory, "test/paragraph2/header")
     Files.list(folder)
         .filter { !it.name.contains("CopyrightDifferentYearTest.kt") }
         .forEach { file ->