fixes for pylama

This commit is contained in:
Pieprzycki Piotr 2016-12-21 13:51:52 +01:00
parent 875ad85bd7
commit f1de371d37
2 changed files with 5 additions and 8 deletions

View File

@ -92,10 +92,10 @@ class VyOSDriver(NetworkDriver):
def open(self):
self.device = ConnectHandler(device_type='vyos',
host=self._hostname,
username=self._username,
password=self._password,
**self.netmiko_optional_args)
host=self._hostname,
username=self._username,
password=self._password,
**self.netmiko_optional_args)
try:
self._scp_client = SCPConn(self.device)
@ -143,7 +143,7 @@ class VyOSDriver(NetworkDriver):
if os.path.exists(filename) is True:
with open(filename) as f:
self.device.send_command("cp "+self._BOOT_FILENAME+" "
+ self._BACKUP_FILENAME)
+ self._BACKUP_FILENAME)
self._new_config = f.read()
cfg = [x for x in self._new_config.split("\n") if x is not ""]
output_loadcmd = self.device.send_config_set(cfg)

View File

@ -37,7 +37,6 @@ class PatchedVyOSDriver(vyos.VyOSDriver):
self.patched_attrs = ['device']
self.device = FakeVyOSDevice()
#self._device = FakeVyOSDevice()
def close(self):
pass
@ -54,8 +53,6 @@ class PatchedVyOSDriver(vyos.VyOSDriver):
class FakeVyOSDevice(BaseTestDouble):
"""VyOS device test double."""
def send_command(self, command, **kwargs):
filename = '{}.text'.format(self.sanitize_text(command))
print filename