replaycli-py/api/wingui/features/steps/steps.py

from behave import *
from cli import Wingui

# =========================== Clip_read & Clip_write ===========================
@given(u'a winGUI client')
def step_impl(context):
    context.client = Wingui ()

@when(u'the clip_write method is called to store the text "There is a light that never goes out"')
def step_impl(context):
    context.client.clip_write("There is a light that never goes out")

@then(u'when the clip_read method is called, it must return the same text')
def step_impl(context):
    assert context.client.clip_read() == "There is a light that never goes out", "Something went wrong with the clipboard."



# =========================== Screen_scale & Screen_size ===========================
@when(u'the screen_scale method is called')
def step_impl(context):
    context.scale = context.client.screen_scale()

@when(u'the screen_size method is also called')
def step_impl(context):
    context.size = context.client.screen_size()

@then(u'the returns of both must be equal')
def step_impl(context):
    assert context.scale == context.size, "Something went wrong while getting the screen dimensions."

# Please note that most of the other screen methods use image recognition and, because of that, it is not possible to test them on every computer, since there is no way to know wich images are being displayed on the screen.
    # Also, the other methods wich do not use image recognition are simply too difficult to test.



# =========================== Proc_exec & Proc_all ===========================
@when(u'the proc_exec method is called with "mspaint"')
def step_impl(context):
    context.client.proc_exec("mspaint")

@when(u'the proc_all method is called')
def step_impl(context):
    context.procs = context.client.proc_all()

@then(u'there must be a process with the name "mspaint.exe" inside the list')
def step_impl(context):
    for proc in context.procs:
        if "mspaint.exe" in proc["Name"]:
            context.client.proc_kill(proc["Pid"])
            assert True
            return
    
    assert False, "Something went wrong executing Microsoft Paint."



# =========================== Proc_name ===========================
@given(u'the initialization of Microsoft Paint')
def step_impl(context):
    context.client.proc_exec("mspaint")

@given(u'the "Name" and "Pid" infos about this process')
def step_impl(context):
    context.procs = context.client.proc_all()

    for proc in context.procs:
        if "mspaint.exe" in proc["Name"]:
            context.proc_name = proc["Name"]
            context.proc_pid = proc["Pid"]

@when(u'the proc_name method is called with the "Pid" collected')
def step_impl(context):
    context.name_return = context.client.proc_name(context.proc_pid)

@then(u'the return must be te same as the "Name" collected')
def step_impl(context):
    assert context.name_return == context.proc_name, "Something went wrong with some of the proc methods."

    context.client.proc_kill(context.proc_pid)



# =========================== Proc_pids ===========================
@when(u'the proc_pids is called')
def step_impl(context):
    context.all_pids = context.client.proc_pids()

@then(u'the "Pid" info collected must be inside the array returned')
def step_impl(context):
    assert context.proc_pid in context.all_pids, "Something went wrong catching all of the pids."
    
    context.client.proc_kill(context.proc_pid)