Bring back test class generation (#403)

Nikita Vlaev · nikitavlaev · commit 43693fb2b541 · 2022-07-04T21:07:54.000+03:00
diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/TestGenerator.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/TestGenerator.kt
@@ -151,7 +151,9 @@ object TestGenerator {
                 }
             }.get()
 
-            return if (FileModificationService.getInstance().preparePsiElementForWrite(testClass)) testClass else null
+            testClass?.let {
+                return if (FileModificationService.getInstance().preparePsiElementForWrite(it)) it else null
+            }
         }
 
         val fileTemplate = FileTemplateManager.getInstance(testDirectory.project).getInternalTemplate(

Original file line number	Diff line number	Diff line change
`@@ -151,7 +151,9 @@ object TestGenerator {`
`151`	`151`	`}`
`152`	`152`	`}.get()`
`153`	`153`
`154`		`- return if (FileModificationService.getInstance().preparePsiElementForWrite(testClass)) testClass else null`
	`154`	`+ testClass?.let {`
	`155`	`+ return if (FileModificationService.getInstance().preparePsiElementForWrite(it)) it else null`
	`156`	`+ }`
`155`	`157`	`}`
`156`	`158`
`157`	`159`	`val fileTemplate = FileTemplateManager.getInstance(testDirectory.project).getInternalTemplate(`