Home | History | Annotate | Download | only in stress 
      1 /*
      2  * Copyright (C) 2010, 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.connectivitymanagertest.stress;
     18 
     19 
     20 import com.android.connectivitymanagertest.ConnectivityManagerStressTestRunner;
     21 import com.android.connectivitymanagertest.ConnectivityManagerTestActivity;
     22 
     23 import android.content.Context;
     24 import android.net.wifi.WifiConfiguration;
     25 import android.net.wifi.WifiConfiguration.KeyMgmt;
     26 import android.net.wifi.WifiConfiguration.AuthAlgorithm;
     27 import android.net.wifi.WifiManager;
     28 import android.os.Environment;
     29 import android.os.IPowerManager;
     30 import android.os.PowerManager;
     31 import android.os.ServiceManager;
     32 import android.os.SystemClock;
     33 import android.test.ActivityInstrumentationTestCase2;
     34 import android.test.suitebuilder.annotation.LargeTest;
     35 import android.util.Log;
     36 
     37 import java.io.BufferedWriter;
     38 import java.io.File;
     39 import java.io.FileWriter;
     40 
     41 /**
     42  * Stress the wifi driver as access point.
     43  */
     44 public class WifiApStress
     45     extends ActivityInstrumentationTestCase2<ConnectivityManagerTestActivity> {
     46     private final static String TAG = "WifiApStress";
     47     private static String NETWORK_ID = "AndroidAPTest";
     48     private static String PASSWD = "androidwifi";
     49     private final static String OUTPUT_FILE = "WifiStressTestOutput.txt";
     50     private ConnectivityManagerTestActivity mAct;
     51     private int iterations;
     52     private BufferedWriter mOutputWriter = null;
     53     private int mLastIteration = 0;
     54     private boolean mWifiOnlyFlag;
     55 
     56     public WifiApStress() {
     57         super(ConnectivityManagerTestActivity.class);
     58     }
     59 
     60     @Override
     61     public void setUp() throws Exception {
     62         super.setUp();
     63         mAct = getActivity();
     64         ConnectivityManagerStressTestRunner mRunner =
     65             (ConnectivityManagerStressTestRunner)getInstrumentation();
     66         iterations = mRunner.mSoftapIterations;
     67         mWifiOnlyFlag = mRunner.mWifiOnlyFlag;
     68         mAct.turnScreenOn();
     69     }
     70 
     71     @Override
     72     public void tearDown() throws Exception {
     73         // write the total number of iterations into output file
     74         mOutputWriter = new BufferedWriter(new FileWriter(new File(
     75                 Environment.getExternalStorageDirectory(), OUTPUT_FILE)));
     76         mOutputWriter.write(String.format("iteration %d out of %d\n", mLastIteration, iterations));
     77         mOutputWriter.flush();
     78         mOutputWriter.close();
     79         super.tearDown();
     80     }
     81 
     82     @LargeTest
     83     public void testWifiHotSpot() {
     84         if (mWifiOnlyFlag) {
     85             Log.v(TAG, this.getName() + " is excluded for wi-fi only test");
     86             return;
     87         }
     88         WifiConfiguration config = new WifiConfiguration();
     89         config.SSID = NETWORK_ID;
     90         config.allowedKeyManagement.set(KeyMgmt.WPA_PSK);
     91         config.allowedAuthAlgorithms.set(AuthAlgorithm.OPEN);
     92         config.preSharedKey = PASSWD;
     93 
     94         // If Wifi is enabled, disable it
     95         if (mAct.mWifiManager.isWifiEnabled()) {
     96             mAct.disableWifi();
     97         }
     98         int i;
     99         for (i = 0; i < iterations; i++) {
    100             Log.v(TAG, "iteration: " + i);
    101             mLastIteration = i;
    102             // enable Wifi tethering
    103             assertTrue(mAct.mWifiManager.setWifiApEnabled(config, true));
    104             // Wait for wifi ap state to be ENABLED
    105             assertTrue(mAct.waitForWifiAPState(WifiManager.WIFI_AP_STATE_ENABLED,
    106                     2 * ConnectivityManagerTestActivity.LONG_TIMEOUT));
    107             // Wait for wifi tethering result
    108             assertEquals(ConnectivityManagerTestActivity.SUCCESS,
    109                     mAct.waitForTetherStateChange(2*ConnectivityManagerTestActivity.SHORT_TIMEOUT));
    110             // Allow the wifi tethering to be enabled for 10 seconds
    111             try {
    112                 Thread.sleep(2 * ConnectivityManagerTestActivity.SHORT_TIMEOUT);
    113             } catch (Exception e) {
    114                 fail("thread in sleep is interrupted");
    115             }
    116             assertTrue("no uplink data connection after Wi-Fi tethering", mAct.pingTest(null));
    117             // Disable soft AP
    118             assertTrue(mAct.mWifiManager.setWifiApEnabled(config, false));
    119             // Wait for 30 seconds until Wi-Fi tethering is stopped
    120             try {
    121                 Thread.sleep(30 * 1000);
    122                 Log.v(TAG, "wait for Wi-Fi tethering to be disabled.");
    123             } catch (Exception e) {
    124                 fail("thread in sleep is interrupted");
    125             }
    126             assertFalse("Wi-Fi AP disable failed", mAct.mWifiManager.isWifiApEnabled());
    127         }
    128         if (i == iterations) {
    129             mLastIteration = iterations;
    130         }
    131     }
    132 
    133 }
    134