From 2bc2551b9107a546054929a0a55991610618eac7 Mon Sep 17 00:00:00 2001
From: Alex Stetson <alexstetson@google.com>
Date: Mon, 9 May 2022 12:20:27 -0700
Subject: [PATCH] Expose IME inset source control for testing

Bug: 229256359
Test: atest DisplaySystemBarsControllerTest
Change-Id: Ia16438d6e9968baf8f29ab2527565557cc671487
---
 .../com/android/wm/shell/common/DisplayImeController.java    | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/libs/WindowManager/Shell/src/com/android/wm/shell/common/DisplayImeController.java b/libs/WindowManager/Shell/src/com/android/wm/shell/common/DisplayImeController.java
index 3b83f1586d8c9..6a2acf438302b 100644
--- a/libs/WindowManager/Shell/src/com/android/wm/shell/common/DisplayImeController.java
+++ b/libs/WindowManager/Shell/src/com/android/wm/shell/common/DisplayImeController.java
@@ -498,6 +498,11 @@ public class DisplayImeController implements DisplayController.OnDisplaysChanged
                 dispatchVisibilityChanged(mDisplayId, isShowing);
             }
         }
+
+        @VisibleForTesting(otherwise = VisibleForTesting.PRIVATE)
+        public InsetsSourceControl getImeSourceControl() {
+            return mImeSourceControl;
+        }
     }
 
     void removeImeSurface() {