import demistomock as demisto  # noqa: F401
from CommonServerPython import *  # noqa: F401


def main():
    try:
        listName = demisto.args()["list"]
        results = execute_command("getList", {"listName": listName})
        if "Item not found" not in results:
            fields = json.loads(results)
            # Set context
            for key, val in fields.items():
                execute_command("Set", {"key": key, "value": val})
        else:
            raise DemistoException(f"UnitTestLoadContextList: list '{listName}' not found")
    except Exception as ex:
        raise ex


if __name__ in ("__main__", "__builtin__", "builtins"):
    main()