timing.core: add testing for less used functions in the segment store

[deliverable/tracecompass.git] / statesystem / org.eclipse.tracecompass.segmentstore.core.tests / src / org / eclipse / tracecompass / segmentstore / core / tests / TreeMapStoreTest.java

diff --git a/statesystem/org.eclipse.tracecompass.segmentstore.core.tests/src/org/eclipse/tracecompass/segmentstore/core/tests/TreeMapStoreTest.java b/statesystem/org.eclipse.tracecompass.segmentstore.core.tests/src/org/eclipse/tracecompass/segmentstore/core/tests/TreeMapStoreTest.java
index f2f96b919ed47a260364097e4f6693ea21a4987c..56f79c4011199e46c51c5d63f4aefdbe2c8c7efb 100644 (file)
--- a/statesystem/org.eclipse.tracecompass.segmentstore.core.tests/src/org/eclipse/tracecompass/segmentstore/core/tests/TreeMapStoreTest.java
+++ b/statesystem/org.eclipse.tracecompass.segmentstore.core.tests/src/org/eclipse/tracecompass/segmentstore/core/tests/TreeMapStoreTest.java
@@ -12,6 +12,8 @@ package org.eclipse.tracecompass.segmentstore.core.tests;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
 
+import java.util.Arrays;
+
 import org.eclipse.jdt.annotation.NonNull;
 import org.eclipse.tracecompass.internal.segmentstore.core.treemap.TreeMapStore;
 import org.eclipse.tracecompass.segmentstore.core.BasicSegment;
@@ -31,6 +33,16 @@ public class TreeMapStoreTest extends AbstractTestSegmentStore {
         return new TreeMapStore<>();
     }
 
+    /**
+     * The TreeMapStore does not have a bulk loader, if it ever gets one, it should be tested here.
+     */
+    @Override
+    protected ISegmentStore<@NonNull ISegment> getSegmentStore(@NonNull ISegment @NonNull [] data) {
+        TreeMapStore<@NonNull ISegment> treeMapStore = new TreeMapStore<>();
+        treeMapStore.addAll(Arrays.asList(data));
+        return treeMapStore;
+    }
+
     /**
      * Try adding duplicate elements, they should be ignored
      */