def __init__(self,
enable,
security=None,
ssh_host="192.168.1.1",
ssh_user="******",
encryption=None,
wifi_password=None,
radius_ip=None,
radius_secret=None,
hidden_ssid="0",
new_ssid=None,
ssh_pwd=None,
interface5ghz='0',
**kwargs):
self.enable = enable
self.hidden_ssid = hidden_ssid
self.new_ssid = new_ssid
self.security = security
self.encryption = encryption
self.wifi_password = wifi_password
self.radius_ip = radius_ip
self.radius_secret = radius_secret
self.ssh_host = ssh_host
self.ssh_user = ssh_user
self.ssh_pwd = ssh_pwd
self.interface5ghz = interface5ghz
base_step.__init__(self, **kwargs)
def __init__(self, device, platform, version=None, **kwargs):
self.relay = Relayed_device(kb_relay_port=device['kb']['relay_tty'],
kb_port=device['kb']['kb_tty'])
self.kb_device_port = device['kb']['port']
self.platform = platform
self.version = version
base_step.__init__(self, **kwargs)
def __init__(self, **kwargs):
base_step.__init__(self, **kwargs)
self.serial = None
self.no_log = False
if "serial" in kwargs:
self.serial = kwargs["serial"]
if "platform" in kwargs:
self.platform = kwargs["platform"]
else:
self.platform = None
if "dessert" in kwargs:
self.dessert = kwargs["dessert"]
else:
self.dessert = None
if "no_log" in kwargs:
self.no_log = kwargs["no_log"]
# if self.no_log is True:
# self.device_info = None
# else
self.device_info = statics.Device(serial=self.serial, platform=self.platform, dessert=self.dessert)
if self.serial not in self.passm:
self.passm = self.passm.replace(" - [PASSED]", " [ {} ] - [PASSED]".format(self.serial))
if self.serial not in self.errorm:
self.errorm = self.errorm.replace(" - [FAILED]", " [ {} ] - [FAILED]".format(self.serial))
def __init__(self, **kwargs):
base_step.__init__(self, **kwargs)
print "Calling mock step {0} with the following parameters: ".format(
self.__class__.__name__)
for key in kwargs:
print "{0} = {1}".format(key, kwargs[key])
if not kwargs:
print "No parameters ware given"
def __init__(self, ssh_host="192.168.1.1", ssh_user="******", **kwargs):
self.radvd_conf_file = os.path.join(
os.path.dirname(os.path.realpath(__file__)), RADVD_CONF_FILE)
self.remote_file = RADVD_CONF_FILE
self.ssh_host = ssh_host
self.ssh_user = ssh_user
self.step_data = None
base_step.__init__(self, **kwargs)
def __init__(self, url, args="", grep_for=None, **kwargs):
self.command = "curl {0} {1}".format(args, url)
self.grep_for = grep_for
base_step.__init__(self, **kwargs)
if self.grep_for:
self.set_passm("Curl for {0} checking \'{1}\'".format(url, self.grep_for))
self.set_errorm("", "Failed curl for {0} checking \'{1}\'".format(url, self.grep_for))
else:
self.set_passm("Curl for {0}".format(url))
self.set_errorm("", "Failed curl for {0}".format(url))
def __init__(self, first_file, second_file, tolerance=0, **kwargs):
self.first_file = first_file
self.second_file = second_file
self.diff = tolerance
base_step.__init__(self, **kwargs)
self.set_passm("Comparing {0} and {1} with {2} tolerance".format(
first_file, second_file, self.diff))
self.set_errorm(
"", "Comparing {0} and {1} with {2} tolerance".format(
first_file, second_file, self.diff))
def __init__(self,
ipv6_ip,
ssh_host="192.168.1.1",
ssh_user="******",
**kwargs):
self.ipv6_ip = ipv6_ip
self.ssh_host = ssh_host
self.ssh_user = ssh_user
self.step_data = None
base_step.__init__(self, **kwargs)
def __init__(self,
config,
value,
ssh_host,
ssh_user,
ssh_pwd=None,
**kwargs):
self.config = config
self.value = value
self.ssh_host = ssh_host
self.ssh_user = ssh_user
self.ssh_pwd = ssh_pwd
base_step.__init__(self, **kwargs)
def __init__(self,
url,
args,
file_name,
version,
delta=0,
success_mess="OK",
**kwargs):
self.url = url
self.args = args
self.file_name = file_name
self.version = version
self.delta = delta
self.ok = success_mess
base_step.__init__(self, **kwargs)
def __init__(self, device, platform="ECS27B", **kwargs):
self.relay = Relayed_device(kb_relay_port=device['kb']['relay_tty'],
kb_port=device['kb']['kb_tty'])
self.touch_relay = Relayed_device(
relay_port=device["relay"]["tty"],
power_port=device["relay"]["power_port"],
v_up_port=device["relay"]["v_up_port"],
v_down_port=device["relay"]["v_down_port"],
always_allow_port=device["relay"]["always_allow_port"],
allow_ok_port=device["relay"]["allow_ok_port"],
)
self.kb_device_port = device['kb']['port']
self.device = device
self.allowed = False
self.platform = platform
base_step.__init__(self, **kwargs)
def __init__(self, **kwargs):
base_step.__init__(self, **kwargs)
self.relay = Relayed_device( # "RLY08B" or "URMC32"
relay_type=kwargs["relay_type"],
# COM port used for communication. Ex: "/dev/ttyACM0"
relay_port=kwargs["relay_port"]
if "relay_port" in kwargs else None,
# number of the relay connected to POWER button
power_port=kwargs["power_port"]
if "power_port" in kwargs else None,
# number of the relay connected to VOL UP button
v_up_port=kwargs["v_up_port"] if "v_up_port" in kwargs else None,
# number of the relay connected to VOL DOWN button
v_down_port=kwargs["v_down_port"]
if "v_down_port" in kwargs else None,
# number of the relay used to cut VC+ of the USB cable connected to the device
USB_VC_cut_port=kwargs["USB_VC_cut_port"]
if "USB_VC_cut_port" in kwargs else None)
def __init__(self,
host,
user,
password,
recipients,
subject,
message=None,
file_path=None,
html_content_type=False,
**kwargs):
self.host = host
self.user = user
self.passwd = password
self.recipients = recipients
self.subject = subject
self.message = message
self.file_path = file_path
self.html = html_content_type
base_step.__init__(self, **kwargs)
def __init__(self,
sender,
recipient,
subject,
body_text=None,
body_html=None,
smtp_server="localhost",
**kwargs):
self.sender = sender
self.recipient = recipient
self.smtp_server = smtp_server
self.subject = subject
self.text = body_text
self.html = body_html
base_step.__init__(self, **kwargs)
self.set_passm("E-mail with '{0}' subject sent to {1}".format(
self.subject, recipient))
self.set_errorm(
"", "Error sending e-mail with '{0}' subject sent to {1}".format(
self.subject, recipient))
def __init__(self,
mode,
security,
ssh_host='192.168.1.1',
ssh_user="******",
encryption=None,
wifi_password=None,
radius_ip=None,
radius_secret=None,
hidden_ssid="0",
new_ssid=None,
channel_bw="20",
channel_no=None,
ssh_pwd=None,
interface5ghz='0',
ipv6_enable=0,
radvd_enable=0,
dhcp_lease=None,
txpwr=None,
ipv4_class='192.168.1.1',
**kwargs):
self.hidden_ssid = hidden_ssid
self.new_ssid = new_ssid
self.mode = mode
self.security = security
self.encryption = encryption
self.wifi_password = wifi_password
self.radius_ip = radius_ip
self.radius_secret = radius_secret
self.ssh_host = ssh_host
self.ssh_user = ssh_user
self.ssh_pwd = ssh_pwd
self.channel_bw = channel_bw
self.channel_no = channel_no
self.interface5ghz = interface5ghz
self.ipv6_enable = ipv6_enable
self.radvd_enable = radvd_enable
self.dhcp_lease = dhcp_lease
self.txpwr = txpwr
self.ipv4_class = ipv4_class
base_step.__init__(self, **kwargs)
def __init__(self, **kwargs):
base_step.__init__(self, **kwargs)
self.available_states_to_config = {
'1_am': self.config_am,
'2_wifi_connected': self.config_wifi_connected,
'3_p2p_connected': self.config_p2p_connected,
'4_wifi_traffic': self.config_wifi_traffic,
'5_p2p_traffic': self.config_p2p_traffic,}
self.scenario_action = {"Owner-Turn-WiFi-OFF": self.turn_wifi_off,
"CLI-Turn-WiFi-OFF": self.turn_wifi_off,
"Owner-Disconnect-Wifi": self.disconnect_wifi,
"CLI-Disconnect-Wifi": self.disconnect_wifi,
"Owner-Deactivate-AM": self.deactivate_am,
"CLI-Deactivate-AM": self.deactivate_am,
"Owner-Activate-AM": self.activate_am,
"CLI-Activate-AM": self.activate_am,
"CLI-Leave-Group": self.leave_group,
"Owner-Dismiss-CLI": self.dismiss_cli,
"Owner-Remove-Group": self.remove_group,
"Owner-Generate-P2P-Traffic": self.generate_p2p_traffic,
"CLI-Generate-P2P-Traffic": self.generate_p2p_traffic,
"CLI-Connect-P2P": self.connect_p2p,
"CLI-Connect-WiFi": self.connect_wifi,
"Owner-Connect-WiFi": self.connect_wifi,
"Owner-Reject-Connect": self.reject_connect,
"Owner-Accept-Connect": self.accept_connect,
"CLI-Generate-WiFi-Traffic": self.generate_wifi_traffic,
"Owner-Generate-WiFi-Traffic": self.generate_wifi_traffic,
"CLI-Scan-WiFi": self.scan_wifi,
"Owner-Scan-WiFi": self.scan_wifi,
"CLI-Connect-timeout": self.connect_timeout,
"Owner-Connect-timeout": self.connect_timeout,
"CLI-Reconnect-Persistent-P2P": self.reconnect_persistent,
"Owner-Reconnect-Persistent-P2P": self.reconnect_persistent,
"Owner-P2P-Twoway-Traffic": self.config_p2p_twoway_traffic
}
self.scenario = scenario
def __init__(self, screen_name, with_log=False, **kwargs):
self.screen_name = screen_name
self.with_log = with_log
base_step.__init__(self, **kwargs)
def __init__(self, serial, image_path, build_no, timeout, **kwargs):
self.serial = serial
self.image_path = image_path
self.build_no = build_no
self.timeout = timeout
base_step.__init__(self, **kwargs)
def __init__(self, view, view_index=0, radio_index=0, **kwargs):
self.view = view
self.radio_instance = radio_index
self.view_instance = view_index
base_step.__init__(self, **kwargs)
def __init__(self, device, **kwargs):
self.relay = Relayed_device(kb_relay_port=device['kb']['relay_tty'],
kb_port=device['kb']['kb_tty'])
self.kb_device_port = device['kb']['port']
base_step.__init__(self, **kwargs)
def __init__(self, info_message):
"""
:param info_message: info message to be logged
"""
base_step.__init__(self)
self._info_message = info_message
def __init__(self, **kwargs):
base_step.__init__(self, **kwargs)
def __init__(self, ssh_host="192.168.1.1", ssh_user="******", **kwargs):
base_step.__init__(self, **kwargs)
def __init__(self, screen_name, **kwargs):
self.screen_name = screen_name
base_step.__init__(self, **kwargs)
def __init__(self, sleep_time=1, **kwargs):
self.ssh_connection = connection_ssh(**kwargs)
base_step.__init__(self, **kwargs)
def __init__(self, db_prop_list, env_prop_list, **kwargs):
self.queries = db_prop_list
self.props = env_prop_list
base_step.__init__(self, **kwargs)
def __init__(self, **kwargs):
self.local_connection = connection_local(**kwargs)
base_step.__init__(self, **kwargs)
def __init__(self, db_prop_list, env_prop_list, ip, **kwargs):
self.db_prop_list = db_prop_list
self.env_prop_list = env_prop_list
self.ip = ip
self.set_errorm("", "After flash operations could not be performed")
base_step.__init__(self, **kwargs)
