Create settings screen for Notification Assistant
Test: this atest Test: manual: change assistant and "adb shell dumpsys notification" Test: manual: verify persistance through reboot (including none) Fixes:120852765 Change-Id: Ie4516c3339246d66d7b6719ac5dd1d65c4d03b57
This commit is contained in:
@@ -0,0 +1,165 @@
|
||||
/*
|
||||
* Copyright (C) 2019 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.settings.notification;
|
||||
|
||||
import static org.junit.Assert.fail;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
import static org.junit.Assert.assertNull;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.mockito.ArgumentMatchers.anyInt;
|
||||
import static org.mockito.ArgumentMatchers.anyString;
|
||||
import static org.mockito.Mockito.RETURNS_SMART_NULLS;
|
||||
import static org.mockito.Mockito.mock;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
import android.content.ComponentName;
|
||||
import android.content.Context;
|
||||
import android.content.pm.PackageManager;
|
||||
import android.content.pm.ServiceInfo;
|
||||
|
||||
import com.android.settingslib.widget.CandidateInfo;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.MockitoAnnotations;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
import org.mockito.invocation.InvocationOnMock;
|
||||
import org.robolectric.RobolectricTestRunner;
|
||||
|
||||
@RunWith(RobolectricTestRunner.class)
|
||||
public class NotificationAssistantPickerTest {
|
||||
|
||||
private NotificationAssistantPicker mFragment;
|
||||
@Mock
|
||||
private Context mContext;
|
||||
@Mock
|
||||
private PackageManager mPackageManager;
|
||||
@Mock
|
||||
private NotificationBackend mNotificationBackend;
|
||||
private static final String TEST_PKG = "test.package";
|
||||
private static final String TEST_SRV = "test.component";
|
||||
private static final String TEST_CMP = TEST_PKG + "/" + TEST_SRV;
|
||||
private static final String TEST_NAME = "Test name";
|
||||
private static final ComponentName TEST_COMPONENT = ComponentName.unflattenFromString(TEST_CMP);
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
mFragment = new TestNotificationAssistantPicker(mContext, mPackageManager,
|
||||
mNotificationBackend);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getCurrentAssistant() {
|
||||
when(mNotificationBackend.getAllowedNotificationAssistant()).thenReturn(TEST_COMPONENT);
|
||||
String key = mFragment.getDefaultKey();
|
||||
assertEquals(key, TEST_CMP);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void getCurrentAssistant_None() {
|
||||
when(mNotificationBackend.getAllowedNotificationAssistant()).thenReturn(null);
|
||||
String key = mFragment.getDefaultKey();
|
||||
assertEquals(key, "");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void setAssistant() {
|
||||
mFragment.setDefaultKey(TEST_CMP);
|
||||
verify(mNotificationBackend).setNotificationAssistantGranted(TEST_COMPONENT);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void setAssistant_None() {
|
||||
mFragment.setDefaultKey("");
|
||||
verify(mNotificationBackend).setNotificationAssistantGranted(null);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void candidateListHasNoneAtEnd() {
|
||||
List<ServiceInfo> list = new ArrayList<>();
|
||||
ServiceInfo serviceInfo = mock(ServiceInfo.class, RETURNS_SMART_NULLS);
|
||||
serviceInfo.packageName = TEST_PKG;
|
||||
serviceInfo.name = TEST_SRV;
|
||||
list.add(serviceInfo);
|
||||
mFragment.onServicesReloaded(list);
|
||||
List<? extends CandidateInfo> candidates = mFragment.getCandidates();
|
||||
assertTrue(candidates.size() > 0);
|
||||
assertEquals(candidates.get(candidates.size() - 1).getKey(), "");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void candidateListHasCorrectCandidate() {
|
||||
List<ServiceInfo> list = new ArrayList<>();
|
||||
ServiceInfo serviceInfo = mock(ServiceInfo.class, RETURNS_SMART_NULLS);
|
||||
serviceInfo.packageName = TEST_PKG;
|
||||
serviceInfo.name = TEST_SRV;
|
||||
list.add(serviceInfo);
|
||||
mFragment.onServicesReloaded(list);
|
||||
List<? extends CandidateInfo> candidates = mFragment.getCandidates();
|
||||
boolean found = false;
|
||||
for (CandidateInfo c : candidates) {
|
||||
if (TEST_CMP.equals(c.getKey())) {
|
||||
found = true;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!found) fail();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void noDialogOnNoAssistantSelected() {
|
||||
when(mContext.getString(anyInt(), anyString())).thenAnswer(
|
||||
(InvocationOnMock invocation) -> {
|
||||
return invocation.getArgument(1);
|
||||
});
|
||||
assertNull(mFragment.getConfirmationMessage(
|
||||
new NotificationAssistantPicker.CandidateNone(mContext)));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void dialogTextHasAssistantName() {
|
||||
CandidateInfo c = mock(CandidateInfo.class);
|
||||
when(mContext.getString(anyInt(), anyString())).thenAnswer(
|
||||
(InvocationOnMock invocation) -> {
|
||||
return invocation.getArgument(1);
|
||||
});
|
||||
when(c.loadLabel()).thenReturn(TEST_NAME);
|
||||
when(c.getKey()).thenReturn(TEST_CMP);
|
||||
CharSequence text = mFragment.getConfirmationMessage(c);
|
||||
assertNotNull(text);
|
||||
assertTrue(text.toString().contains(TEST_NAME));
|
||||
}
|
||||
|
||||
|
||||
private static class TestNotificationAssistantPicker extends NotificationAssistantPicker {
|
||||
TestNotificationAssistantPicker(Context context, PackageManager packageManager,
|
||||
NotificationBackend notificationBackend) {
|
||||
mContext = context;
|
||||
mPm = packageManager;
|
||||
mNotificationBackend = notificationBackend;
|
||||
}
|
||||
}
|
||||
|
||||
}
|
Reference in New Issue
Block a user