Create BubbleSessionTrackerImpl (46ae2739) · Commits · e / os / android_frameworks_base

libs/WindowManager/Shell/multivalentTests/src/com/android/wm/shell/bubbles/logging/BubbleSessionTrackerImplTest.kt

0 → 100644

+74 −0

Original line number	Original line	Diff line number	Diff line
			/*
			* Copyright (C) 2025 The Android Open Source Project
			*
			* Licensed under the Apache License, Version 2.0 (the "License");
			* you may not use this file except in compliance with the License.
			* You may obtain a copy of the License at
			*
			* http://www.apache.org/licenses/LICENSE-2.0
			*
			* Unless required by applicable law or agreed to in writing, software
			* distributed under the License is distributed on an "AS IS" BASIS,
			* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
			* See the License for the specific language governing permissions and
			* limitations under the License.
			*/

			package com.android.wm.shell.bubbles.logging

			import androidx.test.ext.junit.runners.AndroidJUnit4
			import androidx.test.filters.SmallTest
			import com.android.internal.logging.InstanceId
			import com.android.internal.logging.InstanceIdSequence
			import com.android.internal.protolog.ProtoLog
			import com.google.common.truth.Truth.assertThat
			import org.junit.Before
			import org.junit.Test
			import org.junit.runner.RunWith

			/** Unit tests for [BubbleSessionTrackerImpl]. */
			@SmallTest
			@RunWith(AndroidJUnit4::class)
			class BubbleSessionTrackerImplTest {

			private val instanceIdSequence = FakeInstanceIdSequence()
			private val bubbleSessionTracker = BubbleSessionTrackerImpl(instanceIdSequence)

			@Before
			fun setUp() {
			ProtoLog.REQUIRE_PROTOLOGTOOL = false
			ProtoLog.init()
			}

			@Test
			fun startSession_generatesNewSessionId() {
			bubbleSessionTracker.start()
			val firstSessionId = instanceIdSequence.id
			bubbleSessionTracker.stop()

			bubbleSessionTracker.start()
			val secondSessionId = instanceIdSequence.id

			assertThat(firstSessionId).isNotEqualTo(secondSessionId)
			}

			@Test
			fun endSession_shouldNotGenerateNewId() {
			bubbleSessionTracker.start()
			val currentId = instanceIdSequence.id
			bubbleSessionTracker.stop()

			assertThat(currentId).isEqualTo(instanceIdSequence.id)
			}

			class FakeInstanceIdSequence : InstanceIdSequence(/* instanceIdMax= */ 10) {

			var id = -1
			private set

			override fun newInstanceId(): InstanceId {
			id = if (id == -1 \|\| id == 10) 1 else id + 1
			return InstanceId.fakeInstanceId(id)
			}
			}
			}

libs/WindowManager/Shell/src/com/android/wm/shell/bubbles/logging/BubbleSessionTracker.kt

0 → 100644

+31 −0

Original line number	Original line	Diff line number	Diff line
			/*
			* Copyright (C) 2025 The Android Open Source Project
			*
			* Licensed under the Apache License, Version 2.0 (the "License");
			* you may not use this file except in compliance with the License.
			* You may obtain a copy of the License at
			*
			* http://www.apache.org/licenses/LICENSE-2.0
			*
			* Unless required by applicable law or agreed to in writing, software
			* distributed under the License is distributed on an "AS IS" BASIS,
			* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
			* See the License for the specific language governing permissions and
			* limitations under the License.
			*/

			package com.android.wm.shell.bubbles.logging

			/**
			* Keeps track of the current Bubble session.
			*
			* Bubble sessions start when the stack expands and end when the stack collapses.
			*/
			interface BubbleSessionTracker {

			/** Starts tracking a new session. */
			fun start()

			/** Stops tracking the current session. */
			fun stop()
			}

libs/WindowManager/Shell/src/com/android/wm/shell/bubbles/logging/BubbleSessionTrackerImpl.kt

0 → 100644

+59 −0

Original line number	Original line	Diff line number	Diff line
			/*
			* Copyright (C) 2025 The Android Open Source Project
			*
			* Licensed under the Apache License, Version 2.0 (the "License");
			* you may not use this file except in compliance with the License.
			* You may obtain a copy of the License at
			*
			* http://www.apache.org/licenses/LICENSE-2.0
			*
			* Unless required by applicable law or agreed to in writing, software
			* distributed under the License is distributed on an "AS IS" BASIS,
			* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
			* See the License for the specific language governing permissions and
			* limitations under the License.
			*/

			package com.android.wm.shell.bubbles.logging

			import com.android.internal.logging.InstanceId
			import com.android.internal.logging.InstanceIdSequence
			import com.android.internal.protolog.ProtoLog
			import com.android.wm.shell.dagger.Bubbles
			import com.android.wm.shell.dagger.WMSingleton
			import com.android.wm.shell.protolog.ShellProtoLogGroup
			import javax.inject.Inject

			/**
			* Keeps track of the current bubble session and logs when sessions start and end.
			*
			* Sessions are identified using an [InstanceId].
			*/
			@WMSingleton
			class BubbleSessionTrackerImpl @Inject constructor(
			@Bubbles private val instanceIdSequence: InstanceIdSequence
			) : BubbleSessionTracker {

			private var currentSession: InstanceId? = null

			override fun start() {
			if (currentSession != null) {
			ProtoLog.d(
			ShellProtoLogGroup.WM_SHELL_BUBBLES_NOISY,
			"BubbleSessionTracker: starting to track a new session. "
			+ "previous session still active"
			)
			}
			currentSession = instanceIdSequence.newInstanceId()
			}

			override fun stop() {
			if (currentSession == null) {
			ProtoLog.d(
			ShellProtoLogGroup.WM_SHELL_BUBBLES_NOISY,
			"BubbleSessionTracker: session tracking stopped but current session is null"
			)
			}
			currentSession = null
			}
			}