| /* |
| * Copyright (C) 2018 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.systemui.statusbar.policy; |
| |
| import static junit.framework.Assert.assertEquals; |
| import static junit.framework.Assert.assertFalse; |
| import static junit.framework.Assert.assertTrue; |
| |
| import android.os.Handler; |
| import android.os.Looper; |
| import android.provider.Settings; |
| import android.test.suitebuilder.annotation.SmallTest; |
| import android.testing.AndroidTestingRunner; |
| import android.testing.TestableLooper; |
| import android.testing.TestableResources; |
| |
| import com.android.systemui.R; |
| import com.android.systemui.SysuiTestCase; |
| |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| |
| @RunWith(AndroidTestingRunner.class) |
| @TestableLooper.RunWithLooper |
| @SmallTest |
| public class SmartReplyConstantsTest extends SysuiTestCase { |
| |
| private static final int CONTENT_OBSERVER_TIMEOUT_SECONDS = 10; |
| |
| private SmartReplyConstants mConstants; |
| |
| @Before |
| public void setUp() { |
| overrideSetting(null); // No config. |
| TestableResources resources = mContext.getOrCreateTestableResources(); |
| resources.addOverride(R.bool.config_smart_replies_in_notifications_enabled, true); |
| resources.addOverride( |
| R.integer.config_smart_replies_in_notifications_max_squeeze_remeasure_attempts, 7); |
| mConstants = new SmartReplyConstants(new Handler(Looper.getMainLooper()), mContext); |
| } |
| |
| @Test |
| public void testIsEnabledWithNoConfig() { |
| assertTrue(mConstants.isEnabled()); |
| } |
| |
| @Test |
| public void testIsEnabledWithInvalidConfig() { |
| overrideSetting("invalid config"); |
| triggerConstantsOnChange(); |
| assertTrue(mConstants.isEnabled()); |
| } |
| |
| @Test |
| public void testIsEnabledWithValidConfig() { |
| overrideSetting("enabled=false,max_squeeze_remeasure_attempts=5"); |
| triggerConstantsOnChange(); |
| assertFalse(mConstants.isEnabled()); |
| } |
| |
| @Test |
| public void testGetMaxSqueezeRemeasureAttemptsWithNoConfig() { |
| assertTrue(mConstants.isEnabled()); |
| assertEquals(7, mConstants.getMaxSqueezeRemeasureAttempts()); |
| } |
| |
| @Test |
| public void testGetMaxSqueezeRemeasureAttemptsWithInvalidConfig() { |
| overrideSetting("invalid config"); |
| triggerConstantsOnChange(); |
| assertEquals(7, mConstants.getMaxSqueezeRemeasureAttempts()); |
| } |
| |
| @Test |
| public void testGetMaxSqueezeRemeasureAttemptsWithValidConfig() { |
| overrideSetting("enabled=false,max_squeeze_remeasure_attempts=5"); |
| triggerConstantsOnChange(); |
| assertEquals(5, mConstants.getMaxSqueezeRemeasureAttempts()); |
| } |
| |
| private void overrideSetting(String flags) { |
| Settings.Global.putString(mContext.getContentResolver(), |
| Settings.Global.SMART_REPLIES_IN_NOTIFICATIONS_FLAGS, flags); |
| } |
| |
| private void triggerConstantsOnChange() { |
| // Since Settings.Global is mocked in TestableContext, we need to manually trigger the |
| // content observer. |
| mConstants.onChange(false, |
| Settings.Global.getUriFor(Settings.Global.SMART_REPLIES_IN_NOTIFICATIONS_FLAGS)); |
| } |
| } |