1 /* 2 * Copyright (C) 2023 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.systemui.notetask 18 19 import android.content.Context 20 import android.content.Intent 21 import android.graphics.drawable.Icon 22 import android.os.UserHandle 23 import com.android.wm.shell.bubbles.Bubbles 24 import java.util.Optional 25 import kotlinx.coroutines.CoroutineDispatcher 26 27 /** 28 * Fake for [NoteTaskBubblesController] as mocking suspending functions is not supported in the 29 * Android tree's version of mockito. Ideally the [NoteTaskBubblesController] should be implemented 30 * using an interface for effectively providing multiple implementations but as this fake primarily 31 * for dealing with old version of mockito there isn't any benefit in adding complexity. 32 */ 33 class FakeNoteTaskBubbleController( 34 unUsed1: Context, 35 unsUsed2: CoroutineDispatcher, 36 private val optionalBubbles: Optional<Bubbles> 37 ) : NoteTaskBubblesController(unUsed1, unsUsed2) { 38 override suspend fun areBubblesAvailable() = optionalBubbles.isPresent 39 40 override suspend fun showOrHideAppBubble(intent: Intent, userHandle: UserHandle, icon: Icon) { 41 optionalBubbles.ifPresentOrElse( 42 { bubbles -> bubbles.showOrHideAppBubble(intent, userHandle, icon) }, 43 { throw IllegalAccessException() } 44 ) 45 } 46 } 47