ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/test/tck/ArrayBlockingQueueTest.java
(Generate patch)

Comparing jsr166/src/test/tck/ArrayBlockingQueueTest.java (file contents):
Revision 1.4 by dl, Sat Sep 20 18:20:07 2003 UTC vs.
Revision 1.14 by jsr166, Mon Nov 16 05:30:07 2009 UTC

# Line 1 | Line 1
1   /*
2 < * Written by members of JCP JSR-166 Expert Group and released to the
3 < * public domain. Use, modify, and redistribute this code in any way
4 < * without acknowledgement. Other contributors include Andrew Wright,
5 < * Jeffrey Hayes, Pat Fischer, Mike Judd.
2 > * Written by Doug Lea with assistance from members of JCP JSR-166
3 > * Expert Group and released to the public domain, as explained at
4 > * http://creativecommons.org/licenses/publicdomain
5 > * Other contributors include Andrew Wright, Jeffrey Hayes,
6 > * Pat Fisher, Mike Judd.
7   */
8  
9  
# Line 13 | Line 14 | import java.io.*;
14  
15   public class ArrayBlockingQueueTest extends JSR166TestCase {
16      public static void main(String[] args) {
17 <        junit.textui.TestRunner.run (suite());  
17 >        junit.textui.TestRunner.run (suite());
18      }
19      public static Test suite() {
20          return new TestSuite(ArrayBlockingQueueTest.class);
# Line 26 | Line 27 | public class ArrayBlockingQueueTest exte
27      private ArrayBlockingQueue populatedQueue(int n) {
28          ArrayBlockingQueue q = new ArrayBlockingQueue(n);
29          assertTrue(q.isEmpty());
30 <        for(int i = 0; i < n; i++)
30 >        for (int i = 0; i < n; i++)
31              assertTrue(q.offer(new Integer(i)));
32          assertFalse(q.isEmpty());
33          assertEquals(0, q.remainingCapacity());
34          assertEquals(n, q.size());
35          return q;
36      }
37 <
37 >
38      /**
39 <     *
39 >     * A new queue has the indicated capacity
40       */
41      public void testConstructor1() {
42          assertEquals(SIZE, new ArrayBlockingQueue(SIZE).remainingCapacity());
43      }
44  
45      /**
46 <     *
46 >     * Constructor throws IAE if  capacity argument nonpositive
47       */
48      public void testConstructor2() {
49          try {
# Line 53 | Line 54 | public class ArrayBlockingQueueTest exte
54      }
55  
56      /**
57 <     *
57 >     * Initializing from null Collection throws NPE
58       */
59      public void testConstructor3() {
59
60          try {
61              ArrayBlockingQueue q = new ArrayBlockingQueue(1, true, null);
62              shouldThrow();
# Line 65 | Line 65 | public class ArrayBlockingQueueTest exte
65      }
66  
67      /**
68 <     *
68 >     * Initializing from Collection of null elements throws NPE
69       */
70      public void testConstructor4() {
71          try {
# Line 77 | Line 77 | public class ArrayBlockingQueueTest exte
77      }
78  
79      /**
80 <     *
80 >     * Initializing from Collection with some null elements throws NPE
81       */
82      public void testConstructor5() {
83          try {
# Line 91 | Line 91 | public class ArrayBlockingQueueTest exte
91      }
92  
93      /**
94 <     *
94 >     * Initializing from too large collection throws IAE
95       */
96      public void testConstructor6() {
97          try {
# Line 105 | Line 105 | public class ArrayBlockingQueueTest exte
105      }
106  
107      /**
108 <     *
108 >     * Queue contains all elements of collection used to initialize
109       */
110      public void testConstructor7() {
111          try {
# Line 120 | Line 120 | public class ArrayBlockingQueueTest exte
120      }
121  
122      /**
123 <     *
123 >     * Queue transitions from empty to full when elements added
124       */
125      public void testEmptyFull() {
126          ArrayBlockingQueue q = new ArrayBlockingQueue(2);
# Line 135 | Line 135 | public class ArrayBlockingQueueTest exte
135      }
136  
137      /**
138 <     *
138 >     * remainingCapacity decreases on add, increases on remove
139       */
140      public void testRemainingCapacity() {
141          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 152 | Line 152 | public class ArrayBlockingQueueTest exte
152      }
153  
154      /**
155 <     *
155 >     *  offer(null) throws NPE
156       */
157      public void testOfferNull() {
158          try {
159              ArrayBlockingQueue q = new ArrayBlockingQueue(1);
160              q.offer(null);
161              shouldThrow();
162 <        } catch (NullPointerException success) { }  
162 >        } catch (NullPointerException success) { }
163 >    }
164 >
165 >    /**
166 >     *  add(null) throws NPE
167 >     */
168 >    public void testAddNull() {
169 >        try {
170 >            ArrayBlockingQueue q = new ArrayBlockingQueue(1);
171 >            q.add(null);
172 >            shouldThrow();
173 >        } catch (NullPointerException success) { }
174      }
175  
176      /**
177 <     *
177 >     * Offer succeeds if not full; fails if full
178       */
179      public void testOffer() {
180          ArrayBlockingQueue q = new ArrayBlockingQueue(1);
# Line 172 | Line 183 | public class ArrayBlockingQueueTest exte
183      }
184  
185      /**
186 <     *
186 >     * add succeeds if not full; throws ISE if full
187       */
188      public void testAdd() {
189          try {
# Line 182 | Line 193 | public class ArrayBlockingQueueTest exte
193              }
194              assertEquals(0, q.remainingCapacity());
195              q.add(new Integer(SIZE));
196 <        } catch (IllegalStateException success){
197 <        }  
196 >        } catch (IllegalStateException success) {
197 >        }
198      }
199  
200      /**
201 <     *
201 >     *  addAll(null) throws NPE
202       */
203      public void testAddAll1() {
204          try {
# Line 197 | Line 208 | public class ArrayBlockingQueueTest exte
208          }
209          catch (NullPointerException success) {}
210      }
211 +
212      /**
213 <     *
213 >     * addAll(this) throws IAE
214 >     */
215 >    public void testAddAllSelf() {
216 >        try {
217 >            ArrayBlockingQueue q = populatedQueue(SIZE);
218 >            q.addAll(q);
219 >            shouldThrow();
220 >        }
221 >        catch (IllegalArgumentException success) {}
222 >    }
223 >
224 >
225 >    /**
226 >     *  addAll of a collection with null elements throws NPE
227       */
228      public void testAddAll2() {
229          try {
# Line 210 | Line 235 | public class ArrayBlockingQueueTest exte
235          catch (NullPointerException success) {}
236      }
237      /**
238 <     *
238 >     * addAll of a collection with any null elements throws NPE after
239 >     * possibly adding some elements
240       */
241      public void testAddAll3() {
242          try {
# Line 224 | Line 250 | public class ArrayBlockingQueueTest exte
250          catch (NullPointerException success) {}
251      }
252      /**
253 <     *
253 >     * addAll throws ISE if not enough room
254       */
255      public void testAddAll4() {
256          try {
# Line 238 | Line 264 | public class ArrayBlockingQueueTest exte
264          catch (IllegalStateException success) {}
265      }
266      /**
267 <     *
267 >     * Queue contains all elements, in traversal order, of successful addAll
268       */
269      public void testAddAll5() {
270          try {
# Line 256 | Line 282 | public class ArrayBlockingQueueTest exte
282      }
283  
284      /**
285 <     *
285 >     *  put(null) throws NPE
286       */
287       public void testPutNull() {
288          try {
289              ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
290              q.put(null);
291              shouldThrow();
292 <        }
293 <        catch (NullPointerException success){
294 <        }  
292 >        }
293 >        catch (NullPointerException success) {
294 >        }
295          catch (InterruptedException ie) {
296              unexpectedException();
297          }
298       }
299  
300      /**
301 <     *
301 >     * all elements successfully put are contained
302       */
303       public void testPut() {
304           try {
# Line 290 | Line 316 | public class ArrayBlockingQueueTest exte
316      }
317  
318      /**
319 <     *
319 >     * put blocks interruptibly if full
320       */
321      public void testBlockingPut() {
322 +        final ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
323          Thread t = new Thread(new Runnable() {
324                  public void run() {
325                      int added = 0;
326                      try {
300                        ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
327                          for (int i = 0; i < SIZE; ++i) {
328                              q.put(new Integer(i));
329                              ++added;
330                          }
331                          q.put(new Integer(SIZE));
332                          threadShouldThrow();
333 <                    } catch (InterruptedException ie){
333 >                    } catch (InterruptedException ie) {
334                          threadAssertEquals(added, SIZE);
335 <                    }  
335 >                    }
336                  }});
337 <        try {
337 >        try {
338              t.start();
339 <           Thread.sleep(SHORT_DELAY_MS);
339 >           Thread.sleep(MEDIUM_DELAY_MS);
340             t.interrupt();
341             t.join();
342          }
# Line 320 | Line 346 | public class ArrayBlockingQueueTest exte
346      }
347  
348      /**
349 <     *
349 >     * put blocks waiting for take when full
350       */
351      public void testPutWithTake() {
352          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
# Line 337 | Line 363 | public class ArrayBlockingQueueTest exte
363                          q.put(new Object());
364                          ++added;
365                          threadShouldThrow();
366 <                    } catch (InterruptedException e){
366 >                    } catch (InterruptedException e) {
367                          threadAssertTrue(added >= 2);
368                      }
369                  }
# Line 348 | Line 374 | public class ArrayBlockingQueueTest exte
374              q.take();
375              t.interrupt();
376              t.join();
377 <        } catch (Exception e){
377 >        } catch (Exception e) {
378              unexpectedException();
379          }
380      }
381  
382      /**
383 <     *
383 >     * timed offer times out if full and elements not taken
384       */
385      public void testTimedOffer() {
386          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
# Line 366 | Line 392 | public class ArrayBlockingQueueTest exte
392                          threadAssertFalse(q.offer(new Object(), SHORT_DELAY_MS/2, TimeUnit.MILLISECONDS));
393                          q.offer(new Object(), LONG_DELAY_MS, TimeUnit.MILLISECONDS);
394                          threadShouldThrow();
395 <                    } catch (InterruptedException success){}
395 >                    } catch (InterruptedException success) {}
396                  }
397              });
398 <        
398 >
399          try {
400              t.start();
401              Thread.sleep(SHORT_DELAY_MS);
402              t.interrupt();
403              t.join();
404 <        } catch (Exception e){
404 >        } catch (Exception e) {
405              unexpectedException();
406          }
407      }
408  
409      /**
410 <     *
410 >     * take retrieves elements in FIFO order
411       */
412      public void testTake() {
413          try {
# Line 389 | Line 415 | public class ArrayBlockingQueueTest exte
415              for (int i = 0; i < SIZE; ++i) {
416                  assertEquals(i, ((Integer)q.take()).intValue());
417              }
418 <        } catch (InterruptedException e){
418 >        } catch (InterruptedException e) {
419              unexpectedException();
420 <        }  
420 >        }
421      }
422  
423      /**
424 <     *
424 >     * take blocks interruptibly when empty
425       */
426      public void testTakeFromEmpty() {
427          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
# Line 404 | Line 430 | public class ArrayBlockingQueueTest exte
430                      try {
431                          q.take();
432                          threadShouldThrow();
433 <                    } catch (InterruptedException success){ }                
433 >                    } catch (InterruptedException success) { }
434                  }
435              });
436          try {
# Line 412 | Line 438 | public class ArrayBlockingQueueTest exte
438              Thread.sleep(SHORT_DELAY_MS);
439              t.interrupt();
440              t.join();
441 <        } catch (Exception e){
441 >        } catch (Exception e) {
442              unexpectedException();
443          }
444      }
445  
446      /**
447 <     *
447 >     * Take removes existing elements until empty, then blocks interruptibly
448       */
449      public void testBlockingTake() {
450          Thread t = new Thread(new Runnable() {
# Line 430 | Line 456 | public class ArrayBlockingQueueTest exte
456                          }
457                          q.take();
458                          threadShouldThrow();
459 <                    } catch (InterruptedException success){
460 <                    }  
459 >                    } catch (InterruptedException success) {
460 >                    }
461                  }});
462 <        try {
462 >        try {
463              t.start();
464 <            Thread.sleep(SHORT_DELAY_MS);
464 >            Thread.sleep(SHORT_DELAY_MS);
465              t.interrupt();
466              t.join();
467          }
# Line 446 | Line 472 | public class ArrayBlockingQueueTest exte
472  
473  
474      /**
475 <     *
475 >     * poll succeeds unless empty
476       */
477      public void testPoll() {
478          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 457 | Line 483 | public class ArrayBlockingQueueTest exte
483      }
484  
485      /**
486 <     *
486 >     * timed pool with zero timeout succeeds when non-empty, else times out
487       */
488      public void testTimedPoll0() {
489          try {
# Line 466 | Line 492 | public class ArrayBlockingQueueTest exte
492                  assertEquals(i, ((Integer)q.poll(0, TimeUnit.MILLISECONDS)).intValue());
493              }
494              assertNull(q.poll(0, TimeUnit.MILLISECONDS));
495 <        } catch (InterruptedException e){
495 >        } catch (InterruptedException e) {
496              unexpectedException();
497 <        }  
497 >        }
498      }
499  
500      /**
501 <     *
501 >     * timed pool with nonzero timeout succeeds when non-empty, else times out
502       */
503      public void testTimedPoll() {
504          try {
# Line 481 | Line 507 | public class ArrayBlockingQueueTest exte
507                  assertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS)).intValue());
508              }
509              assertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
510 <        } catch (InterruptedException e){
510 >        } catch (InterruptedException e) {
511              unexpectedException();
512 <        }  
512 >        }
513      }
514  
515      /**
516 <     *
516 >     * Interrupted timed poll throws InterruptedException instead of
517 >     * returning timeout status
518       */
519      public void testInterruptedTimedPoll() {
520          Thread t = new Thread(new Runnable() {
# Line 498 | Line 525 | public class ArrayBlockingQueueTest exte
525                              threadAssertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS)).intValue());
526                          }
527                          threadAssertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
528 <                    } catch (InterruptedException success){
529 <                    }  
528 >                    } catch (InterruptedException success) {
529 >                    }
530                  }});
531 <        try {
531 >        try {
532              t.start();
533 <            Thread.sleep(SHORT_DELAY_MS);
533 >            Thread.sleep(SHORT_DELAY_MS);
534              t.interrupt();
535              t.join();
536          }
# Line 513 | Line 540 | public class ArrayBlockingQueueTest exte
540      }
541  
542      /**
543 <     *
543 >     *  timed poll before a delayed offer fails; after offer succeeds;
544 >     *  on interruption throws
545       */
546      public void testTimedPollWithOffer() {
547          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
# Line 524 | Line 552 | public class ArrayBlockingQueueTest exte
552                          q.poll(LONG_DELAY_MS, TimeUnit.MILLISECONDS);
553                          q.poll(LONG_DELAY_MS, TimeUnit.MILLISECONDS);
554                          threadShouldThrow();
555 <                    } catch (InterruptedException success) { }                
555 >                    } catch (InterruptedException success) { }
556                  }
557              });
558          try {
# Line 533 | Line 561 | public class ArrayBlockingQueueTest exte
561              assertTrue(q.offer(zero, SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
562              t.interrupt();
563              t.join();
564 <        } catch (Exception e){
564 >        } catch (Exception e) {
565              unexpectedException();
566          }
567 <    }  
567 >    }
568  
569  
570      /**
571 <     *
571 >     * peek returns next element, or null if empty
572       */
573      public void testPeek() {
574          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 554 | Line 582 | public class ArrayBlockingQueueTest exte
582      }
583  
584      /**
585 <     *
585 >     * element returns next element, or throws NSEE if empty
586       */
587      public void testElement() {
588          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 570 | Line 598 | public class ArrayBlockingQueueTest exte
598      }
599  
600      /**
601 <     *
601 >     * remove removes next element, or throws NSEE if empty
602       */
603      public void testRemove() {
604          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 580 | Line 608 | public class ArrayBlockingQueueTest exte
608          try {
609              q.remove();
610              shouldThrow();
611 <        } catch (NoSuchElementException success){
612 <        }  
611 >        } catch (NoSuchElementException success) {
612 >        }
613      }
614  
615      /**
616 <     *
616 >     * remove(x) removes x and returns true if present
617       */
618      public void testRemoveElement() {
619          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 598 | Line 626 | public class ArrayBlockingQueueTest exte
626          }
627          assertTrue(q.isEmpty());
628      }
629 <        
629 >
630      /**
631 <     *
631 >     * contains(x) reports true when elements added but not yet removed
632       */
633      public void testContains() {
634          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 612 | Line 640 | public class ArrayBlockingQueueTest exte
640      }
641  
642      /**
643 <     *
643 >     * clear removes all elements
644       */
645      public void testClear() {
646          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 622 | Line 650 | public class ArrayBlockingQueueTest exte
650          assertEquals(SIZE, q.remainingCapacity());
651          q.add(one);
652          assertFalse(q.isEmpty());
653 +        assertTrue(q.contains(one));
654          q.clear();
655          assertTrue(q.isEmpty());
656      }
657  
658      /**
659 <     *
659 >     * containsAll(c) is true when c contains a subset of elements
660       */
661      public void testContainsAll() {
662          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 641 | Line 670 | public class ArrayBlockingQueueTest exte
670      }
671  
672      /**
673 <     *
673 >     * retainAll(c) retains only those elements of c and reports true if changed
674       */
675      public void testRetainAll() {
676          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 660 | Line 689 | public class ArrayBlockingQueueTest exte
689      }
690  
691      /**
692 <     *
692 >     * removeAll(c) removes only those elements of c and reports true if changed
693       */
694      public void testRemoveAll() {
695          for (int i = 1; i < SIZE; ++i) {
# Line 675 | Line 704 | public class ArrayBlockingQueueTest exte
704          }
705      }
706  
678
707      /**
708 <     *
708 >     *  toArray contains all elements
709       */
710      public void testToArray() {
711          ArrayBlockingQueue q = populatedQueue(SIZE);
712          Object[] o = q.toArray();
713          try {
714 <        for(int i = 0; i < o.length; i++)
714 >        for (int i = 0; i < o.length; i++)
715              assertEquals(o[i], q.take());
716 <        } catch (InterruptedException e){
716 >        } catch (InterruptedException e) {
717              unexpectedException();
718 <        }    
718 >        }
719      }
720  
721      /**
722 <     *
722 >     * toArray(a) contains all elements
723       */
724      public void testToArray2() {
725          ArrayBlockingQueue q = populatedQueue(SIZE);
726          Integer[] ints = new Integer[SIZE];
727          ints = (Integer[])q.toArray(ints);
728          try {
729 <            for(int i = 0; i < ints.length; i++)
729 >            for (int i = 0; i < ints.length; i++)
730                  assertEquals(ints[i], q.take());
731 <        } catch (InterruptedException e){
731 >        } catch (InterruptedException e) {
732              unexpectedException();
733 <        }    
733 >        }
734      }
735 <    
735 >
736 >    /**
737 >     * toArray(null) throws NPE
738 >     */
739 >    public void testToArray_BadArg() {
740 >        try {
741 >            ArrayBlockingQueue q = populatedQueue(SIZE);
742 >            Object o[] = q.toArray(null);
743 >            shouldThrow();
744 >        } catch (NullPointerException success) {}
745 >    }
746 >
747 >    /**
748 >     * toArray with incompatible array type throws CCE
749 >     */
750 >    public void testToArray1_BadArg() {
751 >        try {
752 >            ArrayBlockingQueue q = populatedQueue(SIZE);
753 >            Object o[] = q.toArray(new String[10] );
754 >            shouldThrow();
755 >        } catch (ArrayStoreException  success) {}
756 >    }
757 >
758 >
759      /**
760 <     *
760 >     * iterator iterates through all elements
761       */
762      public void testIterator() {
763          ArrayBlockingQueue q = populatedQueue(SIZE);
764          Iterator it = q.iterator();
765          try {
766 <            while(it.hasNext()){
766 >            while (it.hasNext()) {
767                  assertEquals(it.next(), q.take());
768              }
769 <        } catch (InterruptedException e){
769 >        } catch (InterruptedException e) {
770              unexpectedException();
771 <        }    
771 >        }
772 >    }
773 >
774 >    /**
775 >     * iterator.remove removes current element
776 >     */
777 >    public void testIteratorRemove () {
778 >        final ArrayBlockingQueue q = new ArrayBlockingQueue(3);
779 >        q.add(two);
780 >        q.add(one);
781 >        q.add(three);
782 >
783 >        Iterator it = q.iterator();
784 >        it.next();
785 >        it.remove();
786 >
787 >        it = q.iterator();
788 >        assertEquals(it.next(), one);
789 >        assertEquals(it.next(), three);
790 >        assertFalse(it.hasNext());
791      }
792  
793      /**
794 <     *
794 >     * iterator ordering is FIFO
795       */
796      public void testIteratorOrdering() {
797          final ArrayBlockingQueue q = new ArrayBlockingQueue(3);
# Line 740 | Line 810 | public class ArrayBlockingQueueTest exte
810      }
811  
812      /**
813 <     *
813 >     * Modifications do not cause iterators to fail
814       */
815      public void testWeaklyConsistentIteration () {
816          final ArrayBlockingQueue q = new ArrayBlockingQueue(3);
# Line 756 | Line 826 | public class ArrayBlockingQueueTest exte
826          catch (ConcurrentModificationException e) {
827              unexpectedException();
828          }
759
829          assertEquals(0, q.size());
830      }
831  
832  
833      /**
834 <     *
834 >     * toString contains toStrings of elements
835       */
836      public void testToString() {
837          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 770 | Line 839 | public class ArrayBlockingQueueTest exte
839          for (int i = 0; i < SIZE; ++i) {
840              assertTrue(s.indexOf(String.valueOf(i)) >= 0);
841          }
842 <    }        
842 >    }
843  
844  
845      /**
846 <     *
846 >     * offer transfers elements across Executor tasks
847       */
848      public void testOfferInExecutor() {
780
849          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
782
850          q.add(one);
851          q.add(two);
785
852          ExecutorService executor = Executors.newFixedThreadPool(2);
787
853          executor.execute(new Runnable() {
854              public void run() {
855                  threadAssertFalse(q.offer(three));
# Line 809 | Line 874 | public class ArrayBlockingQueueTest exte
874                  }
875              }
876          });
877 <        
877 >
878          joinPool(executor);
879  
880      }
881  
882      /**
883 <     *
883 >     * poll retrieves elements across Executor threads
884       */
885      public void testPollInExecutor() {
821
886          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
823
887          ExecutorService executor = Executors.newFixedThreadPool(2);
825
888          executor.execute(new Runnable() {
889              public void run() {
890                  threadAssertNull(q.poll());
# Line 847 | Line 909 | public class ArrayBlockingQueueTest exte
909                  }
910              }
911          });
850        
851        joinPool(executor);
912  
913 +        joinPool(executor);
914      }
915  
916      /**
917 <     *
917 >     * A deserialized serialized queue has same elements in same order
918       */
919      public void testSerialization() {
920          ArrayBlockingQueue q = populatedQueue(SIZE);
# Line 868 | Line 929 | public class ArrayBlockingQueueTest exte
929              ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
930              ArrayBlockingQueue r = (ArrayBlockingQueue)in.readObject();
931              assertEquals(q.size(), r.size());
932 <            while (!q.isEmpty())
932 >            while (!q.isEmpty())
933                  assertEquals(q.remove(), r.remove());
934 <        } catch(Exception e){
934 >        } catch (Exception e) {
935 >            unexpectedException();
936 >        }
937 >    }
938 >
939 >    /**
940 >     * drainTo(null) throws NPE
941 >     */
942 >    public void testDrainToNull() {
943 >        ArrayBlockingQueue q = populatedQueue(SIZE);
944 >        try {
945 >            q.drainTo(null);
946 >            shouldThrow();
947 >        } catch (NullPointerException success) {
948 >        }
949 >    }
950 >
951 >    /**
952 >     * drainTo(this) throws IAE
953 >     */
954 >    public void testDrainToSelf() {
955 >        ArrayBlockingQueue q = populatedQueue(SIZE);
956 >        try {
957 >            q.drainTo(q);
958 >            shouldThrow();
959 >        } catch (IllegalArgumentException success) {
960 >        }
961 >    }
962 >
963 >    /**
964 >     * drainTo(c) empties queue into another collection c
965 >     */
966 >    public void testDrainTo() {
967 >        ArrayBlockingQueue q = populatedQueue(SIZE);
968 >        ArrayList l = new ArrayList();
969 >        q.drainTo(l);
970 >        assertEquals(q.size(), 0);
971 >        assertEquals(l.size(), SIZE);
972 >        for (int i = 0; i < SIZE; ++i)
973 >            assertEquals(l.get(i), new Integer(i));
974 >        q.add(zero);
975 >        q.add(one);
976 >        assertFalse(q.isEmpty());
977 >        assertTrue(q.contains(zero));
978 >        assertTrue(q.contains(one));
979 >        l.clear();
980 >        q.drainTo(l);
981 >        assertEquals(q.size(), 0);
982 >        assertEquals(l.size(), 2);
983 >        for (int i = 0; i < 2; ++i)
984 >            assertEquals(l.get(i), new Integer(i));
985 >    }
986 >
987 >    /**
988 >     * drainTo empties full queue, unblocking a waiting put.
989 >     */
990 >    public void testDrainToWithActivePut() {
991 >        final ArrayBlockingQueue q = populatedQueue(SIZE);
992 >        Thread t = new Thread(new Runnable() {
993 >                public void run() {
994 >                    try {
995 >                        q.put(new Integer(SIZE+1));
996 >                    } catch (InterruptedException ie) {
997 >                        threadUnexpectedException();
998 >                    }
999 >                }
1000 >            });
1001 >        try {
1002 >            t.start();
1003 >            ArrayList l = new ArrayList();
1004 >            q.drainTo(l);
1005 >            assertTrue(l.size() >= SIZE);
1006 >            for (int i = 0; i < SIZE; ++i)
1007 >                assertEquals(l.get(i), new Integer(i));
1008 >            t.join();
1009 >            assertTrue(q.size() + l.size() >= SIZE);
1010 >        } catch (Exception e) {
1011              unexpectedException();
1012          }
1013      }
1014  
1015 +    /**
1016 +     * drainTo(null, n) throws NPE
1017 +     */
1018 +    public void testDrainToNullN() {
1019 +        ArrayBlockingQueue q = populatedQueue(SIZE);
1020 +        try {
1021 +            q.drainTo(null, 0);
1022 +            shouldThrow();
1023 +        } catch (NullPointerException success) {
1024 +        }
1025 +    }
1026 +
1027 +    /**
1028 +     * drainTo(this, n) throws IAE
1029 +     */
1030 +    public void testDrainToSelfN() {
1031 +        ArrayBlockingQueue q = populatedQueue(SIZE);
1032 +        try {
1033 +            q.drainTo(q, 0);
1034 +            shouldThrow();
1035 +        } catch (IllegalArgumentException success) {
1036 +        }
1037 +    }
1038 +
1039 +    /**
1040 +     * drainTo(c, n) empties first max {n, size} elements of queue into c
1041 +     */
1042 +    public void testDrainToN() {
1043 +        ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE*2);
1044 +        for (int i = 0; i < SIZE + 2; ++i) {
1045 +            for (int j = 0; j < SIZE; j++)
1046 +                assertTrue(q.offer(new Integer(j)));
1047 +            ArrayList l = new ArrayList();
1048 +            q.drainTo(l, i);
1049 +            int k = (i < SIZE)? i : SIZE;
1050 +            assertEquals(l.size(), k);
1051 +            assertEquals(q.size(), SIZE-k);
1052 +            for (int j = 0; j < k; ++j)
1053 +                assertEquals(l.get(j), new Integer(j));
1054 +            while (q.poll() != null) ;
1055 +        }
1056 +    }
1057 +
1058  
1059   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines