def test_get_variable_replace_replace_iter_fucnt(self):
kwargs = {'a': r"%VARIABLES{iter_class.mock_func}",
'self': self.blitz_obj,
'section': self.section}
replaced_kwargs = get_variable(**kwargs)
self.assertEqual(replaced_kwargs['a'], 'mock func returned val')
def test_get_variable_replace_replace_list_item(self):
kwargs = {'cmd': r"%VARIABLES{list_item[2]}",
'self': self.blitz_obj,
'section': self.section}
replaced_kwargs = get_variable(**kwargs)
self.assertEqual(replaced_kwargs['cmd'], 13)
def test_get_variable_replace_replace_mid_str(self):
kwargs = {'cmd': r"show %VARIABLES{sub_command}",
'self': self.blitz_obj,
'section': self.section}
replaced_kwargs = get_variable(**kwargs)
self.assertEqual(replaced_kwargs['cmd'], 'show interface')
def test_get_variable_replace(self):
kwargs = {'command': r"%VARIABLES{command}",
'device': 'PE2',
'self': self.blitz_obj,
'section': self.section}
replaced_kwargs = get_variable(**kwargs)
# check if replacement is done
self.assertEqual(replaced_kwargs['command'], self.blitz_obj.parameters['save_variable_name']['command'])
def test_get_variable_replace_replace_keep_type(self):
kwargs = {'self': self.blitz_obj,
'item': {'value': r"%VARIABLES{type_k}",
'val': 11,
'ls': ['a', {'b':1}]},
'section': self.section}
replaced_kwargs = get_variable(**kwargs)
# check if type would stay the same
self.assertEqual(type(replaced_kwargs['item']['value']), int)
def test_custom_start_step_messsage_with_variable(self):
#saved variable
Blitz.parameters['save_variable_name'] = {'command': 'sh version'}
self.blitz_obj = Blitz()
self.blitz_obj.parameters['test_sections'] = [{'section1': [{'action': {'command': 'a'}}]}]
sections = self.blitz_obj._discover()
self.section = sections[0].__testcls__(sections[0])
self.kwargs = {
'self': self.blitz_obj,
'section': self.section,
'custom_start_step_message': 'test command: %VARIABLES{command}'
}
#To get the saved variable
replaced_kwargs = get_variable(**self.kwargs)
self.assertEqual(replaced_kwargs['custom_start_step_message'], 'test command: sh version')