#!/usr/bin/env python3
#
#   Copyright 2018 - The Android Open Source Project
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.

import acts_contrib.test_utils.wifi.wifi_test_utils as wutils
import acts.utils
import time

from acts import asserts
from acts import utils

from acts.test_decorators import test_tracker_info
from acts_contrib.test_utils.wifi.p2p.WifiP2pBaseTest import WifiP2pBaseTest
from acts_contrib.test_utils.wifi.p2p import wifi_p2p_test_utils as wp2putils
from acts_contrib.test_utils.wifi.p2p import wifi_p2p_const as p2pconsts

WPS_PBC = wp2putils.WifiP2PEnums.WpsInfo.WIFI_WPS_INFO_PBC
WPS_DISPLAY = wp2putils.WifiP2PEnums.WpsInfo.WIFI_WPS_INFO_DISPLAY
WPS_KEYPAD = wp2putils.WifiP2PEnums.WpsInfo.WIFI_WPS_INFO_KEYPAD


class WifiP2pGroupTest(WifiP2pBaseTest):
    """Tests for APIs in Android's WifiP2pManager class.

    Test Bed Requirement:
    * At least two Android devices
    """
    def __init__(self, controllers):
        WifiP2pBaseTest.__init__(self, controllers)

    def setup_class(self):
        super().setup_class()
        if not "network_name" in self.user_params.keys():
            self.log.error("Missing mandatory user config \"network_name\"!")
        self.network_name = self.user_params["network_name"]
        if not "passphrase" in self.user_params.keys():
            self.log.error("Missing mandatory user config \"passphrase\"!")
        self.passphrase = self.user_params["passphrase"]
        if not "group_band" in self.user_params.keys():
            self.log.error("Missing mandatory user config \"group_band\"!")
        self.group_band = self.user_params["group_band"]

    def setup_test(self):
        super().setup_test()
        self.dut1.droid.wifiP2pRemoveGroup()
        self.dut2.droid.wifiP2pRemoveGroup()
        time.sleep(p2pconsts.DEFAULT_FUNCTION_SWITCH_TIME)

    def teardown_test(self):
        self.dut1.droid.wifiP2pRemoveGroup()
        self.dut2.droid.wifiP2pRemoveGroup()
        time.sleep(p2pconsts.DEFAULT_FUNCTION_SWITCH_TIME)
        super().teardown_test()

    def p2p_group_join(self, wps_type):
        """ General flow for p2p group join

        Steps:
        1. GO creates a group.
        2. GC joins the group.
        3. connection check via ping from GC to GO
        """
        go_dut = self.dut1
        gc_dut = self.dut2
        # Create a group
        wp2putils.p2p_create_group(go_dut)
        go_dut.ed.pop_event(p2pconsts.CONNECTED_EVENT,
                            p2pconsts.DEFAULT_TIMEOUT)
        time.sleep(p2pconsts.DEFAULT_FUNCTION_SWITCH_TIME)
        # Request the connection
        wp2putils.p2p_connect(gc_dut,
                              go_dut,
                              False,
                              wps_type,
                              p2p_connect_type=p2pconsts.P2P_CONNECT_JOIN)

        go_ip = wp2putils.p2p_go_ip(gc_dut)
        wp2putils.p2p_connection_ping_test(gc_dut, go_ip)
        # trigger disconnect
        wp2putils.p2p_disconnect(go_dut)
        wp2putils.check_disconnect(gc_dut)
        time.sleep(p2pconsts.DEFAULT_FUNCTION_SWITCH_TIME)

    """Test Cases"""
    @test_tracker_info(uuid="c41f8293-5225-430d-917e-c294ddff7c2a")
    def test_p2p_group_join_via_pbc(self):
        """Verify the p2p creates a group and join this group via WPS PBC method.
        """
        self.p2p_group_join(WPS_PBC)

    @test_tracker_info(uuid="56eb339f-d7e4-44f0-9802-6094e9255957")
    def test_p2p_group_join_via_display(self):
        """Verify the p2p creates a group and join this group via WPS DISPLAY method.
        """
        self.p2p_group_join(WPS_DISPLAY)

    @test_tracker_info(uuid="27075cab-7859-49a7-afe9-b6cc6e8faddb")
    def test_p2p_group_with_config(self):
        """Verify the p2p creates a group and join an this group with config.

        Steps:
        1. GO creates a group with config.
        2. GC joins the group with config.
        3. connection check via ping from GC to GO
        """
        go_dut = self.dut1
        gc_dut = self.dut2
        # Create a group
        wp2putils.p2p_create_group_with_config(go_dut, self.network_name,
                                               self.passphrase,
                                               self.group_band)
        time.sleep(p2pconsts.DEFAULT_FUNCTION_SWITCH_TIME)
        # Request the connection. Since config is known, this is reconnection.
        wp2putils.p2p_connect_with_config(gc_dut, go_dut, self.network_name,
                                          self.passphrase, self.group_band)

        go_ip = wp2putils.p2p_go_ip(gc_dut)
        wp2putils.p2p_connection_ping_test(gc_dut, go_ip)
        # trigger disconnect
        wp2putils.p2p_disconnect(go_dut)
        wp2putils.check_disconnect(gc_dut)
        time.sleep(p2pconsts.DEFAULT_FUNCTION_SWITCH_TIME)