def _fw(*args):
    fw = Firewall("127.0.0.1", "admin", "admin", "secret")
    fw._version_info = (9999, 0, 0)

    if len(args) == 0:
        fw.xapi.get = mock.Mock(return_value=object_not_found())
    else:
        prefix = "<response><result>"
        suffix = "</result></response>"
        inner = "".join(x.element_str().decode("utf-8") for x in args)
        fw.xapi.get = mock.Mock(
            return_value=ET.fromstring(prefix + inner + suffix, ))

    return fw
Esempio n. 2
0
def _fw():
    fw = Firewall("127.0.0.1", "admin", "admin", "secret")
    fw._version_info = (9999, 0, 0)
    return fw