UnitTestCase

Type

python

Pack

ContentTesting

Source

import io

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


def main():
    args = demisto.args()
    testName = args.get("testName", "")
    listName = args.get("listName", "")
    addAfter = args.get("addAfter", "")
    try:
        if listName != "":
            listlines = execute_command("getList", {"listName": listName})
            buf = io.StringIO(listlines)
        else:
            raise DemistoException("No test case list provided")

        line = buf.readline()
        while line != "":
            words = line.split("|", 1)
            testType = words[0].strip()
            # Process commands
            if testType == "Automation" or testType == "Subplaybook" or testType == "Playbook":
                if testType == "Playbook":
                    testList = ""
                    playbooks = words[1].strip()
                else:
                    testList = words[1].strip()
                    playbooks = ""
                args = {"playbook": playbooks, "addAfter": addAfter, "testType": testType, "listName": testList}
                demisto.executeCommand("UnitTest", args)
            line = buf.readline()

    except Exception as ex:
        demisto.error(traceback.format_exc())
        return_error(f"UnitTestCase: {testName} Exception failed to execute. Error: {ex!s}")


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

README