From 0f62c1a3d9958191671b06e37f61cbfdda5c0f52 Mon Sep 17 00:00:00 2001 From: Simon Oliver Tveit Date: Wed, 30 Aug 2023 09:56:53 +0200 Subject: [PATCH] Add fixtures --- tests/unittests/portadmin/portadmin_test.py | 53 +++++++++++++++++++++ 1 file changed, 53 insertions(+) diff --git a/tests/unittests/portadmin/portadmin_test.py b/tests/unittests/portadmin/portadmin_test.py index 024d22c357..e018110ac8 100644 --- a/tests/unittests/portadmin/portadmin_test.py +++ b/tests/unittests/portadmin/portadmin_test.py @@ -1,6 +1,7 @@ from mock import Mock import unittest +import pytest from nav.oids import OID from nav.enterprise.ids import VENDOR_ID_HEWLETT_PACKARD, VENDOR_ID_CISCOSYSTEMS @@ -10,6 +11,58 @@ from nav.portadmin.vlan import FantasyVlan +@pytest.fixture +def profile(): + profile = Mock() + profile.snmp_version = 2 + profile.snmp_community = "public" + return profile + + +@pytest.fixture +def netbox_hp(profile): + vendor = Mock() + vendor.id = u'hp' + + netbox_type = Mock() + netbox_type.vendor = vendor + netbox_type.get_enterprise_id.return_value = VENDOR_ID_HEWLETT_PACKARD + + netbox = Mock() + netbox.type = netbox_type + netbox.ip = '10.240.160.39' + netbox.get_preferred_snmp_management_profile.return_value = profile + + return netbox + + +@pytest.fixture +def netbox_cisco(profile): + vendor = Mock() + vendor.id = u'cisco' + + netbox_type = Mock() + netbox_type.vendor = vendor + netbox_type.get_enterprise_id.return_value = VENDOR_ID_CISCOSYSTEMS + + netbox = Mock() + netbox.type = netbox_type + netbox.ip = '10.240.160.38' + netbox.get_preferred_snmp_management_profile.return_value = profile + + return netbox + + +@pytest.fixture +def handler_hp(netbox_hp): + return ManagementFactory.get_instance(netbox_hp) + + +@pytest.fixture +def handler_cisco(netbox_cisco): + return ManagementFactory.get_instance(netbox_cisco) + + class PortadminResponseTest(unittest.TestCase): def setUp(self): self.profile = Mock()