JAL-2446 added contains, delete, checks for inclusion on add + tests

[jalview.git] / test / jalview / datamodel / features / NCNodeTest.java

diff --git a/test/jalview/datamodel/features/NCNodeTest.java b/test/jalview/datamodel/features/NCNodeTest.java
index da0aa4e..73f957e 100644 (file)
--- a/test/jalview/datamodel/features/NCNodeTest.java
+++ b/test/jalview/datamodel/features/NCNodeTest.java
@@ -1,8 +1,11 @@
 package jalview.datamodel.features;
 
 import static org.testng.Assert.assertEquals;
+import static org.testng.Assert.assertFalse;
 import static org.testng.Assert.assertTrue;
 
+import jalview.datamodel.SequenceFeature;
+
 import java.util.ArrayList;
 import java.util.List;
 
@@ -75,4 +78,25 @@ public class NCNodeTest
     assertTrue(entries.contains(r1));
     assertTrue(entries.contains(r2));
   }
+
+  /**
+   * Tests for the contains method (uses entry.equals() test)
+   */
+  @Test(groups = "Functional")
+  public void testContains()
+  {
+    SequenceFeature sf1 = new SequenceFeature("type", "desc", 1, 10, 2f,
+            "group");
+    SequenceFeature sf2 = new SequenceFeature("type", "desc", 1, 10, 2f,
+            "group");
+    SequenceFeature sf3 = new SequenceFeature("type", "desc", 1, 10, 2f,
+            "anothergroup");
+    NCNode<SequenceFeature> node = new NCNode<SequenceFeature>(sf1);
+
+    assertFalse(node.contains(null));
+    assertTrue(node.contains(sf1));
+    assertTrue(node.contains(sf2)); // sf1.equals(sf2)
+    assertFalse(node.contains(sf3)); // !sf1.equals(sf3)
+  }
+
 }