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

from behave import *
from cli import Chrome

# ========================== Chrome.start () ==========================
@given ("the Chrome client")
def step_impl (context):
    context.client = Chrome()

@when ("the start method is called")
def step_impl (context):
    context.start_return = context.client.start()

@then ("Google Chrome should open")
def step_impl (context):
    assert context.start_return == "", "Google Chrome did not opened as expected."
    context.client.stop()


# ========================== Chrome.start_headless () ==========================
@when ("the start_headless method is called")
def step_impl (context):
    context.start_headless_return = context.client.start_headless()

@then ("Google Chrome should open headless")
def step_impl (context):
    assert context.start_headless_return == "", "Google Chrome did not opened as expected."
    context.client.stop()


# ========================== Chrome.stop () ==========================
@given ("a Chrome instance")
def step_impl (context):
    context.start_return = context.client.start()

@when ("the stop method is called upon it")
def step_impl (context):
    context.stop_return = context.client.stop()

@then ("this Google Chrome instance should close")
def step_impl (context):
    assert context.stop_return == "", "Google Chrome did not closed as expected."


# ========================== Chrome.new () ==========================