feat: improve expected/actual payload when comparing data in checks

from dataclasses import dataclass

from deepdiff import DeepDiff

from prettydiff import get_annotated_lines_from_diff, diff_json, Flag

from robot.api import logger

@dataclass

class Theme:

    added: str

    removed: str

    reset: str

def compare_dictionaries_ignoring_keys(expected, actual, exclude_regex_paths, group_by=None):

                       group_by=group_by)

    else:

        res = DeepDiff(expected, actual, exclude_regex_paths=exclude_regex_paths, ignore_order=True, verbose_level=1)

    if len(res) > 0:

        output_pretty_diff(expected, actual, Theme(added="", removed="", reset=""))

    return res

def output_pretty_diff(a, b, theme, indent_size: int = 2, console=False):

    logger.info("Dictionary comparison failed with -> ", also_console=True)

    lines = get_annotated_lines_from_diff(diff_json(a, b))

    msg = ""

    for line in lines:

        if Flag.ADDED in line.flags:

            flags = f"{theme.added}+ "

        elif Flag.REMOVED in line.flags:

            flags = f"{theme.removed}- "

        else:

            flags = f"{theme.reset} "

        msg = msg + flags + " " * (indent_size * line.indent) + line.s + "\n"

    logger.info(msg, also_console=True)

from __future__ import unicode_literals

from __future__ import division

from pygments import highlight, lexers, formatters

from json import dumps, JSONDecodeError, loads

from robot.api import logger

from robot.api.deco import keyword

@keyword(name="Output", tags=("I/O",))

def output(response, console=True):

def output(response, description, console=True):

    """*Request and response are output to terminal and file (in JSON).*

    :param response: response to a request

    :param description: explains what request is being made

    :param console: If false, the JSON is not written to terminal. Default is true.

    """

    pretty_request_json = dumps(request_json, indent=4, sort_keys=False, separators=(",", ": "))

    pretty_response_json = dumps(response_json, indent=4, sort_keys=False, separators=(",", ": "))

    logger.info(pretty_request_json)

    logger.info(pretty_response_json)

    if console:

        pretty_request_json_colored = highlight(

            pretty_request_json, lexers.JsonLexer(), formatters.TerminalFormatter()

        )

        pretty_response_json_colored = highlight(

            pretty_response_json, lexers.JsonLexer(), formatters.TerminalFormatter()

        )

        logger.console(pretty_request_json_colored)

        logger.console(pretty_response_json_colored)

    logger.info("\n" + description, also_console=True)

    logger.info("Request ->", also_console=True)

    logger.info(pretty_request_json, also_console=True)

    logger.info("Response ->", also_console=True)

    logger.info(pretty_response_json, also_console=True)

Delete Entity by Id Returning Response

    [Arguments]    ${id}

    ${response}=    DELETE    url=${url}/${ENTITIES_ENDPOINT_PATH}${id}    expected_status=any

    Output    ${response}

    Output    ${response}    Delete Entity by Id Returning Response

    RETURN    ${response}

Delete Entity by Id

    [Arguments]    ${id}

    ${response}=    DELETE    url=${url}/${ENTITIES_ENDPOINT_PATH}${id}    expected_status=any

    Output    ${response}

    Output    ${response}    Delete Entity by Id

    RETURN    ${response}

Query Entity

    ...    headers=${headers}

    ...    params=${params}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Query Entity

    RETURN    ${response}

Query Entities

    ...    headers=${headers}

    ...    params=${params}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Query Entities

    RETURN    ${response}

Query Entities Via POST

    ...    json=${params}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Query Entities Via POST

    RETURN    ${response}

Retrieve Entity by Id

        ...    Link=<${context}>; rel="http://www.w3.org/ns/json-ld#context";type="application/ld+json"

    END

    ${response}=    GET    url=${url}/${ENTITIES_ENDPOINT_PATH}${id}    headers=${headers}    expected_status=any

    Output    ${response}

    Output    ${response}    Retrieve Entity by Id

    RETURN    ${response}

Create Entity Selecting Content Type

    ...    json=${entity}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Create Entity Selecting Content Type

    RETURN    ${response}

Append Entity Attributes

    ...    data=${file_content}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Append Entity Attributes

    RETURN    ${response}

