common/util/src/test/java/org/opendaylight/yangtools/util/concurrent/TrackingLinkedBlockingQueueTest.java

   1 /*
   2  * Copyright (c) 2014 Brocade Communications Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.yangtools.util.concurrent;
  10
  11 import static org.junit.Assert.assertEquals;
  12 import static org.junit.Assert.fail;
  13
  14 import java.util.Arrays;
  15 import java.util.concurrent.TimeUnit;
  16
  17 import org.junit.Test;
  18
  19 /**
  20  * Unit tests for TrackingLinkedBlockingQueue.
  21  *
  22  * @author Thomas Pantelis
  23  */
  24 public class TrackingLinkedBlockingQueueTest {
  25
  26     @Test
  27     public void testOffer() throws InterruptedException {
  28
  29         TrackingLinkedBlockingQueue<String> queue = new TrackingLinkedBlockingQueue<>( 2 );
  30
  31         assertEquals( "offer", true, queue.offer( "1" ) );
  32         assertEquals( "getLargestQueueSize", 1, queue.getLargestQueueSize() );
  33         assertEquals( "size", 1, queue.size() );
  34
  35         assertEquals( "offer", true, queue.offer( "2", 1, TimeUnit.MILLISECONDS ) );
  36         assertEquals( "getLargestQueueSize", 2, queue.getLargestQueueSize() );
  37         assertEquals( "size", 2, queue.size() );
  38
  39         assertEquals( "offer", false, queue.offer( "3" ) );
  40         assertEquals( "getLargestQueueSize", 2, queue.getLargestQueueSize() );
  41         assertEquals( "size", 2, queue.size() );
  42
  43         assertEquals( "offer", false, queue.offer( "4", 1, TimeUnit.MILLISECONDS ) );
  44         assertEquals( "getLargestQueueSize", 2, queue.getLargestQueueSize() );
  45         assertEquals( "size", 2, queue.size() );
  46     }
  47
  48     @Test
  49     public void testPut() throws InterruptedException {
  50
  51         TrackingLinkedBlockingQueue<String> queue = new TrackingLinkedBlockingQueue<>();
  52
  53         queue.put( "1" );
  54         assertEquals( "getLargestQueueSize", 1, queue.getLargestQueueSize() );
  55         assertEquals( "size", 1, queue.size() );
  56
  57         queue.put( "2" );
  58         assertEquals( "getLargestQueueSize", 2, queue.getLargestQueueSize() );
  59         assertEquals( "size", 2, queue.size() );
  60     }
  61
  62     @Test
  63     public void testAdd() {
  64
  65         TrackingLinkedBlockingQueue<String> queue = new TrackingLinkedBlockingQueue<>( 2 );
  66
  67         assertEquals( "add", true, queue.add( "1" ) );
  68         assertEquals( "getLargestQueueSize", 1, queue.getLargestQueueSize() );
  69         assertEquals( "size", 1, queue.size() );
  70
  71         assertEquals( "add", true, queue.add( "2" ) );
  72         assertEquals( "getLargestQueueSize", 2, queue.getLargestQueueSize() );
  73         assertEquals( "size", 2, queue.size() );
  74
  75         try {
  76             queue.add( "3" );
  77             fail( "Expected IllegalStateException" );
  78         } catch( IllegalStateException e ) {
  79             // Expected
  80             assertEquals( "getLargestQueueSize", 2, queue.getLargestQueueSize() );
  81             assertEquals( "size", 2, queue.size() );
  82         }
  83     }
  84
  85     @Test
  86     public void testAddAll() {
  87
  88         TrackingLinkedBlockingQueue<String> queue = new TrackingLinkedBlockingQueue<>( 3 );
  89
  90         queue.addAll( Arrays.asList( "1", "2" ) );
  91         assertEquals( "getLargestQueueSize", 2, queue.getLargestQueueSize() );
  92         assertEquals( "size", 2, queue.size() );
  93
  94         try {
  95             queue.addAll( Arrays.asList( "3", "4" ) );
  96             fail( "Expected IllegalStateException" );
  97         } catch( IllegalStateException e ) {
  98             // Expected
  99             assertEquals( "getLargestQueueSize", 3, queue.getLargestQueueSize() );
 100             assertEquals( "size", 3, queue.size() );
 101         }
 102     }
 103 }