From 86f06b333a9501bfb3206cdf978c77702bb75cc4 Mon Sep 17 00:00:00 2001
From: Kavin <20838718+FireMasterK@users.noreply.github.com>
Date: Mon, 14 Nov 2022 00:05:31 +0000
Subject: [PATCH] Address review.

---
 .../org/schabi/newpipe/extractor/stream/AudioStream.java    | 2 ++
 .../youtube/stream/YoutubeStreamExtractorDefaultTest.java   | 6 ++++++
 2 files changed, 8 insertions(+)

diff --git a/extractor/src/main/java/org/schabi/newpipe/extractor/stream/AudioStream.java b/extractor/src/main/java/org/schabi/newpipe/extractor/stream/AudioStream.java
index dd8efc61f..3db70b599 100644
--- a/extractor/src/main/java/org/schabi/newpipe/extractor/stream/AudioStream.java
+++ b/extractor/src/main/java/org/schabi/newpipe/extractor/stream/AudioStream.java
@@ -415,6 +415,7 @@ public final class AudioStream extends Stream {
      *
      * @return the id of the audio track
      */
+    @Nullable
     public String getAudioTrackId() {
         return audioTrackId;
     }
@@ -424,6 +425,7 @@ public final class AudioStream extends Stream {
      *
      * @return the name of the audio track
      */
+    @Nullable
     public String getAudioTrackName() {
         return audioTrackName;
     }
diff --git a/extractor/src/test/java/org/schabi/newpipe/extractor/services/youtube/stream/YoutubeStreamExtractorDefaultTest.java b/extractor/src/test/java/org/schabi/newpipe/extractor/services/youtube/stream/YoutubeStreamExtractorDefaultTest.java
index 4bfe2b855..defbfd2be 100644
--- a/extractor/src/test/java/org/schabi/newpipe/extractor/services/youtube/stream/YoutubeStreamExtractorDefaultTest.java
+++ b/extractor/src/test/java/org/schabi/newpipe/extractor/services/youtube/stream/YoutubeStreamExtractorDefaultTest.java
@@ -494,6 +494,12 @@ public class YoutubeStreamExtractorDefaultTest {
                             .stream()
                             .anyMatch(audioStream -> audioStream.getAudioTrackName().equals("English"))
             );
+
+            assertTrue(
+                    extractor.getAudioStreams()
+                            .stream()
+                            .anyMatch(audioStream -> audioStream.getAudioTrackName().equals("Hindi"))
+            );
         }
 
     }