add queue test

billryan · billryan · commit 5d1d755dbdaa · 2019-08-10T11:37:28.000+08:00
diff --git a/codes/algorithm-java/src/test/java/basic/DataStructureTest.java b/codes/algorithm-java/src/test/java/basic/DataStructureTest.java
@@ -5,6 +5,7 @@
 import org.slf4j.LoggerFactory;
 
 import java.util.Deque;
+import java.util.Queue;
 
 /**
  * @author billryan
@@ -14,6 +15,23 @@ public class DataStructureTest {
 
     private static final Logger log = LoggerFactory.getLogger(DataStructureTest.class);
 
+    @Test
+    public void testQueue() {
+        DataStructure<Integer> ds = new DataStructure<>();
+        Queue<Integer> queue = ds.getQueue();
+        queue.offer(1);
+        queue.offer(2);
+        queue.offer(3);
+
+        int peek = 1;
+        while (!queue.isEmpty()) {
+            int queuePeek = queue.poll();
+            log.info("queue peek: {}", queuePeek);
+            assert peek == queuePeek;
+            peek++;
+        }
+    }
+
     @Test
     public void testStack() {
         DataStructure<Integer> ds = new DataStructure<>();
