Ejemplo n.º 1
0
class ArrayParsingTests(TestCase):
    def setUp(self):
        self.sheet = ArrayParsing(get_sheet_test('arrayParsing_test.xlsx'))

    def test_get_param_by_index(self):
        expected = '10.2.2.254'
        got = self.sheet.get_value_of_var_by_index_and_param('105', 'Gateway')
        self.assertEqual(expected, got)

    def test_get_param_by_index2(self):
        expected = '105'
        got = self.sheet.get_value_of_var_by_index_and_param('105', 'Vlans ID')
        self.assertEqual(expected, got)

    def test_get_param_by_index_failed(self):
        try:
            expected = KeyError
            got = self.sheet.get_value_of_var_by_index_and_param(
                '101', 'invalid_param')
            self.assertRaises(expected, got)
        except KeyError:
            pass

    def test_set_param_by_index(self):
        index_value, param_value, updated_value = '104', 'Gateway', '1.1.1.1'
        self.sheet.set_value_by_index_and_param(index_value, param_value,
                                                updated_value)
        got = self.sheet.get_value_of_var_by_index_and_param(
            index_value, param_value)
        self.assertEqual(updated_value, got)

    def test_set_param_by_index_failed(self):
        try:
            index_value, param_value, updated_value = '000', 'Gateway', '1.1.1.1'
            expected = KeyError
            got = self.sheet.set_value_by_index_and_param(
                index_value, param_value, updated_value)
            self.assertRaises(expected, got)
        except KeyError:
            pass

    def test_get_all_param_by_index(self):
        expected = {
            'Vlans ID': '105',
            'Name': 'VLAN-VOICE-02',
            'Subnet': '10.2.2.0',
            'Mask': '255.255.255.0',
            'Gateway': '10.2.2.254',
            'Description': '',
            'Subnet_arp': '10.2.2.224',
            'Wildcard_arp': '0.0.0.15',
            'Template Default': 'Default',
            'Template Snooping': 'Snooping',
            'Template Inspection': 'Inspection'
        }
        got = self.sheet.get_all_param_by_index('105')
        self.assertDictEqual(expected, got)

    def test_get_nbr_of_rows(self):
        expected = 23
        got = self.sheet.get_nbr_of_rows()
        self.assertEqual(expected, got)

    def test_get_nbr_of_cols(self):
        expected = 11
        got = self.sheet.get_nbr_of_cols()
        self.assertEqual(expected, got)

    def test_get_nbr_of_cols_in_row(self):
        expected = 11
        got = self.sheet.get_nbr_of_cols_in_row(0)
        self.assertEquals(expected, got)

    def test_get_all_headers(self):
        expected = [
            'Vlans ID', 'Name', 'Subnet', 'Mask', 'Gateway', 'Description',
            'Subnet_arp', 'Wildcard_arp', 'Template Default',
            'Template Snooping', 'Template Inspection'
        ]
        got = self.sheet.get_all_headers()
        self.assertListEqual(expected, got)

    def test_get_all_indexes(self):
        expected = [
            '21', '22', '23', '41', '42', '43', '101', 'test', '103', '104',
            '105', '106', '107', '108', '109'
        ]
        got = self.sheet.get_all_indexes()
        self.assertListEqual(expected, got)

    def test_is_duplication_False(self):
        expected = (False, [])
        got = self.sheet.is_duplication(self.sheet.get_all_headers())
        self.assertEqual(expected, got)

    def test_is_duplication_True(self):
        list_of_superheroes = [
            'Superman', 'Superman', 'Superman', 'Batman', 'Batman', 'Ironman',
            'Thor'
        ]
        expected = (True, ['Superman', 'Batman'])
        got = self.sheet.is_duplication(list_of_superheroes)
        self.assertEqual(expected, got)

    def test_get_row_where_value(self):
        expected = 11
        got = self.sheet.get_row_where_value(105)
        self.assertEqual(expected, got)

    def test_get_row_where_value_failed(self):
        expected = -1
        got = self.sheet.get_row_where_value(0000)
        self.assertEqual(expected, got)

    def test_delimitation_between_indexes_and_commands(self):
        expected = 17
        got = self.sheet.delimitation_between_indexes_and_commands()
        self.assertEqual(expected, got)

    def test_template_content_by_name1(self):
        expected = '{{Name}} and {{Subnet}} and\n{{Gateway}}'
        got = self.sheet.template_content_by_name('default')
        self.assertEqual(expected, got)

    def test_template_content_by_name2(self):
        expected = ''
        got = self.sheet.template_content_by_name('inspection')
        self.assertEqual(expected, got)
