X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=test%2Fjalview%2Fdatamodel%2FAllColsIteratorTest.java;h=0dfcc21c826a8daa118456bb908923eec04ae0ca;hb=6206b38ff971c795daa4dbdc846fd0fca2247f3a;hp=ee82340ec5eb2a0abb86e2ea32532bb57d6710a0;hpb=2ca081eebf90c6731cd79bde140ca2fecc9c9644;p=jalview.git

diff --git a/test/jalview/datamodel/AllColsIteratorTest.java b/test/jalview/datamodel/AllColsIteratorTest.java
index ee82340..0dfcc21 100644
--- a/test/jalview/datamodel/AllColsIteratorTest.java
+++ b/test/jalview/datamodel/AllColsIteratorTest.java
@@ -29,13 +29,13 @@ import org.testng.annotations.Test;
 
 public class AllColsIteratorTest
 {
-  ColumnSelection hiddenCols;
+  HiddenColumns hiddenCols;
   
-  @BeforeClass
+  @BeforeClass(alwaysRun = true)
   public void setup()
   {
-   hiddenCols = new ColumnSelection();
-   hiddenCols.hideColumns(2,4);
+    hiddenCols = new HiddenColumns();
+    hiddenCols.hideColumns(2, 4);
   }
   
 
@@ -82,4 +82,21 @@ public class AllColsIteratorTest
     AllColsIterator it = new AllColsIterator(0, 3, hiddenCols);
     it.remove();
   }
+
+  /*
+   * Test iterator behaves correctly when there is only one element in the collection
+   */
+  @Test(groups = { "Functional" })
+  public void testOneElement()
+  {
+    HiddenColumns hidden = new HiddenColumns();
+    AllColsIterator it = new AllColsIterator(0, 0, hidden);
+    int count = 0;
+    while (it.hasNext())
+    {
+      it.next();
+      count++;
+    }
+    assertTrue(count == 1, "hasNext() is false after 1 iteration");
+  }
 }