tests/remote/config.py

   1 # Environment configuration
   2 # Copyright (c) 2016, Tieto Corporation
   3 #
   4 # This software may be distributed under the terms of the BSD license.
   5 # See README for more details.
   6
   7 #
   8 # Currently static definition, in the future this could be a config file,
   9 # or even common database with host management.
  10 #
  11
  12 import logging
  13 logger = logging.getLogger()
  14
  15 #
  16 # You can put your settings in cfg.py file with setup_params, devices
  17 # definitions in the format as below. In other case HWSIM cfg will be used.
  18 #
  19 setup_params = { "setup_hw" : "./tests/setup_hw.sh",
  20                  "hostapd" : "./tests/hostapd",
  21                  "wpa_supplicant" : "./tests/wpa_supplicant",
  22                  "iperf" : "iperf",
  23                  "wlantest" : "./tests/wlantest",
  24                  "wlantest_cli" : "./tests/wlantest_cli",
  25                  "country" : "US",
  26                  "log_dir" : "/tmp/",
  27                  "ipv4_test_net" : "192.168.12.0",
  28                  "trace_start" : "./tests/trace_start.sh",
  29                  "trace_stop" : "./tests/trace_stop.sh",
  30                  "perf_start" : "./tests/perf_start.sh",
  31                  "perf_stop" : "./tests/perf_stop.sh" }
  32
  33 #
  34 #devices = [{"hostname": "192.168.254.58", "ifname" : "wlan0", "port": "9877", "name" : "t2-ath9k", "flags" : "AP_HT40 STA_HT40"},
  35 #           {"hostname": "192.168.254.58", "ifname" : "wlan1", "port": "9877", "name" : "t2-ath10k", "flags" : "AP_VHT80"},
  36 #           {"hostname": "192.168.254.58", "ifname" : "wlan3", "port": "9877", "name" : "t2-intel7260", "flags" : "STA_VHT80"},
  37 #           {"hostname": "192.168.254.55", "ifname" : "wlan0, wlan1, wlan2", "port": "", "name" : "t3-monitor"},
  38 #           {"hostname": "192.168.254.50", "ifname" : "wlan0", "port": "9877", "name" : "t1-ath9k"},
  39 #           {"hostname": "192.168.254.50", "ifname" : "wlan1", "port": "9877", "name" : "t1-ath10k"}]
  40
  41 #
  42 # HWSIM - ifaces available after modprobe mac80211_hwsim
  43 #
  44 devices = [{"hostname": "localhost", "ifname": "wlan0", "port": "9868", "name": "hwsim0", "flags": "AP_VHT80 STA_VHT80"},
  45            {"hostname": "localhost", "ifname": "wlan1", "port": "9878", "name": "hwsim1", "flags": "AP_VHT80 STA_VHT80"},
  46            {"hostname": "localhost", "ifname": "wlan2", "port": "9888", "name": "hwsim2", "flags": "AP_VHT80 STA_VHT80"},
  47            {"hostname": "localhost", "ifname": "wlan3", "port": "9898", "name": "hwsim3", "flags": "AP_VHT80 STA_VHT80"},
  48            {"hostname": "localhost", "ifname": "wlan4", "port": "9908", "name": "hwsim4", "flags": "AP_VHT80 STA_VHT80"}]
  49
  50
  51 def get_setup_params(filename="cfg.py"):
  52     try:
  53        mod = __import__(filename.split(".")[0])
  54        return mod.setup_params
  55     except:
  56        logger.debug("__import__(" + filename + ") failed, using static settings")
  57        pass
  58     return setup_params
  59
  60 def get_devices(filename="cfg.py"):
  61     try:
  62        mod = __import__(filename.split(".")[0])
  63        return mod.devices
  64     except:
  65        logger.debug("__import__(" + filename + ") failed, using static settings")
  66        pass
  67     return devices
  68
  69 def get_device(devices, name=None, flags=None, lock=False):
  70     if name is None and flags is None:
  71         raise Exception("Failed to get device")
  72     for device in devices:
  73         if device['name'] == name:
  74             return device
  75     for device in devices:
  76         try:
  77             device_flags = device['flags']
  78             if device_flags.find(flags) != -1:
  79                 return device
  80         except:
  81             pass
  82     raise Exception("Failed to get device " + name)
  83
  84 def put_device(devices, name):
  85     pass