Fix CompatUILayoutTest#testOnClickForSizeCompatHint

Bug: 321225807
Test: atest CompatUILayoutTest
Change-Id: I33f14c30394805dd6e10e2be5bfdae5251f34734

libs/WindowManager/Shell/tests/unittest/src/com/android/wm/shell/compatui/CompatUILayoutTest.java

@@ -127,6 +127,7 @@ public class CompatUILayoutTest extends ShellTestCase {
     @Test
     public void testOnClickForSizeCompatHint() {
         mWindowManager.mHasSizeCompat = true;
+        doReturn(true).when(mWindowManager).shouldShowSizeCompatRestartButton(mTaskInfo);
         mWindowManager.createLayout(/* canShow= */ true);
         final LinearLayout sizeCompatHint = mLayout.findViewById(R.id.size_compat_hint);
         sizeCompatHint.performClick();