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
def _fw(): fw = Firewall("127.0.0.1", "admin", "admin", "secret") fw._version_info = (9999, 0, 0) return fw