tests/carservice_unit_test/src/com/android/car/admin/NotificationHelperValidImportanceTest.java

/*
 * Copyright (C) 2021 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.car.admin;

import static android.app.NotificationManager.IMPORTANCE_DEFAULT;
import static android.app.NotificationManager.IMPORTANCE_HIGH;

import static com.android.car.admin.NotificationHelper.CHANNEL_ID_DEFAULT;
import static com.android.car.admin.NotificationHelper.CHANNEL_ID_HIGH;
import static com.android.car.admin.NotificationHelper.newNotificationBuilder;
import static com.android.internal.R.string.android_system_label;

import static com.google.common.truth.Truth.assertWithMessage;

import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import android.app.Notification;
import android.app.NotificationChannel;
import android.app.NotificationManager;
import android.content.Context;
import android.os.Bundle;

import androidx.test.platform.app.InstrumentationRegistry;

import com.android.car.R;

import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;

import java.util.Arrays;
import java.util.List;


@RunWith(Parameterized.class)
public final class NotificationHelperValidImportanceTest {

    private static final String SYSTEM_LABEL = "System, I am your Label!";
    private static final String IMPORTANCE_NAME = "The name is Bond, Importance Bond!";

    @Rule
    public final MockitoRule mMockitoRule = MockitoJUnit.rule();

    private Context mRealContext = InstrumentationRegistry.getInstrumentation().getContext();

    private Context mSpiedContext;

    @Mock
    private NotificationManager mNotificationMgr;

    private final int mImportance;
    private final String mChannelId;
    private final int mResId;

    public NotificationHelperValidImportanceTest(int importance, String channelId, int resId) {
        mImportance = importance;
        mChannelId = channelId;
        mResId = resId;
    }

    @Before
    public void setContext() {
        mSpiedContext = spy(mRealContext);
        when(mSpiedContext.getSystemService(NotificationManager.class))
                .thenReturn(mNotificationMgr);
        when(mSpiedContext.getString(android_system_label)).thenReturn(SYSTEM_LABEL);
        when(mSpiedContext.getString(mResId)).thenReturn(IMPORTANCE_NAME);
    }

    @Test
    public void testNewNotificationBuilder() {
        Notification.Builder builder = newNotificationBuilder(mSpiedContext, mImportance);
        assertWithMessage("builder").that(builder).isNotNull();

        // Check custom label
        Bundle extras = builder.getExtras();
        assertWithMessage("extras").that(extras).isNotNull();
        String label = extras.getString(Notification.EXTRA_SUBSTITUTE_APP_NAME);
        assertWithMessage("label (extra %s)", Notification.EXTRA_SUBSTITUTE_APP_NAME).that(label)
                .isEqualTo(SYSTEM_LABEL);

        // Check notification arguments
        ArgumentCaptor<NotificationChannel> captor = ArgumentCaptor
                .forClass(NotificationChannel.class);
        verify(mNotificationMgr).createNotificationChannel(captor.capture());
        NotificationChannel channel = captor.getValue();
        assertWithMessage("channel id").that(channel.getId()).isEqualTo(mChannelId);
        assertWithMessage("importance").that(channel.getImportance()).isEqualTo(mImportance);
        assertWithMessage("name").that(channel.getName()).isEqualTo(IMPORTANCE_NAME);
    }

    @Parameterized.Parameters
    public static List<Object[]> provideParams() {
        return Arrays.asList(new Object[][] {
            { IMPORTANCE_DEFAULT, CHANNEL_ID_DEFAULT, R.string.importance_default },
            { IMPORTANCE_HIGH, CHANNEL_ID_HIGH, R.string.importance_high }
        });
    }
}