Append Entity Attributes With Parameters

    ...    json=${fragment_payload}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Append Entity Attributes With Parameters

    RETURN    ${response}

Update Entity Attributes

    ...    data=${file_content}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Update Entity Attributes

    RETURN    ${response}

Delete Entity Attributes

    ...    url=${url}/${ENTITIES_ENDPOINT_PATH}${entityId}/attrs/${attributeId}?${params_as_string}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Delete Entity Attributes

    RETURN    ${response}

Partial Update Entity Attributes

    ...    json=${fragment_payload}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Partial Update Entity Attributes

    RETURN    ${response}

Retrieve Entity Types

    ...    headers=${headers}

    ...    params=${params}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Retrieve Entity Types

    RETURN    ${response}

Retrieve Entity Type

    ...    url=${url}/${ENTITIES_TYPES_ENDPOINT_PATH}/${type}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Retrieve Entity Type

    RETURN    ${response}

Retrieve Attributes

    ...    headers=${headers}

    ...    params=${params}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Retrieve Attributes

    RETURN    ${response}

Retrieve Attribute

    ...    url=${url}/${ATTRIBUTES_ENDPOINT_PATH}/${attribute_name}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Retrieve Attribute

    RETURN    ${response}

Create Context Source Registration With Return

    ...    json=${payload}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Create Context Source Registration

    RETURN    ${response}

Update Context Source Registration With Return

    ...    json=${fragment}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Update Context Source Registration

    RETURN    ${response}

Delete Context Source Registration With Return

    ${response}=    DELETE

    ...    url=${url}/${CONTEXT_SOURCE_REGISTRATION_ENDPOINT_PATH}/${registration_id}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Delete Context Source Registration

    RETURN    ${response}

Create Entity

    ...    json=${entity}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Create Entity

    RETURN    ${response}

Create Or Update Temporal Representation Of Entity Selecting Content Type

    ...    json=${temporal_entity_representation}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Create Or Update Temporal Representation Of Entity Selecting Content Type

    RETURN    ${response}

Create Temporal Representation Of Entity Selecting Content Type

    ...    data=${file_content}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Create Temporal Representation Of Entity Selecting Content Type

    RETURN    ${response}

Append Attribute To Temporal Entity

    ...    data=${file_content}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Append Attribute To Temporal Entity

    RETURN    ${response}

Modify Attribute Instance From Temporal Entity

    ...    json=${fragment_payload}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Modify Attribute Instance From Temporal Entity

    RETURN    ${response}

Delete Attribute From Temporal Entity

    ...    url=${url}/${TEMPORAL_ENTITIES_ENDPOINT_PATH}/${entityId}/attrs/${attributeId}?${params_as_string}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Delete Attribute From Temporal Entity

    RETURN    ${response}

Delete Temporal Representation Of Entity With Returning Response

    ${response}=    DELETE

    ...    url=${url}/${TEMPORAL_ENTITIES_ENDPOINT_PATH}/${temporal_entity_representation_id}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Delete Temporal Representation Of Entity

    RETURN    ${response}

Get Temporal Representation Of Entity

    ...    headers=${headers}

    ...    params=${params}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Get Temporal Representation Of Entity

    RETURN    ${response}

Delete Attribute Instance From Temporal Entity

    ...    url=${url}/${TEMPORAL_ENTITIES_ENDPOINT_PATH}/${temporal_entity_id}/attrs/${attributeId}/${instanceId}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Delete Attribute Instance From Temporal Entity

    RETURN    ${response}

Update Temporal Representation Of Entity Selecting Content Type

    ...    json=${temporal_entity_fragment}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Update Temporal Representation Of Entity Selecting Content Type

    RETURN    ${response}

Batch Create Entities

    ...    json=@{entities_to_be_created}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Batch Create Entities

    RETURN    ${response}

Batch Upsert Entities

    ...    json=@{entities_to_be_upserted}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Batch Upsert Entities

    RETURN    ${response}

Batch Update Entities

    ...    json=@{entities_to_be_updated}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Batch Update Entities

    RETURN    ${response}

Batch Delete Entities

    ...    json=@{entities_ids_to_be_deleted}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Batch Delete Entities

    RETURN    ${response}

