def setUp(self):
dir_name = os.path.dirname(os.path.abspath(__file__))
f, self.jobfile = tempfile.mkstemp()
init_runtime(runtime)
runtime.configuration.load()
runtime.job = Job(jobfile = self.jobfile,
runtime = runtime,
**runtime.configuration.components.job)
mgr = runtime.tasks
task = mgr.Task(testscript = os.path.join(dir_name, 'mock_yamls/trigger_datafile.yaml'),
taskid = 'awesome')
self.testbed = load(os.path.join(dir_name, 'mock_testbeds/testbed.yaml'))
self.mock_testbed_devices()
Blitz.parameters = ParameterMap()
Blitz.uid = 'test.dev'
Blitz.parameters['testbed'] = self.testbed
self.blitz_obj = Blitz()
self.uid = self.blitz_obj.uid
self.blitz_obj.parent = self
self.blitz_obj.parent.parameters = mock.Mock()
self.dev = Device( name='PE1', os='iosxe')
self.dev.custom = {'abstraction': {'order': ['os']}}
self.blitz_obj.parameters['test_sections'] = [{'section1': [{'execute': {'command': 'cmd', 'device': 'PE1'}}]}]
sections = self.blitz_obj._discover()
self.kwargs = {'self': self.blitz_obj,
'testbed': self.testbed,
'section': sections[0].__testcls__(sections[0]),
'name': ''}
def setUp(self):
dir_name = os.path.dirname(os.path.abspath(__file__))
mfc = MockFuncClass()
self.testbed = load(
os.path.join(dir_name, 'mock_testbeds/testbed.yaml'))
Blitz.parameters = ParameterMap()
Blitz.uid = 'test.dev'
Blitz.parameters['testbed'] = self.testbed
Blitz.parameters['save_variable_name'] = {
'dev_name': 'PE2',
'command': 'sh version',
'sub_command': 'interface',
'type_k': 1500,
'list_item': [177, 24, 13, 45],
'iter_class': mfc,
'dict1': {
'st': "name"
}
}
self.blitz_obj = Blitz()
self.uid = self.blitz_obj.uid
self.blitz_obj.parent = self
self.blitz_obj.parent.parameters = mock.Mock()
self.blitz_obj.parameters['test_sections'] = [{
'section1': [{
'execute': {
'command': 'cmd',
'device': 'PE1'
}
}]
}]
sections = self.blitz_obj._discover()
self.section = sections[0].__testcls__(sections[0])
def setUp(self):
dir_name = os.path.dirname(os.path.abspath(__file__))
self.testbed = load(os.path.join(dir_name, 'mock_testbeds/testbed.yaml'))
Blitz.parameters = ParameterMap()
Blitz.uid = 'test.dev'
Blitz.parameters['testbed'] = self.testbed
self.blitz_obj = Blitz()
self.dev = Device( name='PE1', os='iosxe')
self.dev.custom = {'abstraction': {'order': ['os']}}
self.blitz_obj.parameters['test_sections'] = [{'section1': [{'action': {'command': 'a'}}]}]
sections = self.blitz_obj._discover()
self.kwargs = {'self': self.blitz_obj,
'section': sections[0],
'name': ''}
def _initiate_blitz_cls(self, yaml_file):
dir_name = os.path.dirname(os.path.abspath(__file__))
self.blitz_cls = Blitz
self.testbed = load(os.path.join(dir_name, 'mock_testbeds/testbed.yaml'))
self.blitz_cls.parameters = ParameterMap()
self.blitz_cls.parameters['testbed'] = self.testbed
self._mock_testbed_devs()
self.datafile = yaml.safe_load(yaml_file)
for key, value in self.datafile.items():
self.blitz_cls.uid = "{}.{}".format(key, value['devices'][0])
self.blitz_cls.parameters['test_sections'] = value['test_sections']
if value.get('description'):
self.blitz_cls.description = value['description']
def setUp(self):
dir_name = os.path.dirname(os.path.abspath(__file__))
f, self.jobfile = tempfile.mkstemp()
init_runtime(runtime)
runtime.configuration.load()
runtime.job = Job(jobfile=self.jobfile,
runtime=runtime,
**runtime.configuration.components.job)
mgr = runtime.tasks
task = mgr.Task(testscript=os.path.join(
dir_name, 'mock_yamls/trigger_datafile.yaml'),
taskid='awesome')
self.testbed = load(
os.path.join(dir_name, 'mock_testbeds/testbed.yaml'))
self.mock_testbed_devs()
Blitz.parameters = ParameterMap()
Blitz.uid = 'test.dev'
Blitz.parameters['testbed'] = self.testbed
self.blitz_obj = Blitz()
self.blitz_obj.parameters['test_sections'] = [{
'section1': [{
'parallel': [{
'action': {
'arg1': 'cmd'
}
}]
}]
}]
sections = self.blitz_obj._discover()
self.kwargs = {
'self': self.blitz_obj,
'testbed': self.testbed,
'section': sections[0].__testcls__(sections[0]),
'name': ''
}
def setUp(self):
dir_name = os.path.dirname(os.path.abspath(__file__))
mfc = MockFuncClass()
self.testbed = load(
os.path.join(dir_name, 'mock_testbeds/testbed.yaml'))
Blitz.parameters = ParameterMap()
Blitz.uid = 'test.dev'
Blitz.parameters['testbed'] = self.testbed
Blitz.parameters['save_variable_name'] = {
'dev_name': 'PE2',
'command': 'sh version',
'sub_command': 'interface',
'type_k': 1500,
'list_item': [177, 24, 13, 45],
'iter_class': mfc,
'dict1': {
'st': "name"
}
}
self.blitz_obj = Blitz()
