Home | History | Annotate | Download | only in setup 
      1 /*
      2  * Copyright (C) 2017 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.tv.settings.connectivity.setup;
     18 
     19 import static org.mockito.Mockito.verify;
     20 import static org.robolectric.shadow.api.Shadow.extract;
     21 
     22 import android.arch.lifecycle.ViewModelProviders;
     23 import android.net.wifi.WifiConfiguration;
     24 
     25 import com.android.settingslib.wifi.AccessPoint;
     26 import com.android.tv.settings.TvSettingsRobolectricTestRunner;
     27 import com.android.tv.settings.connectivity.util.State;
     28 import com.android.tv.settings.connectivity.util.StateMachine;
     29 import com.android.tv.settings.testutils.ShadowStateMachine;
     30 
     31 import org.junit.Before;
     32 import org.junit.Test;
     33 import org.junit.runner.RunWith;
     34 import org.mockito.Mock;
     35 import org.mockito.MockitoAnnotations;
     36 import org.robolectric.Robolectric;
     37 import org.robolectric.annotation.Config;
     38 
     39 @RunWith(TvSettingsRobolectricTestRunner.class)
     40 @Config(shadows = ShadowStateMachine.class)
     41 public class AddStartStateTest {
     42     private WifiSetupActivity mActivity;
     43     private AddStartState mAddStartState;
     44     private UserChoiceInfo mUserChoiceInfo;
     45     @Mock
     46     private State.StateCompleteListener mStateCompleteListener;
     47 
     48     @Before
     49     public void setUp() {
     50         MockitoAnnotations.initMocks(this);
     51         mActivity = Robolectric.buildActivity(WifiSetupActivity.class).get();
     52         StateMachine stateMachine = ViewModelProviders.of(mActivity).get(StateMachine.class);
     53         mUserChoiceInfo = ViewModelProviders.of(mActivity).get(UserChoiceInfo.class);
     54         ShadowStateMachine shadowStateMachine = (ShadowStateMachine) extract(stateMachine);
     55         shadowStateMachine.setListener(mStateCompleteListener);
     56         mAddStartState = new AddStartState(mActivity);
     57     }
     58 
     59     @Test
     60     public void testForward_WEP_NeedPassword() {
     61         mUserChoiceInfo.init();
     62         mUserChoiceInfo.setWifiSecurity(AccessPoint.SECURITY_WEP);
     63         mUserChoiceInfo.setWifiConfiguration(new WifiConfiguration());
     64         mAddStartState.processForward();
     65         verify(mStateCompleteListener).onComplete(StateMachine.PASSWORD);
     66     }
     67 
     68     @Test
     69     public void testForward_WPA_NeedPassword() {
     70         mUserChoiceInfo.init();
     71         mUserChoiceInfo.setWifiSecurity(AccessPoint.SECURITY_PSK);
     72         mUserChoiceInfo.setWifiConfiguration(new WifiConfiguration());
     73         mAddStartState.processForward();
     74         verify(mStateCompleteListener).onComplete(StateMachine.PASSWORD);
     75     }
     76 
     77     @Test
     78     public void testForward_EAP_NeedPassword() {
     79         mUserChoiceInfo.init();
     80         mUserChoiceInfo.setWifiSecurity(AccessPoint.SECURITY_EAP);
     81         mUserChoiceInfo.setWifiConfiguration(new WifiConfiguration());
     82         mAddStartState.processForward();
     83         verify(mStateCompleteListener).onComplete(StateMachine.PASSWORD);
     84     }
     85 
     86     @Test
     87     public void testForward_AlreadyHasPassword() {
     88         mUserChoiceInfo.init();
     89         mUserChoiceInfo.setWifiSecurity(AccessPoint.SECURITY_PSK);
     90         WifiConfiguration config = new WifiConfiguration();
     91         config.preSharedKey = "PasswordTest";
     92         mUserChoiceInfo.setWifiConfiguration(config);
     93         mAddStartState.processForward();
     94         verify(mStateCompleteListener).onComplete(StateMachine.CONNECT);
     95     }
     96 
     97     @Test
     98     public void testForward_DoNotNeedPassword() {
     99         mUserChoiceInfo.init();
    100         mUserChoiceInfo.setWifiSecurity(AccessPoint.SECURITY_NONE);
    101         mAddStartState.processForward();
    102         verify(mStateCompleteListener).onComplete(StateMachine.CONNECT);
    103     }
    104 }
    105