feat: MainUI tests

pom.xml

@@ -49,6 +49,12 @@
             <version>3.24.2</version>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>org.awaitility</groupId>
+            <artifactId>awaitility</artifactId>
+            <version>4.2.0</version>
+            <scope>test</scope>
+        </dependency>
     </dependencies>
 
     <build>

src/test/java/com/cys4/sensitivediscoverer/MainUITest.java

@@ -0,0 +1,81 @@
+package com.cys4.sensitivediscoverer;
+
+import burp.IBurpExtenderCallbacks;
+import com.cys4.sensitivediscoverer.mock.BurpExtenderCallbacksMock;
+import com.cys4.sensitivediscoverer.model.ScannerOptions;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+import javax.swing.*;
+
+import static java.util.concurrent.TimeUnit.SECONDS;
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.awaitility.Awaitility.await;
+
+class MainUITest {
+    private MainUI mainUI;
+    private IBurpExtenderCallbacks burpExtenderCallbacks;
+
+    @BeforeEach
+    void setUp() throws Exception {
+        this.burpExtenderCallbacks = new BurpExtenderCallbacksMock();
+        this.mainUI = new MainUI(burpExtenderCallbacks);
+        assertThat(this.mainUI.isInterfaceInitialized()).isFalse();
+    }
+
+    @Test
+    void testGetScannerOptions() {
+        assertThat(this.mainUI.getScannerOptions())
+                .isNotNull()
+                .isInstanceOf(ScannerOptions.class);
+    }
+
+    @Test
+    void testGetMainPanel() {
+        assertThat(this.mainUI.getMainPanel()).isNull();
+
+        this.mainUI.initializeUI();
+        await().atMost(30, SECONDS).until(this.mainUI::isInterfaceInitialized);
+        assertThat(this.mainUI.isInterfaceInitialized()).isTrue();
+        assertThat(this.mainUI.getMainPanel()).isNotNull();
+
+        assertThat(this.mainUI.getMainPanel().getTabCount()).isEqualTo(3);
+        assertThat(this.mainUI.getMainPanel().getComponentAt(0)).isInstanceOf(JPanel.class);
+        assertThat(this.mainUI.getMainPanel().getTitleAt(0)).isEqualTo("Logger");
+        assertThat(this.mainUI.getMainPanel().getComponentAt(1)).isInstanceOf(JPanel.class);
+        assertThat(this.mainUI.getMainPanel().getTitleAt(1)).isEqualTo("Options");
+        assertThat(this.mainUI.getMainPanel().getComponentAt(2)).isInstanceOf(JPanel.class);
+        assertThat(this.mainUI.getMainPanel().getTitleAt(2)).isEqualTo("About");
+    }
+
+    @Test
+    void testGetCallbacks() {
+        assertThat(this.burpExtenderCallbacks).isNotNull();
+        assertThat(this.mainUI.getCallbacks()).isSameAs(this.burpExtenderCallbacks);
+    }
+
+    @Test
+    void testGetGeneralRegexList() {
+        assertThat(this.mainUI.getGeneralRegexList()).isNotNull();
+    }
+
+    @Test
+    void testGetExtensionsRegexList() {
+        assertThat(this.mainUI.getExtensionsRegexList()).isNotNull();
+    }
+
+    @Test
+    void testGetNameExtension() {
+        assertThat(mainUI.getNameExtension()).isEqualTo("Sensitive Discoverer");
+    }
+
+    @Test
+    void testGetTabCaption() {
+        assertThat(this.mainUI.getTabCaption()).isEqualTo(this.mainUI.getNameExtension());
+    }
+
+    @Test
+    void testGetUiComponent() {
+        assertThat(this.mainUI.getUiComponent()).isSameAs(this.mainUI.getMainPanel());
+    }
+}

src/test/java/com/cys4/sensitivediscoverer/mock/BurpExtenderCallbacksMock.java

@@ -9,6 +9,7 @@
 import java.net.URL;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.function.Predicate;
 
 public class BurpExtenderCallbacksMock implements IBurpExtenderCallbacks {
@@ -44,17 +45,31 @@ public void setIsInScopePredicate(Predicate<URL> isInScopePredicate) {
     }
 
     @Override
-    public void setExtensionName(String s) {
-        throw new NotImplementedException();
+    public void addSuiteTab(ITab tab) {
+        if (Objects.isNull(tab)) throw new IllegalArgumentException("Tab is null");
+    }
+
+    @Override
+    public void customizeUiComponent(Component component) {
     }
 
     @Override
     public OutputStream getStdout() {
-        throw new NotImplementedException();
+        return System.out;
     }
 
     @Override
     public OutputStream getStderr() {
+        return System.err;
+    }
+
+    @Override
+    public ITextEditor createTextEditor() {
+        return new TextEditorMock();
+    }
+
+    @Override
+    public void setExtensionName(String s) {
         throw new NotImplementedException();
     }
 
@@ -253,21 +268,11 @@ public void unloadExtension() {
         throw new NotImplementedException();
     }
 
-    @Override
-    public void addSuiteTab(ITab iTab) {
-        throw new NotImplementedException();
-    }
-
     @Override
     public void removeSuiteTab(ITab iTab) {
         throw new NotImplementedException();
     }
 
-    @Override
-    public void customizeUiComponent(Component component) {
-        throw new NotImplementedException();
-    }
-
     @Override
     public IMessageEditor createMessageEditor(IMessageEditorController iMessageEditorController, boolean b) {
         throw new NotImplementedException();
@@ -288,11 +293,6 @@ public String loadExtensionSetting(String s) {
         throw new NotImplementedException();
     }
 
-    @Override
-    public ITextEditor createTextEditor() {
-        throw new NotImplementedException();
-    }
-
     @Override
     public void sendToRepeater(String s, int i, boolean b, byte[] bytes, String s1) {
         throw new NotImplementedException();

src/test/java/com/cys4/sensitivediscoverer/mock/TextEditorMock.java

@@ -0,0 +1,49 @@
+package com.cys4.sensitivediscoverer.mock;
+
+import burp.ITextEditor;
+import org.apache.commons.lang3.NotImplementedException;
+
+import javax.swing.*;
+import java.awt.*;
+
+public class TextEditorMock implements ITextEditor {
+    @Override
+    public Component getComponent() {
+        return new JPanel();
+    }
+
+    @Override
+    public void setEditable(boolean b) {
+        throw new NotImplementedException();
+    }
+
+    @Override
+    public byte[] getText() {
+        throw new NotImplementedException();
+    }
+
+    @Override
+    public void setText(byte[] bytes) {
+        throw new NotImplementedException();
+    }
+
+    @Override
+    public boolean isTextModified() {
+        throw new NotImplementedException();
+    }
+
+    @Override
+    public byte[] getSelectedText() {
+        throw new NotImplementedException();
+    }
+
+    @Override
+    public int[] getSelectionBounds() {
+        throw new NotImplementedException();
+    }
+
+    @Override
+    public void setSearchExpression(String s) {
+        throw new NotImplementedException();
+    }
+}