From ceeb309c48dad16ba3b8aefcb7a80ac4df49ca9b Mon Sep 17 00:00:00 2001 From: Jerry Chang Date: Wed, 28 Oct 2020 13:08:50 +0800 Subject: [PATCH] Add back switch user test in TaskTests Add back testSwitchUser which accidently removed by ag/12874703. Bug: 171198324 Test: atest TaskTests Change-Id: Ia3f5364cb3a47aa2beda6b9ea089173eb1999cde --- .../src/com/android/server/wm/TaskTests.java | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/services/tests/wmtests/src/com/android/server/wm/TaskTests.java b/services/tests/wmtests/src/com/android/server/wm/TaskTests.java index 3203ccb400d63..91e55ed7ab6ae 100644 --- a/services/tests/wmtests/src/com/android/server/wm/TaskTests.java +++ b/services/tests/wmtests/src/com/android/server/wm/TaskTests.java @@ -181,6 +181,21 @@ public class TaskTests extends WindowTestsBase { }); } + @Test + public void testSwitchUser() { + final Task rootTask = createTaskStackOnDisplay(mDisplayContent); + final Task childTask = createTaskInStack(rootTask, 0 /* userId */); + final Task leafTask1 = createTaskInStack(childTask, 10 /* userId */); + final Task leafTask2 = createTaskInStack(childTask, 0 /* userId */); + assertEquals(1, rootTask.getChildCount()); + assertEquals(leafTask2, childTask.getTopChild()); + + doReturn(true).when(leafTask1).showToCurrentUser(); + rootTask.switchUser(10); + assertEquals(1, rootTask.getChildCount()); + assertEquals(leafTask1, childTask.getTopChild()); + } + @Test public void testEnsureActivitiesVisible() { final Task rootTask = createTaskStackOnDisplay(mDisplayContent);