[TEST] extend & fix structure view test

SeeSharpSoft · May 9, 2019 · 4406831 · 4406831
1 parent 20cf72a
commit 4406831
Showing 1 changed file with 96 additions and 1 deletion.
diff --git a/src/test/java/net/seesharpsoft/intellij/plugins/csv/structureview/CsvStructureViewTest.java b/src/test/java/net/seesharpsoft/intellij/plugins/csv/structureview/CsvStructureViewTest.java
@@ -4,6 +4,7 @@
 import com.intellij.ide.util.treeView.smartTree.TreeElement;
 import com.intellij.navigation.ItemPresentation;
 import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase;
+import net.seesharpsoft.intellij.plugins.csv.editor.CsvEditorSettingsExternalizable;
 
 public class CsvStructureViewTest extends LightCodeInsightFixtureTestCase {
 
@@ -23,8 +24,9 @@ private void doCheckTreeElement(TreeElement element, Class expectedClazz, String
         }
     }
 
-    public void testStructureView() {
+    public void testStructureViewWithoutFileEndLineBreakSupport() {
         myFixture.configureByFile("StructureViewTestData.csv");
+        CsvEditorSettingsExternalizable.getInstance().setFileEndLineBreak(false);
         myFixture.testStructureView(structureViewComponent -> {
             StructureViewTreeElement root = structureViewComponent.getTreeModel().getRoot();
             doCheckTreeElement(root, CsvStructureViewElement.File.class, "FirstName, LastName\n" +
@@ -124,5 +126,98 @@ public void testStructureView() {
         });
     }
 
+    public void testStructureViewFileEndLineBreakSupport() {
+        myFixture.configureByFile("StructureViewTestData.csv");
+        CsvEditorSettingsExternalizable.getInstance().setFileEndLineBreak(true);
+        myFixture.testStructureView(structureViewComponent -> {
+            StructureViewTreeElement root = structureViewComponent.getTreeModel().getRoot();
+            doCheckTreeElement(root, CsvStructureViewElement.File.class, "FirstName, LastName\n" +
+                    "Peter,Lustig,42\n" +
+                    "Martin\n" +
+                    ",Fuchs\n", null);
+            assertEquals(3, root.getChildren().length);
+
+            TreeElement header = root.getChildren()[0];
+            doCheckTreeElement(
+                    header,
+                    CsvStructureViewElement.Header.class,
+                    "FirstName",
+                    "Header (3 entries)"
+            );
+
+            TreeElement field = header.getChildren()[0];
+            doCheckTreeElement(
+                    field,
+                    CsvStructureViewElement.Field.class,
+                    "Peter",
+                    "(1)"
+            );
+            field = header.getChildren()[1];
+            doCheckTreeElement(
+                    field,
+                    CsvStructureViewElement.Field.class,
+                    "Martin",
+                    "(2)"
+            );
+            field = header.getChildren()[2];
+            doCheckTreeElement(
+                    field,
+                    CsvStructureViewElement.Field.class,
+                    "",
+                    "(3)"
+            );
 
+            /**
+             * LastName header
+             */
+            header = root.getChildren()[1];
+            doCheckTreeElement(
+                    header,
+                    CsvStructureViewElement.Header.class,
+                    "LastName",
+                    "Header (3 entries)"
+            );
+
+            field = header.getChildren()[0];
+            doCheckTreeElement(
+                    field,
+                    CsvStructureViewElement.Field.class,
+                    "Lustig",
+                    "(1)"
+            );
+            field = header.getChildren()[1];
+            doCheckTreeElement(
+                    field,
+                    CsvStructureViewElement.Field.class,
+                    "<undefined>",
+                    "(2)"
+            );
+            field = header.getChildren()[2];
+            doCheckTreeElement(
+                    field,
+                    CsvStructureViewElement.Field.class,
+                    "Fuchs",
+                    "(3)"
+            );
+
+            /**
+             * Empty header
+             */
+            header = root.getChildren()[2];
+            doCheckTreeElement(
+                    header,
+                    CsvStructureViewElement.Header.class,
+                    "<undefined>",
+                    "Header (1 entries)"
+            );
+
+            field = header.getChildren()[0];
+            doCheckTreeElement(
+                    field,
+                    CsvStructureViewElement.Field.class,
+                    "42",
+                    "(1)"
+            );
+        });
+    }
 }