tests/hwsim/test_p2p_grpform.py

   1 #!/usr/bin/python
   2 #
   3 # P2P group formation test cases
   4 # Copyright (c) 2013, Jouni Malinen <j@w1.fi>
   5 #
   6 # This software may be distributed under the terms of the BSD license.
   7 # See README for more details.
   8
   9 import logging
  10 logger = logging.getLogger(__name__)
  11
  12 import hwsim_utils
  13
  14 def go_neg_pin_authorized(i_dev, r_dev, i_intent=None, r_intent=None, expect_failure=False, i_go_neg_status=None):
  15     r_dev.p2p_listen()
  16     i_dev.p2p_listen()
  17     pin = r_dev.wps_read_pin()
  18     logger.info("Start GO negotiation " + i_dev.ifname + " -> " + r_dev.ifname)
  19     r_dev.p2p_go_neg_auth(i_dev.p2p_dev_addr(), pin, "display", go_intent=r_intent)
  20     i_res = i_dev.p2p_go_neg_init(r_dev.p2p_dev_addr(), pin, "enter", timeout=15, go_intent=i_intent, expect_failure=expect_failure)
  21     r_res = r_dev.p2p_go_neg_auth_result(expect_failure=expect_failure)
  22     logger.debug("i_res: " + str(i_res))
  23     logger.debug("r_res: " + str(r_res))
  24     r_dev.dump_monitor()
  25     i_dev.dump_monitor()
  26     if i_go_neg_status:
  27         if i_res['result'] != 'go-neg-failed':
  28             raise Exception("Expected GO Negotiation failure not reported")
  29         if i_res['status'] != i_go_neg_status:
  30             raise Exception("Expected GO Negotiation status not seen")
  31     if expect_failure:
  32         return
  33     logger.info("Group formed")
  34     hwsim_utils.test_connectivity_p2p(r_dev, i_dev)
  35
  36 def test_grpform(dev):
  37     go_neg_pin_authorized(i_dev=dev[0], i_intent=15, r_dev=dev[1], r_intent=0)
  38     dev[0].remove_group()
  39     try:
  40         dev[1].remove_group()
  41     except:
  42         pass
  43
  44 def test_grpform2(dev):
  45     go_neg_pin_authorized(i_dev=dev[0], i_intent=0, r_dev=dev[1], r_intent=15)
  46     dev[0].remove_group()
  47     try:
  48         dev[1].remove_group()
  49     except:
  50         pass
  51
  52 def test_both_go_intent_15(dev):
  53     go_neg_pin_authorized(i_dev=dev[0], i_intent=15, r_dev=dev[1], r_intent=15, expect_failure=True, i_go_neg_status=9)
  54
  55 def add_tests(tests):
  56     tests.append(test_grpform)
  57     tests.append(test_grpform2)
  58     tests.append(test_both_go_intent_15)