server/site_tests/network_WiFiChaosPSK/network_WiFiChaosPSK.py

# Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import logging

from autotest_lib.server import autotest, test


class network_WiFiChaosPSK(test.test):
    """Tests connecting to APs using PSK security setting."""

    version = 1


    def run_once(self, host, helper, ap_info, tries=1):
        """ Main entry function for autotest.

        @param host: an Autotest host object, DUT.
        @param helper: a WiFiChaosConnectionTest object, ready to use.
        @param ap_info: a dict of attributes of a specific AP.
        @param tries: an integer, number of connection attempts.
        """
        # Override PSK password in base helper class
        helper.psk_password = 'chromeos'

        # Install all of the autotest libraries on the client
        client_at = autotest.Autotest(host)
        client_at.install()

        helper.run_ap_test(ap_info, tries, self.outputdir)

        logging.info('Client test complete, powering down router.')
        helper.power_down(ap_info['configurator'])
        helper.check_test_error()