1 /*
2  * Copyright (C) 2020 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.settings.display;
18 
19 import static com.android.settings.core.BasePreferenceController.AVAILABLE;
20 import static com.android.settings.core.BasePreferenceController.UNSUPPORTED_ON_DEVICE;
21 import static com.android.settings.display.PeakRefreshRatePreferenceController.DEFAULT_REFRESH_RATE;
22 
23 import static com.google.common.truth.Truth.assertThat;
24 
25 import android.content.Context;
26 import android.provider.Settings;
27 import android.view.Display;
28 
29 import androidx.preference.SwitchPreference;
30 
31 import org.junit.Before;
32 import org.junit.Test;
33 import org.junit.runner.RunWith;
34 import org.robolectric.RobolectricTestRunner;
35 import org.robolectric.RuntimeEnvironment;
36 import org.robolectric.annotation.Config;
37 
38 @RunWith(RobolectricTestRunner.class)
39 public class PeakRefreshRatePreferenceControllerTest {
40 
41     private Context mContext;
42     private PeakRefreshRatePreferenceController mController;
43     private SwitchPreference mPreference;
44 
45     @Before
setUp()46     public void setUp() {
47         mContext = RuntimeEnvironment.application;
48         mController = new PeakRefreshRatePreferenceController(mContext, "key");
49         mPreference = new SwitchPreference(RuntimeEnvironment.application);
50     }
51 
52     @Test
53     @Config(qualifiers = "mcc999")
getAvailabilityStatus_withConfigNoShow_returnUnsupported()54     public void getAvailabilityStatus_withConfigNoShow_returnUnsupported() {
55         assertThat(mController.getAvailabilityStatus()).isEqualTo(UNSUPPORTED_ON_DEVICE);
56     }
57 
58     @Test
getAvailabilityStatus_refreshRateLargerThanDefault_returnAvailable()59     public void getAvailabilityStatus_refreshRateLargerThanDefault_returnAvailable() {
60         mController.mPeakRefreshRate = DEFAULT_REFRESH_RATE + 1;
61 
62         assertThat(mController.getAvailabilityStatus()).isEqualTo(AVAILABLE);
63     }
64 
65     @Test
getAvailabilityStatus_refreshRateEqualToDefault_returnUnsupported()66     public void getAvailabilityStatus_refreshRateEqualToDefault_returnUnsupported() {
67         mController.mPeakRefreshRate = DEFAULT_REFRESH_RATE;
68 
69         assertThat(mController.getAvailabilityStatus()).isEqualTo(UNSUPPORTED_ON_DEVICE);
70     }
71 
72     @Test
setChecked_enableSmoothDisplay_setCurrentRefreshRate()73     public void setChecked_enableSmoothDisplay_setCurrentRefreshRate() {
74         mController.mPeakRefreshRate = 88f;
75         mController.setChecked(true);
76 
77         assertThat(Settings.System.getFloat(mContext.getContentResolver(),
78                 Settings.System.PEAK_REFRESH_RATE, DEFAULT_REFRESH_RATE))
79                 .isEqualTo(88.0f);
80     }
81 
82     @Test
setChecked_disableSmoothDisplay_setDefaultRefreshRate()83     public void setChecked_disableSmoothDisplay_setDefaultRefreshRate() {
84         mController.mPeakRefreshRate = 88f;
85         mController.setChecked(false);
86 
87         assertThat(Settings.System.getFloat(mContext.getContentResolver(),
88                 Settings.System.PEAK_REFRESH_RATE, DEFAULT_REFRESH_RATE))
89                 .isEqualTo(DEFAULT_REFRESH_RATE);
90     }
91 
92     @Test
isChecked_enableSmoothDisplay_returnTrue()93     public void isChecked_enableSmoothDisplay_returnTrue() {
94         enableSmoothDisplayPreference();
95 
96         assertThat(mController.isChecked()).isTrue();
97     }
98 
99     @Test
isChecked_disableSmoothDisplay_returnFalse()100     public void isChecked_disableSmoothDisplay_returnFalse() {
101         disableSmoothDisplayPreference();
102 
103         assertThat(mController.isChecked()).isFalse();
104     }
105 
106     @Test
findPeakRefreshRate_moreThanOneHigherThanDefault()107     public void findPeakRefreshRate_moreThanOneHigherThanDefault() {
108         Display.Mode lower = new Display.Mode(0, 0, 0, DEFAULT_REFRESH_RATE - 1);
109         Display.Mode def = new Display.Mode(0, 0, 0, DEFAULT_REFRESH_RATE);
110         Display.Mode higher = new Display.Mode(0, 0, 0, DEFAULT_REFRESH_RATE + 1);
111         Display.Mode higher1 = new Display.Mode(0, 0, 0, DEFAULT_REFRESH_RATE + 2);
112 
113         assertThat(mController.findPeakRefreshRate(
114                 new Display.Mode[] {lower, def, higher, higher1}))
115                 .isEqualTo(DEFAULT_REFRESH_RATE + 2);
116         assertThat(mController.findPeakRefreshRate(
117                 new Display.Mode[] {lower, def, higher1, higher}))
118                 .isEqualTo(DEFAULT_REFRESH_RATE + 2);
119     }
120 
enableSmoothDisplayPreference()121     private void enableSmoothDisplayPreference() {
122         mController.mPeakRefreshRate = 88f;
123 
124         Settings.System.putFloat(
125                 mContext.getContentResolver(),
126                 Settings.System.PEAK_REFRESH_RATE,
127                 mController.mPeakRefreshRate);
128     }
129 
disableSmoothDisplayPreference()130     private void disableSmoothDisplayPreference() {
131         mController.mPeakRefreshRate = 88f;
132 
133         Settings.System.putFloat(
134                 mContext.getContentResolver(),
135                 Settings.System.PEAK_REFRESH_RATE,
136                 DEFAULT_REFRESH_RATE);
137     }
138 }
139