class InterfaceTests(TestCase): def setUp(self): self.interface = Interface(get_sheet_test('interface_test.xlsx')) def test_get_by_equipment_and_function(self): ref_list = ['Gigabit', '1', '0', '1-2'] self.assertListEqual(ref_list, self.interface.get_by_equipment_and_function('CISCO WS-C3750v2-48PS-S', 'Uplink')) def test_get_by_equipment_and_function_in_stack(self): value_expected = 'Gigabit 1/0/1-2,Gigabit 2/0/1-2,Gigabit 3/0/1-2' self.assertEqual(value_expected, self.interface.get_by_equipment_and_function_in_stack('CISCO WS-C3750v2-48PS-S', 'Uplink', 3)) def test_in_string(self): value_expected = 'Gigabit 1/0/1-2' data = ['Gigabit', '1', '0', '1-2'] self.assertEqual(value_expected, self.interface.in_string(data))
class InterfaceTests(TestCase): def setUp(self): self.interface = Interface(get_sheet_test('interface_test.xlsx')) def test_get_by_equipment_and_function(self): ref_list = ['Gigabit', '1', '0', '1-2'] self.assertListEqual( ref_list, self.interface.get_by_equipment_and_function( 'CISCO WS-C3750v2-48PS-S', 'Uplink')) def test_get_by_equipment_and_function_in_stack(self): value_expected = 'Gigabit 1/0/1-2,Gigabit 2/0/1-2,Gigabit 3/0/1-2' self.assertEqual( value_expected, self.interface.get_by_equipment_and_function_in_stack( 'CISCO WS-C3750v2-48PS-S', 'Uplink', 3)) def test_in_string(self): value_expected = 'Gigabit 1/0/1-2' data = ['Gigabit', '1', '0', '1-2'] self.assertEqual(value_expected, self.interface.in_string(data))
def main(excel_file, template_file, directory): wb = get_excel_workbook(excel_file) sheet_names = wb.sheet_names() workbook = dict() template = open_file(template_file) for sheet in sheet_names: xl_sheet = wb.sheet_by_name(sheet) if (xl_sheet.cell(0, 0).value == 'Function' and xl_sheet.cell(0, 1).value == 'Variable' and xl_sheet.cell(0, 2).value == 'Value'): workbook[sheet] = ListParsing(xl_sheet) elif xl_sheet.cell(0, 0).value == 'Text': workbook[sheet] = TextParsing(xl_sheet) elif sheet == 'Interfaces': workbook[sheet] = Interface(xl_sheet) else: workbook[sheet] = ArrayParsing(xl_sheet) if not directory: directory = os.path.dirname(template_file) is_directory = False version_number = 1 while is_directory is False: directory_name = ''.join(['scripts-', os.path.splitext(os.path.basename(excel_file))[0], '-v', str(version_number)]) if os.path.isdir(os.path.join(directory, directory_name)) is False: directory = os.path.join(directory, directory_name) os.mkdir(directory) is_directory = True version_number += 1 list_of_equipments = list() for hostname in workbook['Global'].get_all_indexes(): equipment = Equipment(hostname, template, workbook) list_of_equipments.append(equipment) equipment.save_script_as(directory, hostname)
def setUp(self): self.interface = Interface(get_sheet_test('interface_test.xlsx'))