/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.apache.flink.graph.types.valuearray; import org.apache.flink.types.NullValue; import org.junit.Test; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertTrue; public class NullValueArrayTest { @Test public void testUnboundedArray() { int count = 4096; ValueArray<NullValue> nva = new NullValueArray(); // add several elements for (int i = 0 ; i < count ; i++) { assertFalse(nva.isFull()); assertEquals(i, nva.size()); assertTrue(nva.add(NullValue.getInstance())); assertEquals(i + 1, nva.size()); } // array never fills assertFalse(nva.isFull()); assertEquals(count, nva.size()); // verify the array values int idx = 0; for (NullValue nv : nva) { assertEquals(NullValue.getInstance(), nv); } // add element past end of array assertTrue(nva.add(NullValue.getInstance())); assertTrue(nva.addAll(nva)); // test copy assertEquals(nva, nva.copy()); // test copyTo NullValueArray nva_to = new NullValueArray(); nva.copyTo(nva_to); assertEquals(nva, nva_to); // test mark/reset int size = nva.size(); nva.mark(); assertTrue(nva.add(NullValue.getInstance())); assertEquals(size + 1, nva.size()); nva.reset(); assertEquals(size, nva.size()); // test clear nva.clear(); assertEquals(0, nva.size()); } }