hbase-common/src/test/java/org/apache/hadoop/hbase/util/TestBoundedArrayQueue.java

   1 /**
   2  * Licensed to the Apache Software Foundation (ASF) under one
   3  * or more contributor license agreements.  See the NOTICE file
   4  * distributed with this work for additional information
   5  * regarding copyright ownership.  The ASF licenses this file
   6  * to you under the Apache License, Version 2.0 (the
   7  * "License"); you may not use this file except in compliance
   8  * with the License.  You may obtain a copy of the License at
   9  *
  10  *     http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  */
  18 package org.apache.hadoop.hbase.util;
  19
  20 import static org.junit.Assert.assertEquals;
  21 import static org.junit.Assert.assertFalse;
  22 import static org.junit.Assert.assertNull;
  23 import static org.junit.Assert.assertTrue;
  24
  25 import org.apache.hadoop.hbase.testclassification.MiscTests;
  26 import org.apache.hadoop.hbase.testclassification.SmallTests;
  27 import org.junit.Test;
  28 import org.junit.experimental.categories.Category;
  29
  30 @Category({ MiscTests.class, SmallTests.class })
  31 public class TestBoundedArrayQueue {
  32
  33   private int qMaxElements = 5;
  34   private BoundedArrayQueue<Integer> queue = new BoundedArrayQueue<>(qMaxElements);
  35
  36   @Test
  37   public void testBoundedArrayQueueOperations() throws Exception {
  38     assertEquals(0, queue.size());
  39     assertNull(queue.poll());
  40     assertNull(queue.peek());
  41     for(int i=0;i<qMaxElements;i++){
  42       assertTrue(queue.offer(i));
  43     }
  44     assertEquals(qMaxElements, queue.size());
  45     assertFalse(queue.offer(0));
  46     assertEquals(0, queue.peek().intValue());
  47     assertEquals(0, queue.peek().intValue());
  48     for (int i = 0; i < qMaxElements; i++) {
  49       assertEquals(i, queue.poll().intValue());
  50     }
  51     assertEquals(0, queue.size());
  52     assertNull(queue.poll());
  53     // Write after one cycle is over
  54     assertTrue(queue.offer(100));
  55     assertTrue(queue.offer(1000));
  56     assertEquals(100, queue.peek().intValue());
  57     assertEquals(100, queue.poll().intValue());
  58     assertEquals(1000, queue.poll().intValue());
  59   }
  60 }