From 363bbf5fd3d3fa0ceb060f4d0552db9f062fcff7 Mon Sep 17 00:00:00 2001
From: Eric Xu <xeric.2002@gmail.com>
Date: Sat, 22 May 2021 00:00:39 -0400
Subject: [PATCH] Test getItem(int)

---
 .../player/playqueue/PlayQueueTest.java       | 30 +++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/app/src/test/java/org/schabi/newpipe/player/playqueue/PlayQueueTest.java b/app/src/test/java/org/schabi/newpipe/player/playqueue/PlayQueueTest.java
index b3641547f..587d54b0f 100644
--- a/app/src/test/java/org/schabi/newpipe/player/playqueue/PlayQueueTest.java
+++ b/app/src/test/java/org/schabi/newpipe/player/playqueue/PlayQueueTest.java
@@ -1,6 +1,7 @@
 package org.schabi.newpipe.player.playqueue;
 
 import org.junit.Before;
+import org.junit.BeforeClass;
 import org.junit.Test;
 import org.junit.experimental.runners.Enclosed;
 import org.junit.runner.RunWith;
@@ -10,6 +11,7 @@ import org.schabi.newpipe.extractor.stream.StreamType;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
+import java.util.Objects;
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotEquals;
@@ -110,6 +112,34 @@ public class PlayQueueTest {
         }
     }
 
+    public static class GetItemTests {
+        private static List<PlayQueueItem> streams;
+        private PlayQueue queue;
+
+        @BeforeClass
+        public static void init() {
+            streams = new ArrayList<>(Collections.nCopies(5, makeItemWithUrl("OTHER_URL")));
+            streams.set(3, makeItemWithUrl("TARGET_URL"));
+        }
+
+        @Before
+        public void setup() {
+            queue = mockPlayQueue(0, streams);
+        }
+
+        @Test
+        public void inBounds() {
+            assertEquals("TARGET_URL", Objects.requireNonNull(queue.getItem(3)).getUrl());
+            assertEquals("OTHER_URL", Objects.requireNonNull(queue.getItem(1)).getUrl());
+        }
+
+        @Test
+        public void outOfBounds() {
+            assertNull(queue.getItem(-1));
+            assertNull(queue.getItem(5));
+        }
+    }
+
     public static class EqualsTests {
         private final PlayQueueItem item1 = makeItemWithUrl("URL_1");
         private final PlayQueueItem item2 = makeItemWithUrl("URL_2");