class FakeFortiOSDevice:
@staticmethod
def read_txt_file(filename):
with open(filename) as data_file:
return data_file.read().splitlines()
def execute_command(self, command):
return self.read_txt_file('fortios/mock_data/{}.txt'.format(command.replace(' ', '_')))
def load_config(self, config_block):
self.running_config = FortiConfig('running')
self.running_config.parse_config_output(self.read_txt_file('fortios/mock_data/{}.txt'.format(
config_block.replace(' ', '_'))))
class FakeFortiOSDevice:
@staticmethod
def read_txt_file(filename):
with open(filename) as data_file:
return data_file.read().splitlines()
def execute_command(self, command):
return self.read_txt_file('fortios/mock_data/{}.txt'.format(
command.replace(' ', '_').replace('|', '').replace(':', '')))
def load_config(self, config_block):
self.running_config = FortiConfig('running')
self.running_config.parse_config_output(
self.read_txt_file('fortios/mock_data/{}.txt'.format(
config_block.replace(' ', '_'))))
class FakeDevice(BaseTestDouble):
"""Device test double."""
def open(self):
pass
def close(self):
pass
def execute_command(self, command):
filename = '{}.txt'.format(self.sanitize_text(command))
full_path = self.find_file(filename)
return self.read_txt_file(full_path).splitlines()
def load_config(self, config_block):
filename = '{}.txt'.format(self.sanitize_text(config_block))
full_path = self.find_file(filename)
self.running_config = FortiConfig('running')
self.running_config.parse_config_output(self.read_txt_file(full_path))
class FakeDevice(BaseTestDouble):
"""Device test double."""
def open(self):
pass
def close(self):
pass
def execute_command(self, command):
filename = '{}.txt'.format(self.sanitize_text(command))
full_path = self.find_file(filename)
return self.read_txt_file(full_path).splitlines()
def load_config(self, config_block):
filename = '{}.txt'.format(self.sanitize_text(config_block))
full_path = self.find_file(filename)
self.running_config = FortiConfig('running')
self.running_config.parse_config_output(self.read_txt_file(full_path))
