Ny test for getCells -metoden som inkluderer null

src/test/java/no/uib/inf101/colorgrid/TestColorGrid.java

@@ -91,6 +91,28 @@ public class TestColorGrid {
     }
   }
 
+  @Test
+  public void testGetCellsWithNull() {
+    IColorGrid grid = new WColorGrid(3, 1);
+
+    grid.set(new CellPosition(2, 0), Color.GREEN);
+
+    List<CellColor> expected = Arrays.asList(
+        new CellColor(new CellPosition(0, 0), null),
+        new CellColor(new CellPosition(1, 0), null),
+        new CellColor(new CellPosition(2, 0), Color.GREEN)
+    );
+    List<CellColor> actual = grid.getCells();
+
+    assertEquals(expected.size(), actual.size());
+    for (CellColor cp : actual) {
+      assertTrue(expected.contains(cp));
+    }
+    for (CellColor cp : expected) {
+      assertTrue(actual.contains(cp));
+    }
+  }
+
   @Test
   public void testIndexOutOfBoundsException() {
     IColorGrid grid = new WColorGrid(10, 20);