Add test

duckduckgo · Jan 10, 2025 · f628b0f · f628b0f
1 parent 0a1ae92
commit f628b0f
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/DuckDuckGoTests/YoutublePlayerNavigationHandlerTests.swift b/DuckDuckGoTests/YoutublePlayerNavigationHandlerTests.swift
@@ -240,6 +240,23 @@ class DuckPlayerNavigationHandlerTests: XCTestCase {
         }
     }
 
+    @MainActor
+    func testHandleNavigation_WithReferrerInURL_UpdatesDuckPlayerReferrer() async {
+        // Arrange
+        let youtubeURL = URL(string: "https://www.youtube.com/watch?v=abc123&dp_referrer=serp")!
+        let navigationAction = MockNavigationAction(request: URLRequest(url: youtubeURL))
+        playerSettings.mode = .alwaysAsk
+        playerSettings.openInNewTab = true
+        featureFlagger.enabledFeatures = [.duckPlayer, .duckPlayerOpenInNewTab]
+
+        // Act
+        handler.handleDuckNavigation(navigationAction, webView: mockWebView)
+
+        // Assert
+        XCTAssertEqual(handler.referrer, .serp)
+
+    }
+
     @MainActor
     func testHandleDelegateNavigation_DuckPlayerURL_CancelNavigationAndLoadsDuckPlayerWithParamsInTab() async {
         // Arrange