from behave import *
from cli import Replay

from datetime import datetime

# ============================= ENVIRONMENT VARIABLES GETTERS =============================
@given(u'the Replay client')
def step_impl (context):
    context.client = Replay ()

@when(u'each one of the getters is called')
def step_impl(context):
    context.returns = {context.client.replay_env_queue_id(): "context.client.replay_env_queue_id()", context.client.replay_env_alias(): "context.client.replay_env_alias()", context.client.replay_env_feature_dir(): "context.client.replay_env_feature_dir()", context.client.replay_env_root(): "context.client.replay_env_root()", context.client.replay_env_addr(): "context.client.replay_env_addr()", context.client.replay_env_home_dir(): "context.client.replay_env_home_dir()", context.client.replay_env_repo(): "context.client.replay_env_repo()", context.client.replay_env_ver(): "context.client.replay_env_ver()", context.client.replay_env_repo_dir(): "context.client.replay_env_repo_dir()", context.client.replay_env_data_dir(): "context.client.replay_env_data_dir()", context.client.replay_env_instance_alias(): "context.client.replay_env_instance_alias()", context.client.replay_env_api_key(): "context.client.replay_env_api_key()"}

@then(u'all of the returns must be either of the type string or the type None')
def step_impl(context):
    for ret in context.returns.keys():
        assert ( type(ret) == type(None) or type(ret) == type("") ), f'Something went wrong with the {context.returns[ret]} variable getter.'



# ============================= LOG REGISTERS =============================
@given(u'a Log created')
def step_impl(context):
    context.log = context.client.new_log()

@when(u'the log method is called')
def step_impl(context):
    context.dt_log_end = datetime.now().isoformat() + "Z"
    context.log_return = context.client.log(context.log)

@then(u'this new log must be registered in the Replay Data Base')
def step_impl(context):
    assert (context.client.sql("select * from logs")["data"][-1]["dtlogend"] == context.dt_log_end and context.log_return == "ok")