def _mock_manager(self, *args, **kwargs): if kwargs: # if 'path' in kwargs and 'detail' in kwargs: # return self._read_file('dir_list_detail.xml') if 'path' in kwargs: if kwargs['path'] == 'test/stat/decode_dir': return self._read_file('file-list_dir.xml') elif kwargs['path'] == 'test/stat/decode_file': return self._read_file('file-list_file.xml') elif kwargs['path'] == 'test/checksum': return self._read_file('checksum.xml') elif kwargs['path'] == 'test/stat/decode_symbolic_link': return self._read_file('file-list_symlink.xml') if 'directory' in kwargs: if kwargs['directory'] == 'change/directory': return self._read_file('set-cli-working-directory.xml') if 'filename' in kwargs: if kwargs['filename'] == 'test/cat.txt': return self._read_file('file-show.xml') device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) elif args: if args[0].tag == 'command': if args[0].text == 'show cli directory': return self._read_file('show-cli-directory.xml') elif args[0].tag == 'get-system-storage': return self._read_file('get-system-storage.xml') elif args[0].tag == 'request-system-storage-cleanup': return self._read_file('request-system-storage-cleanup.xml') elif args[0].tag == 'file-archive': return self._read_file('file-archive.xml')
def _mock_manager(self, *args, **kwargs): if kwargs: device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) elif args: if args[0].tag == 'command': if args[0].text == 'show cli directory': return self._read_file('show-cli-directory.xml') if args[0].text == 'show interface terse': return self._read_file('show-interface-terse.json') elif args[0].text == 'show configuration': return self._read_file('show-configuration.xml') elif args[0].text == 'show system alarms': return self._read_file('show-system-alarms.xml') elif args[0].text == 'show system uptime | display xml rpc': return self._read_file('show-system-uptime-rpc.xml') elif args[0].text == 'show configuration interfaces': return self._read_file('show-configuration-interfaces.xml') elif args[0].text == 'show interfaces terse asdf': return self._read_file('show-interfaces-terse-asdf.xml') else: raise RpcError else: if args[0].attrib.get('format')=='json': return self._read_file(args[0].tag + '.json') return self._read_file(args[0].tag + '.xml')
def mock_manager(*args, **kwargs): if 'device_params' in kwargs: # open connection device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) elif args: # rpc request rpc_request = args[0].tag rpc_command = str(args[0].text) rpc_command = rpc_command.strip() rpc_command = rpc_command.replace(" ", "_") if rpc_request in rpc_reply_dict: xml = rpc_reply_dict[rpc_request] elif 'dir' in rpc_reply_dict: fname = os.path.join(rpc_reply_dict['dir'], 'rpc-reply', rpc_command, rpc_request + '.xml') with open(fname, 'r') as f: xml = f.read() else: _rpc_reply_dict['dir'] fname = os.path.join(os.path.dirname(__file__), 'rpc-reply', rpc_command, rpc_request + '.xml') with open(fname, 'r') as f: xml = f.read() rpc_reply = NCElement(xml, dev._conn._device_handler.transform_reply()) return rpc_reply
def _mock_manager(self, *args, **kwargs): if kwargs: device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) if args: return self._read_file(args[0].tag + '.xml')
def _mock_manager(self, *args, **kwargs): if kwargs: device_params = kwargs["device_params"] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) if args: return self._read_file("chassis_" + args[0].tag + ".xml")
def _mock_manager(self, *args, **kwargs): if kwargs: device_params = kwargs["device_params"] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) if args: if "version_RE0-RE0" in self.facts: return self._read_file(args[0].tag + "_RE0-RE0.xml") return self._read_file(args[0].tag + ".xml")
def _mock_manager(self, *args, **kwargs): if kwargs: if args and ("normalize" in kwargs or "filter_xml" in kwargs): return self._read_file(args[0].tag + ".xml") device_params = kwargs["device_params"] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) if args: return self._read_file(args[0].tag + ".xml")
def _mock_manager(self, *args, **kwargs): if kwargs and "ignore_warning" not in kwargs: # Little hack for mocked execute if "dev_timeout" in kwargs: if ( args and args[0].findtext("package-name") == "/var/tmp/test_no_result.tgz" ): return self._read_file(args[0].tag + ".no_result.xml") elif ( args and args[0].findtext("package-name") == "/var/tmp/test_no_mx80_packages.tgz" ): return self._read_file(args[0].tag + ".no_mx80_packages.xml") else: return self._read_file(args[0].tag + ".xml") if "path" in kwargs: if kwargs["path"] == "/packages": return self._read_file("file-list_dir.xml") if args and self._testMethodName == "test_sw_zeroize": return self._read_file("request-zeroize.xml") device_params = kwargs["device_params"] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) elif args: if self._testMethodName == "test_sw_reboot_output_in_reply": return self._read_file("request-reboot-output.xml") elif args[0].find("at") is not None: return self._read_file("request-reboot-at.xml") elif self._testMethodName == "test_sw_check_pending_install": if args[0].text == "request-package-check-pending-install": return self._read_file( "request-package-check-pending-install-error.xml" ) elif ( self._testMethodName == "test_sw_check_pending_install_RpcError_continue" ): if args[0].text == "request-package-check-pending-install": xml = """<rpc-error> <error-type>protocol</error-type> <error-tag>operation-failed</error-tag> <error-severity>error</error-severity> <error-message>syntax error</error-message> <error-info> <bad-element>request-package-check-pendings-install</bad-element> </error-info> </rpc-error>""" return RpcError(rsp=etree.fromstring(xml)) else: return self._read_file(args[0].tag + ".xml")
def _mock_manager(self, *args, **kwargs): if kwargs: if 'path' in kwargs: if kwargs['path'] == '/packages': return self._read_file('file-list_dir.xml') device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) elif args: return self._read_file(args[0].tag + '.xml')
def _mock_manager(self, *args, **kwargs): if kwargs: device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) if args: if args[0].xpath('//configuration/system/login/user'): return self._read_file('get-configuration-user.xml') else: return self._read_file(args[0].tag + '.xml')
def _mock_manager(self, *args, **kwargs): if kwargs: if 'normalize' in kwargs and args: return self._read_file(args[0].tag + '.xml') device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) if args: if len(args[0]) > 0 and args[0][0].tag == 'bgp': return self._read_file(args[0].tag + '_bgp_openconfig.xml') return self._read_file(args[0].tag + '.xml')
def _mock_manager(self, *args, **kwargs): if kwargs: if "normalize" in kwargs and args: return self._read_file(args[0].tag + ".xml") device_params = kwargs["device_params"] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) if args: if len(args[0]) > 0 and args[0][0].tag == "bgp": return self._read_file(args[0].tag + "_bgp_openconfig.xml") elif (args[0].attrib.get("format") == "json" and args[0].tag == "get-configuration"): return self._read_file(args[0].tag + ".json") return self._read_file(args[0].tag + ".xml")
def _mock_manager(self, *args, **kwargs): if kwargs: device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) if args: if self.vc is True and args[0].tag == 'get-virtual-chassis-information': if self.vct is True: return True elif self.vcf is True: return self._read_file('get-virtual-chassis-information_mmvcf.xml') else: return self._read_file('get-virtual-chassis-information.xml') return self._read_file(args[0].tag + '_' + self.mode + '.xml')
def _mock_manager(self, *args, **kwargs): if kwargs: # Little hack for mocked execute if 'dev_timeout' in kwargs: return self._read_file(args[0].tag + '.xml') if 'path' in kwargs: if kwargs['path'] == '/packages': return self._read_file('file-list_dir.xml') device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) elif args: if args[0].find('at') is not None: return self._read_file('request-reboot-at.xml') else: return self._read_file(args[0].tag + '.xml')
def _mock_manager(self, *args, **kwargs): if kwargs and 'ignore_warning' not in kwargs: # Little hack for mocked execute if 'dev_timeout' in kwargs: if (args and args[0].findtext('package-name') == '/var/tmp/test_no_result.tgz'): return self._read_file(args[0].tag + '.no_result.xml') else: return self._read_file(args[0].tag + '.xml') if 'path' in kwargs: if kwargs['path'] == '/packages': return self._read_file('file-list_dir.xml') device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) elif args: if args[0].find('at') is not None: return self._read_file('request-reboot-at.xml') else: return self._read_file(args[0].tag + '.xml')
def _mock_manager(self, *args, **kwargs): if kwargs: device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) elif args: if args[0].tag == 'command': if args[0].text == 'show cli directory': return self._read_file('show-cli-directory.xml') elif args[0].text == 'show configuration': return self._read_file('show-configuration.xml') elif args[0].text == 'show system alarms': return self._read_file('show-system-alarms.xml') elif args[0].text == 'show system uptime | display xml rpc': return self._read_file('show-system-uptime-rpc.xml') else: raise RpcError else: return self._read_file(args[0].tag + '.xml')
def _mock_manager(self, *args, **kwargs): if kwargs and "normalize" not in kwargs: device_params = kwargs["device_params"] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) elif args: if args[0].tag == "command": if args[0].text == "show cli directory": return self._read_file("show-cli-directory.xml") if args[0].text == "show interface terse": return self._read_file("show-interface-terse.json") elif args[0].text == "show configuration": return self._read_file("show-configuration.xml") elif args[0].text == "show system alarms": return self._read_file("show-system-alarms.xml") elif args[0].text == "show system uptime| display xml rpc": return self._read_file("show-system-uptime-rpc.xml") elif args[0].text == "show configuration interfaces": return self._read_file("show-configuration-interfaces.xml") elif args[0].text == "show interfaces terse asdf": return self._read_file("show-interfaces-terse-asdf.xml") elif (args[0].text == "show interfaces ge-0/0/0.0 " "routing-instance all media"): return self._read_file( "show-interfaces-routing-instance-media.xml") elif (args[0].text == "show interfaces ge-0/0/0.0 " "routing-instance all media| display " "xml rpc"): return self._read_file( "show-interfaces-routing-instance-media-rpc.xml") else: raise RpcError else: if args[0].attrib.get("format") == "json": return self._read_file(args[0].tag + ".json") return self._read_file(args[0].tag + ".xml")
def _mock_manager(self, *args, **kwargs): if kwargs and 'ignore_warning' not in kwargs: # Little hack for mocked execute if 'dev_timeout' in kwargs: if (args and args[0].findtext('package-name') == '/var/tmp/test_no_result.tgz'): return self._read_file(args[0].tag + '.no_result.xml') else: return self._read_file(args[0].tag + '.xml') if 'path' in kwargs: if kwargs['path'] == '/packages': return self._read_file('file-list_dir.xml') device_params = kwargs['device_params'] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) elif args: if args[0].find('at') is not None: return self._read_file('request-reboot-at.xml') elif self._testMethodName == 'test_sw_check_pending_install': if args[0].text == 'request-package-check-pending-install': return self._read_file( 'request-package-check-pending-install-error.xml') elif self._testMethodName == 'test_sw_check_pending_install_RpcError_continue': if args[0].text == 'request-package-check-pending-install': xml = '''<rpc-error> <error-type>protocol</error-type> <error-tag>operation-failed</error-tag> <error-severity>error</error-severity> <error-message>syntax error</error-message> <error-info> <bad-element>request-package-check-pendings-install</bad-element> </error-info> </rpc-error>''' return RpcError(rsp=etree.fromstring(xml)) else: return self._read_file(args[0].tag + '.xml')
def _mock_manager(self, *args, **kwargs): if kwargs: # if 'path' in kwargs and 'detail' in kwargs: # return self._read_file('dir_list_detail.xml') if "path" in kwargs: if kwargs["path"] == "test/stat/decode_dir": return self._read_file("file-list_dir.xml") elif kwargs["path"] == "test/stat/decode_file": return self._read_file("file-list_file.xml") elif kwargs["path"] == "test/checksum": return self._read_file("checksum.xml") elif kwargs["path"] == "test/stat/decode_symbolic_link": return self._read_file("file-list_symlink.xml") if "directory" in kwargs: if kwargs["directory"] == "change/directory": return self._read_file("set-cli-working-directory.xml") if "filename" in kwargs: if kwargs["filename"] == "test/cat.txt": return self._read_file("file-show.xml") device_params = kwargs["device_params"] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler) elif args: if args[0].tag == "command": if args[0].text == "show cli directory": return self._read_file("show-cli-directory.xml") elif args[0].tag == "get-system-storage": return self._read_file("get-system-storage.xml") elif args[0].tag == "get-directory-usage-information": return self._read_file("get-directory-usage-information.xml") elif args[0].tag == "request-system-storage-cleanup": return self._read_file("request-system-storage-cleanup.xml") elif args[0].tag == "file-archive": return self._read_file("file-archive.xml")
def _mock_manager_setup(self, *args, **kwargs): if kwargs: device_params = kwargs["device_params"] device_handler = make_device_handler(device_params) session = SSHSession(device_handler) return Manager(session, device_handler)