# Copyright (c) 2018 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.

from autotest_lib.server import autotest
from autotest_lib.server import site_linux_system
from autotest_lib.server.cros.network import hostap_config
from autotest_lib.server.cros.network import wifi_cell_test_base


class policy_WiFiAutoconnectServer(wifi_cell_test_base.WiFiCellTestBase):
    version = 1


    def parse_additional_arguments(self, commandline_args, additional_params):
        """
        Hook into super class to take control files parameters.

        @param commandline_args: dict of parsed parameters from the autotest.
        @param additional_params: HostapConfig object.

        """
        self._ap_config = additional_params


    def run_once(self, host, autoconnect=None):
        """
        Set up AP then run the client side autoconnect tests.

        @param host: A host object representing the DUT.
        @param autoconnect: Autoconnect setting for network policy.

        """
        self.context.router.require_capabilities(
                [site_linux_system.LinuxSystem.CAPABILITY_MULTI_AP])
        self.context.router.deconfig()
        self.context.configure(self._ap_config, multi_interface=True)

        client_at = autotest.Autotest(host)

        client_at.run_test('policy_WiFiAutoconnect',
                           ssid=self.context.router.get_ssid(instance=0),
                           autoconnect=autoconnect,
                           check_client_result=True)

        self.context.router.deconfig()