1 /*
2  * Copyright (C) 2018 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 package com.android.settings.wifi;
17 
18 import static org.mockito.ArgumentMatchers.any;
19 import static org.mockito.ArgumentMatchers.anyInt;
20 import static org.mockito.ArgumentMatchers.eq;
21 import static org.mockito.ArgumentMatchers.isNull;
22 import static org.mockito.ArgumentMatchers.nullable;
23 import static org.mockito.Mockito.doReturn;
24 import static org.mockito.Mockito.never;
25 import static org.mockito.Mockito.spy;
26 import static org.mockito.Mockito.times;
27 import static org.mockito.Mockito.verify;
28 import static org.mockito.Mockito.when;
29 
30 import android.content.ContentResolver;
31 import android.content.Context;
32 import android.content.DialogInterface;
33 import android.content.Intent;
34 import android.net.wifi.WifiManager;
35 import android.provider.Settings;
36 
37 import androidx.appcompat.app.AlertDialog;
38 import androidx.fragment.app.Fragment;
39 import androidx.fragment.app.FragmentActivity;
40 
41 import com.android.settings.R;
42 
43 import org.junit.Before;
44 import org.junit.Test;
45 import org.junit.runner.RunWith;
46 import org.mockito.Mock;
47 import org.mockito.MockitoAnnotations;
48 import org.robolectric.RobolectricTestRunner;
49 import org.robolectric.RuntimeEnvironment;
50 
51 @RunWith(RobolectricTestRunner.class)
52 public class WifiScanningRequiredFragmentTest {
53 
54     private WifiScanningRequiredFragment mFragment;
55     private Context mContext;
56     private ContentResolver mResolver;
57     @Mock
58     private WifiManager mWifiManager;
59     @Mock
60     private Fragment mCallbackFragment;
61     @Mock
62     private AlertDialog.Builder mBuilder;
63     @Mock
64     private FragmentActivity mActivity;
65 
66     @Before
setUp()67     public void setUp() {
68         MockitoAnnotations.initMocks(this);
69         mFragment = spy(WifiScanningRequiredFragment.newInstance());
70         mContext = spy(RuntimeEnvironment.application);
71         when(mContext.getSystemService(WifiManager.class)).thenReturn(mWifiManager);
72         mResolver = mContext.getContentResolver();
73         doReturn(mActivity).when(mFragment).getActivity();
74 
75         doReturn(mContext).when(mFragment).getContext();
76         mFragment.setTargetFragment(mCallbackFragment, 1000);
77         when(mWifiManager.isScanAlwaysAvailable()).thenReturn(false);
78     }
79 
80     @Test
onClick_positiveButtonSetsWifiScanningOn()81     public void onClick_positiveButtonSetsWifiScanningOn()
82             throws Settings.SettingNotFoundException {
83         mFragment.onClick(null, DialogInterface.BUTTON_POSITIVE);
84 
85         verify(mWifiManager).setScanAlwaysAvailable(true);
86     }
87 
88     @Test
onClick_positiveButtonCallsBackToActivity()89     public void onClick_positiveButtonCallsBackToActivity() {
90         mFragment.onClick(null, DialogInterface.BUTTON_POSITIVE);
91 
92         verify(mCallbackFragment).onActivityResult(anyInt(), anyInt(), isNull());
93     }
94 
95     @Test
learnMore_launchesHelpWhenIntentFound()96     public void learnMore_launchesHelpWhenIntentFound() {
97         doReturn("").when(mContext).getString(eq(R.string.help_uri_wifi_scanning_required));
98         mFragment.addButtonIfNeeded(mBuilder);
99         verify(mBuilder, never())
100                 .setNeutralButton(anyInt(), nullable(DialogInterface.OnClickListener.class));
101 
102         doReturn("help").when(mContext).getString(eq(R.string.help_uri_wifi_scanning_required));
103         mFragment.addButtonIfNeeded(mBuilder);
104         verify(mBuilder, times(1))
105                 .setNeutralButton(anyInt(), nullable(DialogInterface.OnClickListener.class));
106     }
107 
108     @Test
learnMore_launchesHelp_shouldStartActivityForResult()109     public void learnMore_launchesHelp_shouldStartActivityForResult() {
110         doReturn(new Intent()).when(mFragment).getHelpIntent(mContext);
111         mFragment.addButtonIfNeeded(mBuilder);
112 
113         mFragment.onClick(null, DialogInterface.BUTTON_NEUTRAL);
114 
115         verify(mActivity, times(1)).startActivityForResult(any(), anyInt());
116     }
117 }
118