Request Entity From File

    ...    data=${file_content}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Request Entity From File

    RETURN    ${response}

Batch Request Entities From File

    ...    data=${file_content}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Batch Request Entities From File

    RETURN    ${response}

Create Temporal Representation Of Entity

    ...    json=${temporal_entity_representation}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Create Temporal Representation Of Entity

    RETURN    ${response}

Retrieve Temporal Representation Of Entity

    ...    headers=${headers}

    ...    params=${params}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Retrieve Temporal Representation Of Entity

    RETURN    ${response}

Query Temporal Representation Of Entities

    ...    headers=${headers}

    ...    params=${params}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Query Temporal Representation Of Entities

    RETURN    ${response}

Query Temporal Representation Of Entities Via Post

    ...    json=${query_payload}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Query Temporal Representation Of Entities Via Post

    RETURN    ${response}

Delete Temporal Representation Of Entity

    ${response}=    DELETE

    ...    url=${url}/${TEMPORAL_ENTITIES_ENDPOINT_PATH}/${temporal_entity_representation_id}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Delete Temporal Representation Of Entity

    RETURN    ${response}

Create Context Source Registration

    ...    json=${context_source_registration_payload}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Create Context Source Registration

    RETURN    ${response}

Update Context Source Registration

    ...    url=${url}/${CONTEXT_SOURCE_REGISTRATION_ENDPOINT_PATH}/${context_source_registration_id}

    ...    json=${update_fragment}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Update Context Source Registration

    RETURN    ${response}

Query Context Source Registrations

    ...    headers=${headers}

    ...    params=${params}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Query Context Source Registrations

    RETURN    ${response}

Delete Context Source Registration

    [Arguments]    ${context_source_registration_id}

    ${response}=    DELETE    url=${url}/${CONTEXT_SOURCE_REGISTRATION_ENDPOINT_PATH}/${context_source_registration_id}

    Output    ${response}

    Output    ${response}    Delete Context Source Registration

    RETURN    ${response}

Retrieve Context Source Registration

    ...    url=${url}/${CONTEXT_SOURCE_REGISTRATION_ENDPOINT_PATH}/${context_source_registration_id}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Retrieve Context Source Registration

    RETURN    ${response}

Create Context Source Registration Subscription

    ...    json=${subscription_payload}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Create Context Source Registration Subscription

    RETURN    ${response}

Update Context Source Registration Subscription

    ...    url=${url}/${CONTEXT_SOURCE_REGISTRATION_SUBSCRIPTION_ENDPOINT_PATH}/${subscription_id}

    ...    json=${subscription_update_fragment}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Update Context Source Registration Subscription

    RETURN    ${response}

Retrieve Context Source Registration Subscription

    ...    url=${url}/${CONTEXT_SOURCE_REGISTRATION_SUBSCRIPTION_ENDPOINT_PATH}/${subscription_id}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Retrieve Context Source Registration Subscription

    RETURN    ${response}

Query Context Source Registration Subscriptions

    ...    headers=${headers}

    ...    params=${params}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Query Context Source Registration Subscriptions

    RETURN    ${response}

Delete Context Source Registration Subscription

    [Arguments]    ${subscription_id}

    ${response}=    DELETE    ${url}/${CONTEXT_SOURCE_REGISTRATION_SUBSCRIPTION_ENDPOINT_PATH}/${subscription_id}

    Output    ${response}

    Output    ${response}    Delete Context Source Registration Subscription

    RETURN    ${response}

Update Context Source Registration Subscription From File

    ...    url=${url}/${CONTEXT_SOURCE_REGISTRATION_SUBSCRIPTION_ENDPOINT_PATH}/${subscription_id}

    ...    data=${file_content}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Update Context Source Registration Subscription From File

    RETURN    ${response}

Create Subscription

    ...    json=${subscription}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Create Subscription

    RETURN    ${response}

Create Subscription From Subscription Payload

    ...    json=${subscription_payload}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Create Subscription From Subscription Payload

    RETURN    ${response}

Create Subscription From File

    ...    data=${file_content}

    ...    headers=${headers}

    ...    expected_status=any

    Output    ${response}

    Output    ${response}    Create Subscription From File

    RETURN    ${response}

Update Subscription

    ...    json=${subscription_update_fragment}