added "Assert is not null" action.

Author: Henady Zakalusky

src/main/java/com/bruce/intellijplugin/generatesetter/CommonConstants.java

@@ -24,6 +24,7 @@ public class CommonConstants {
     public static final String GENERATE_ACCESSORS_METHOD = "Generate accessors chain call";
     public static final String GENERATE_SETTER_METHOD_NO_DEFAULT_VALUE = "Generate all setter no default value";
     public static final String ASSERT_ALL_PROPS = "Assert all getters";
+    public static final String ASSERT_NOT_NULL = "Assert is not null";
     public static final String GENERATE_CONVERTER_FROM_METHOD = "Generate setter getter converter";
     public static final String BUILDER_CONVERTER_FROM_METHOD = "Generate builder getter converter";
     public static final String BUILDER_METHOD_NAME = "builder";

src/main/java/com/bruce/intellijplugin/generatesetter/actions/AssertAllGetterAction.java

@@ -80,8 +80,8 @@ enum TestEngine {ASSERT, JUNIT4, JUNIT5, TESTNG, ASSERTJ}
             .put(ASSERTJ, "assertThat")
             .build();
 
-    private Project project;
-    private PsiFile containingFile;
+    protected Project project;
+    protected PsiFile containingFile;
 
     public AssertAllGetterAction() {
         setGenerateAllHandler(new GenerateAllAssertsHandlerAdapter(true));
@@ -118,14 +118,19 @@ public boolean isAvailable(@NotNull Project project, Editor editor, @NotNull Psi
 
     class GenerateAllAssertsHandlerAdapter extends GenerateAllHandlerAdapter {
         private final boolean generateWithDefaultValues;
-        private final Set<TestEngine> currentFileImportedEngines = new HashSet<>();
-        private TestEngine currentFileTestEngine = TestEngine.ASSERT;
+
+        protected final Set<TestEngine> currentFileImportedEngines = new HashSet<>();
+        protected TestEngine currentFileTestEngine = TestEngine.ASSERT;
 
         public GenerateAllAssertsHandlerAdapter(boolean generateWithDefaultValues) {
             this.generateWithDefaultValues = generateWithDefaultValues;
         }
 
-        private TestEngine detectCurrentTestEngine(Project project, PsiFile containingFile) {
+        public void detectCurrentTestEngine(Project project, PsiFile containingFile) {
+            currentFileTestEngine = detectCurrentTestEngineInternal(project, containingFile);
+        }
+
+        private TestEngine detectCurrentTestEngineInternal(Project project, PsiFile containingFile) {
 
             if (containingFile instanceof PsiJavaFile) {
                 PsiJavaFile javaFile = (PsiJavaFile) containingFile;
@@ -246,7 +251,7 @@ public String formatLine(String line) {
                 value = "";
             }
 
-            currentFileTestEngine = detectCurrentTestEngine(project, containingFile);
+            detectCurrentTestEngine(project, containingFile);
 
             switch (currentFileTestEngine) {
                 case JUNIT4:

src/main/java/com/bruce/intellijplugin/generatesetter/actions/AssertNotNullAction.java

@@ -0,0 +1,67 @@
+package com.bruce.intellijplugin.generatesetter.actions;
+
+import com.bruce.intellijplugin.generatesetter.CommonConstants;
+import com.google.common.collect.ImmutableMap;
+import com.intellij.psi.PsiMethod;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.List;
+import java.util.Map;
+import java.util.Set;
+
+import static com.bruce.intellijplugin.generatesetter.actions.AssertAllGetterAction.TestEngine.ASSERT;
+import static com.bruce.intellijplugin.generatesetter.actions.AssertAllGetterAction.TestEngine.ASSERTJ;
+import static com.bruce.intellijplugin.generatesetter.actions.AssertAllGetterAction.TestEngine.JUNIT4;
+import static com.bruce.intellijplugin.generatesetter.actions.AssertAllGetterAction.TestEngine.JUNIT5;
+import static com.bruce.intellijplugin.generatesetter.actions.AssertAllGetterAction.TestEngine.TESTNG;
+
+public class AssertNotNullAction extends AssertAllGetterAction {
+    // imports to add when generating asserts.
+    private static final Map<TestEngine, String> engineImports = ImmutableMap.<TestEngine, String>builder()
+            .put(JUNIT4, "static org.junit.Assert.assertNotNull")
+            .put(JUNIT5, "static org.junit.jupiter.api.Assertions.assertNotNull")
+            .put(TESTNG, "static org.testng.Assert.assertNotNull")
+            .put(ASSERTJ, "static org.assertj.core.api.Assertions.assertThat")
+            .put(ASSERT, "java.util.Objects")
+            .build();
+
+    private final GenerateAllAssertsHandlerAdapter generateAllHandler;
+
+    public AssertNotNullAction() {
+        generateAllHandler = new GenerateAllAssertsHandlerAdapter(true);
+        setGenerateAllHandler(generateAllHandler);
+    }
+
+    @NotNull
+    @Override
+    public String getText() {
+        return CommonConstants.ASSERT_NOT_NULL;
+    }
+
+    @NotNull
+    protected String generateStringForNoParam(String generateName,
+                                              List<PsiMethod> methodList, String splitText,
+                                              Set<String> newImportList, boolean hasGuava) {
+        generateAllHandler.detectCurrentTestEngine(project, containingFile);
+
+        TestEngine currentFileTestEngine = generateAllHandler.currentFileTestEngine;
+        Set<TestEngine> currentFileImportedEngines = generateAllHandler.currentFileImportedEngines;
+
+        if (!currentFileImportedEngines.contains(currentFileTestEngine)) {
+            newImportList.add(engineImports.get(currentFileTestEngine));
+        }
+
+        switch (currentFileTestEngine) {
+            case JUNIT4:
+            case JUNIT5:
+            case TESTNG:
+                return splitText + "assertNotNull(" + generateName + ");";
+            case ASSERTJ:
+                return splitText + "assertThat(" + generateName + ").isNotNull();";
+            case ASSERT:
+                return splitText + "assert " + generateName + " != null";
+            default:
+                throw new Error("Unknown case: " + currentFileTestEngine);
+        }
+    }
+}

src/main/java/com/bruce/intellijplugin/generatesetter/actions/GenerateAllSetterBase.java

@@ -446,7 +446,7 @@ private void handleWithLocalVariable(PsiLocalVariable localVariable,
     }
 
     @NotNull
-    private String generateStringForNoParam(String generateName,
+    protected String generateStringForNoParam(String generateName,
                                             List<PsiMethod> methodList, String splitText,
                                             Set<String> newImportList, boolean hasGuava) {
         StringBuilder builder = new StringBuilder();

src/main/resources/META-INF/plugin.xml

@@ -159,6 +159,11 @@
             <category>GenerateAllSetter</category>
         </intentionAction>
 
+        <intentionAction>
+            <className>com.bruce.intellijplugin.generatesetter.actions.AssertNotNullAction</className>
+            <category>GenerateAllSetter</category>
+        </intentionAction>
+
         <intentionAction>
             <className>com.bruce.intellijplugin.generatesetter.actions.AssertAllGetterAction</className>
             <category>GenerateAllSetter</category>

src/main/resources/intentionDescriptions/AssertNotNullAction/after.java.template

@@ -0,0 +1,2 @@
+User user = userService.findByName("admin");
+assertNotNull(user);

src/main/resources/intentionDescriptions/AssertNotNullAction/before.java.template

@@ -0,0 +1 @@
+User user = userService.findByName("admin");

src/main/resources/intentionDescriptions/AssertNotNullAction/description.html

@@ -0,0 +1,5 @@
+<html>
+<body>
+Generates assertNotNull call depends on the module's testing engine.
+</body>
+</html>