added tests for new feature

Author: nickolas-dimitrakas

mParticle-Apple-Media-SDKTests/mParticle_Apple_MediaTests.swift

@@ -413,6 +413,73 @@ class mParticle_Apple_MediaTests: XCTestCase, MPListenerProtocol {
         mediaSession?.logAdBreakEnd()
         self.waitForExpectations(timeout: defaultTimeout, handler: nil)
     }
+    
+    func testAdBreakExclusionDisabledDoesNotPauseContentTime() {
+        let adBreak = MPMediaAdBreak(title: "foo adbreak title", id: "12345")
+        mediaSession?.excludeAdBreaksFromContentTime = false
+
+        // Start content playback and accumulate 0.2s before the ad break
+        XCTAssertEqual(mediaSession!.mediaContentTimeSpent, 0)
+        mediaSession?.logPlay()
+        Thread.sleep(forTimeInterval: 0.2)
+        
+        XCTAssertNotNil(mediaSession?.currentPlaybackStartTimestamp)
+        XCTAssertEqual(mediaSession!.mediaContentTimeSpent, 0.2, accuracy: 0.08)
+
+        // 0.2s should count toward content time.
+        mediaSession?.logAdBreakStart(adBreak: adBreak)
+        Thread.sleep(forTimeInterval: 0.2)
+        
+        XCTAssertEqual(mediaSession!.mediaContentTimeSpent, 0.4, accuracy: 0.08)
+        
+        mediaSession?.logAdBreakEnd()
+        XCTAssertNil(mediaSession?.adBreak)
+
+        mediaSession?.logPause()
+        
+        XCTAssertEqual(mediaSession!.mediaContentTimeSpent, 0.4, accuracy: 0.08)
+        XCTAssertNil(mediaSession?.currentPlaybackStartTimestamp)
+    }
+    
+    func testAdBreakExclusionEnabledExcludesAdTime() {
+        let adBreak = MPMediaAdBreak(title: "foo adbreak title", id: "12345")
+        mediaSession?.excludeAdBreaksFromContentTime = true
+
+        // Start content playback and accumulate 0.2s before the ad break
+        XCTAssertEqual(mediaSession!.mediaContentTimeSpent, 0)
+        mediaSession?.logPlay()
+        Thread.sleep(forTimeInterval: 0.2)
+
+        XCTAssertNotNil(mediaSession?.currentPlaybackStartTimestamp)
+        XCTAssertEqual(mediaSession!.storedPlaybackTime, 0)
+        XCTAssertEqual(mediaSession!.mediaContentTimeSpent, 0.2, accuracy: 0.08)
+
+        // Start ad break content time should pause & be stored
+        mediaSession?.logAdBreakStart(adBreak: adBreak)
+        Thread.sleep(forTimeInterval: 0.2)
+
+        // Content tracking paused: playhead cleared, stored captured 0.2s
+        XCTAssertEqual(mediaSession!.storedPlaybackTime, mediaSession!.mediaContentTimeSpent)
+        XCTAssertEqual(mediaSession!.mediaContentTimeSpent, 0.2, accuracy: 0.08)
+        XCTAssertNil(mediaSession?.currentPlaybackStartTimestamp)
+
+        // End ad break auto-resume content tracking
+        mediaSession?.logAdBreakEnd()
+        Thread.sleep(forTimeInterval: 0.2)
+        
+        XCTAssertEqual(mediaSession!.storedPlaybackTime, 0.2, accuracy: 0.08)
+        XCTAssertEqual(mediaSession!.mediaContentTimeSpent, 0.4, accuracy: 0.08)
+        XCTAssertNotNil(mediaSession?.currentPlaybackStartTimestamp)
+        XCTAssertNil(mediaSession?.adBreak)
+
+        // Watch a bit more content after the ad
+        Thread.sleep(forTimeInterval: 0.2)
+
+        mediaSession?.logPause()
+
+        XCTAssertEqual(mediaSession!.mediaContentTimeSpent, 0.6, accuracy: 0.08)
+        XCTAssertNil(mediaSession?.currentPlaybackStartTimestamp)
+    }
 
     func testLogSegmentStart() {
         let segment = MPMediaSegment(title: "foo segment title", index: 3, duration: 30000)