From dca3357a776cbfd7b490db41ab4489c2a3dc5bdc Mon Sep 17 00:00:00 2001
From: shubang <shubang@google.com>
Date: Wed, 19 Apr 2023 13:19:48 -0700
Subject: [PATCH] Fix a bug in timeshift mode

Test: atest TvInputServiceTest#verifyCommandTimeShiftSetMode
Bug: 273369119
Change-Id: Idfedfdd07de2f05b6e3fa5080b9ba0342507cb54
---
 media/java/android/media/tv/ITvInputSessionWrapper.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/media/java/android/media/tv/ITvInputSessionWrapper.java b/media/java/android/media/tv/ITvInputSessionWrapper.java
index 80a3e70fbd288..d749b91e38894 100644
--- a/media/java/android/media/tv/ITvInputSessionWrapper.java
+++ b/media/java/android/media/tv/ITvInputSessionWrapper.java
@@ -223,7 +223,8 @@ public class ITvInputSessionWrapper extends ITvInputSession.Stub implements Hand
                 break;
             }
             case DO_TIME_SHIFT_SET_MODE: {
-                mTvInputSessionImpl.timeShiftSetMode((Integer) msg.obj);
+                mTvInputSessionImpl.timeShiftSetMode(msg.arg1);
+                break;
             }
             case DO_TIME_SHIFT_ENABLE_POSITION_TRACKING: {
                 mTvInputSessionImpl.timeShiftEnablePositionTracking((Boolean) msg.obj);