Ejemplo n.º 2
0
class ArrayParsingTests(TestCase):

    def setUp(self):
        self.sheet = ArrayParsing(get_sheet_test('arrayParsing_test.xlsx'))

    def test_get_param_by_index(self):
        expected = '10.2.2.254'
        got = self.sheet.get_value_of_var_by_index_and_param('105', 'Gateway')
        self.assertEqual(expected, got)

    def test_get_param_by_index2(self):
        expected = '105'
        got = self.sheet.get_value_of_var_by_index_and_param('105', 'Vlans ID')
        self.assertEqual(expected, got)

    def test_get_param_by_index_failed(self):
        try:
            expected = KeyError
            got = self.sheet.get_value_of_var_by_index_and_param('101', 'invalid_param')
            self.assertRaises(expected, got)
        except KeyError:
            pass

    def test_set_param_by_index(self):
        index_value, param_value, updated_value = '104', 'Gateway', '1.1.1.1'
        self.sheet.set_value_by_index_and_param(index_value, param_value, updated_value)
        got = self.sheet.get_value_of_var_by_index_and_param(index_value, param_value)
        self.assertEqual(updated_value, got)

    def test_set_param_by_index_failed(self):
        try:
            index_value, param_value, updated_value = '000', 'Gateway', '1.1.1.1'
            expected = KeyError
            got = self.sheet.set_value_by_index_and_param(index_value, param_value, updated_value)
            self.assertRaises(expected, got)
        except KeyError:
            pass

    def test_get_all_param_by_index(self):
        expected = {'Vlans ID': '105',
                    'Name': 'VLAN-VOICE-02',
                    'Subnet': '10.2.2.0',
                    'Mask': '255.255.255.0',
                    'Gateway': '10.2.2.254',
                    'Description': '',
                    'Subnet_arp': '10.2.2.224',
                    'Wildcard_arp': '0.0.0.15',
                    'Template Default': 'Default',
                    'Template Snooping': 'Snooping',
                    'Template Inspection': 'Inspection'}
        got = self.sheet.get_all_param_by_index('105')
        self.assertDictEqual(expected, got)

    def test_get_nbr_of_rows(self):
        expected = 23
        got = self.sheet.get_nbr_of_rows()
        self.assertEqual(expected, got)

    def test_get_nbr_of_cols(self):
        expected = 11
        got = self.sheet.get_nbr_of_cols()
        self.assertEqual(expected, got)

    def test_get_nbr_of_cols_in_row(self):
        expected = 11
        got = self.sheet.get_nbr_of_cols_in_row(0)
        self.assertEquals(expected, got)

    def test_get_all_headers(self):
        expected = ['Vlans ID', 'Name', 'Subnet', 'Mask', 'Gateway', 'Description', 'Subnet_arp', 'Wildcard_arp',
                    'Template Default', 'Template Snooping', 'Template Inspection']
        got = self.sheet.get_all_headers()
        self.assertListEqual(expected, got)

    def test_get_all_indexes(self):
        expected = ['21', '22', '23', '41', '42', '43', '101', 'test', '103', '104', '105', '106', '107', '108', '109']
        got = self.sheet.get_all_indexes()
        self.assertListEqual(expected, got)

    def test_is_duplication_False(self):
        expected = (False, [])
        got = self.sheet.is_duplication(self.sheet.get_all_headers())
        self.assertEqual(expected, got)

    def test_is_duplication_True(self):
        list_of_superheroes = ['Superman', 'Superman', 'Superman', 'Batman', 'Batman', 'Ironman', 'Thor']
        expected = (True, ['Superman', 'Batman'])
        got = self.sheet.is_duplication(list_of_superheroes)
        self.assertEqual(expected, got)

    def test_get_row_where_value(self):
        expected = 11
        got = self.sheet.get_row_where_value(105)
        self.assertEqual(expected, got)

    def test_get_row_where_value_failed(self):
        expected = -1
        got = self.sheet.get_row_where_value(0000)
        self.assertEqual(expected, got)

    def test_delimitation_between_indexes_and_commands(self):
        expected = 17
        got = self.sheet.delimitation_between_indexes_and_commands()
        self.assertEqual(expected, got)

    def test_template_content_by_name1(self):
        expected = '{{Name}} and {{Subnet}} and\n{{Gateway}}'
        got = self.sheet.template_content_by_name('default')
        self.assertEqual(expected, got)

    def test_template_content_by_name2(self):
        expected = ''
        got = self.sheet.template_content_by_name('inspection')
        self.assertEqual(expected, got)