client/site_tests/network_DefaultProfileServices/network_DefaultProfileServices.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 os

from autotest_lib.client.bin import test
from autotest_lib.client.common_lib import error
from autotest_lib.client.common_lib import utils

# This hacks the path so that we can import shill_proxy.
# pylint: disable=W0611
from autotest_lib.client.cros import flimflam_test_path
# pylint: enable=W0611
import shill_proxy

class network_DefaultProfileServices(test.test):
    """The Default Profile Services class.

    Wipe the default profile, start shill, configure a service, restart
    shill, and check that the service exists.

    The service name is chosen such that it is unlikely match any SSID
    that is present over-the-air.

    """
    DEFAULT_PROFILE_PATH = '/var/cache/shill/default.profile'
    OUR_SSID = 'org.chromium.DfltPrflSrvcsTest'
    version = 1


    def run_once(self):
        """Test main loop."""
        utils.run('stop shill')
        os.remove(self.DEFAULT_PROFILE_PATH)
        utils.run('start shill')
        shill = shill_proxy.ShillProxy.get_proxy()
        if shill is None:
            raise error.TestFail('Could not connect to shill')

        shill.manager.PopAllUserProfiles()
        path = shill.manager.ConfigureService({
                shill.SERVICE_PROPERTY_TYPE: 'wifi',
                shill.SERVICE_PROPERTY_MODE: 'managed',
                shill.SERVICE_PROPERTY_SSID: self.OUR_SSID,
                shill.SERVICE_PROPERTY_HIDDEN: True,
                shill.SERVICE_PROPERTY_SECURITY: 'none',
                })

        utils.run('stop shill')
        utils.run('start shill')
        shill = shill_proxy.ShillProxy.get_proxy()
        if shill is None:
            raise error.TestFail('Could not connect to shill')

        shill.manager.PopAllUserProfiles()
        service = shill.find_object('AnyService',
                                    {'Name': self.OUR_SSID})
        if not service:
            raise error.TestFail('Network not found after restart.')