devicemanager/tests/test_device_interfaces_class.py - master - creator/ecstasy

devicemanager/tests/test_device_interfaces_class.py
145 строк8 KB
irudenko
v 15.4
28 июн 2023, 14:52
28 июн 2023, 14:52bb45693
from django.test import SimpleTestCase
from devicemanager.device import Interfaces
from devicemanager.device.interfaces import Interface


class TestInterfacesClass(SimpleTestCase):
    def test_interface_init(self):
        # Создание списка интерфейсов.
        interfaces_list = [
            ["eth1", "up", "description1"],
            ["eth2", "down", "description2"],
            ["eth3", "admin down", "description3"],
            ["eth4", "up", "description4"],
        ]
        # Создание нового экземпляра класса Interfaces.
        intf = Interfaces(interfaces_list)

        # Проверка того, что второй элемент существует и он верный.
        self.assertEqual(str(intf[1]), str(Interface(*interfaces_list[1])))
        # Проверяем, что количество интерфейсов равно 4.
        self.assertEqual(intf.count, 4)
        # Проверяем, что количество интерфейсов со статусом up равно 2.
        self.assertEqual(intf.up(only_count=True), 2)
        # Проверяем, что количество интерфейсов со статусом down равно 1.
        self.assertEqual(intf.down(only_count=True), 2)
        # Проверяем, что количество интерфейсов со статусом admin_down равно 1.
        self.assertEqual(intf.admin_down(only_count=True), 1)

        # Проверка того, что метод with_vlans возвращает правильный интерфейс.
        self.assertEqual(str(intf.with_vlans([11])), str(Interfaces()))

        # Проверка того, что метод filter_by_desc возвращает правильный ответ.
        self.assertEqual(str(intf.filter_by_desc("description")), str(intf))
        # Проверка того, что метод filter_by_desc возвращает правильный ответ.
        self.assertEqual(
            str(intf.filter_by_desc("description2")),
            str(Interfaces([interfaces_list[1]])),
        )
        # Проверка того, что метод filter_by_desc возвращает правильный ответ.
        # Для не существующего результата по фильтру описания
        self.assertEqual(str(intf.filter_by_desc("invalid")), str(Interfaces()))

        interfaces_as_dict_list = [
            {"Interface": "eth1", "Status": "up", "Description": "description1"},
            {"Interface": "eth2", "Status": "down", "Description": "description2"},
            {
                "Interface": "eth3",
                "Status": "admin down",
                "Description": "description3",
            },
            {"Interface": "eth4", "Status": "up", "Description": "description4"},
        ]
        # Сравниваем, что интерфейсы созданные на основе списков и на основе словарей дают одинаковые данные.
        self.assertEqual(str(Interfaces(interfaces_as_dict_list)), str(Interfaces(interfaces_list)))

    def test_interface_init_vlans(self):
        # Создание списка интерфейсов.
        interfaces_list = [
            ["eth1", "up", "description1", [11, 21]],
            ["eth2", "down", "description2", [12, 22]],
            ["eth3", "admin down", "description3", [13, 23]],
            ["eth4", "up", "description4", [14, 24]],
        ]
        # Создание нового экземпляра класса Interfaces.
        intf = Interfaces(interfaces_list)

        # Проверка того, что второй элемент существует и он верный.
        self.assertEqual(str(intf[1]), str(Interface(*interfaces_list[1])))
        # Проверяем, что количество интерфейсов равно 4.
        self.assertEqual(intf.count, 4)
        # Проверяем, что количество интерфейсов со статусом up равно 2.
        self.assertEqual(intf.up(only_count=True), 2)
        # Проверяем, что количество интерфейсов со статусом down равно 1.
        self.assertEqual(intf.down(only_count=True), 2)
        # Проверяем, что количество интерфейсов со статусом admin_down равно 1.
        self.assertEqual(intf.admin_down(only_count=True), 1)
        # Проверка того, что метод with_vlans возвращает правильный интерфейс.
        self.assertEqual(str(intf.with_vlans([11])), str(Interfaces([interfaces_list[0]])))
        # Проверка того, что метод filter_by_desc возвращает правильный ответ.
        self.assertEqual(str(intf.filter_by_desc("description")), str(intf))
        # Проверка того, что метод filter_by_desc возвращает правильный ответ.
        self.assertEqual(
            str(intf.filter_by_desc("description2")),
            str(Interfaces([interfaces_list[1]])),
        )
        # Проверка того, что метод filter_by_desc возвращает правильный ответ.
        # Для не существующего результата по фильтру описания
        self.assertEqual(str(intf.filter_by_desc("invalid")), str(Interfaces()))

        interfaces_dict = [
            {
                "Interface": "eth1",
                "Status": "up",
                "Description": "description1",
                "VLAN's": [11, 21],
            },
            {
                "Interface": "eth2",
                "Status": "down",
                "Description": "description2",
                "VLAN's": [12, 22],
            },
            {
                "Interface": "eth3",
                "Status": "admin down",
                "Description": "description3",
                "VLAN's": [13, 23],
            },
            {
                "Interface": "eth4",
                "Status": "up",
                "Description": "description4",
                "VLAN's": [14, 24],
            },
        ]
        # Сравниваем, что интерфейсы созданные на основе списков и на основе словарей дают одинаковые данные.
        self.assertEqual(str(Interfaces(interfaces_dict)), str(Interfaces(interfaces_list)))

    def test_interface_init_mixed_interface_list(self):
        # Создание списка интерфейсов.
        interfaces_list = [
            ["eth1", "up", "description1", [11, 21]],
            ["eth2", "down", "description2", [12, 22]],
            ["eth3", "admin down", "description3", [13, 23]],
            ["eth4", "up", "description4", [14, 24]],
        ]
        # Создание смешанного списка интерфейсов.
        interfaces_mixed = [
            ["eth1", "up", "description1", [11, 21]],
            {
                "Interface": "eth2",
                "Status": "down",
                "Description": "description2",
                "VLAN's": [12, 22],
            },
            ["eth3", "admin down", "description3", [13, 23]],
            {
                "Interface": "eth4",
                "Status": "up",
                "Description": "description4",
                "VLAN's": [14, 24],
            },
        ]
        # Сравниваем, что интерфейсы созданные на основе списков и на смешанной основе дают одинаковые данные.
        self.assertEqual(str(Interfaces(interfaces_mixed)), str(Interfaces(interfaces_list)))