Add more tests

tests/unittests/portadmin/portadmin_poe_cisco_test.py

@@ -58,3 +58,20 @@ def test_use_interfaces_from_db_if_empty_interfaces_arg(self, handler_cisco):
  handler_cisco.netbox.interfaces = [interface]
  state = handler_cisco.get_poe_states()
  assert interface.ifname in state
+
+
+class TestGetPoeIndexesForInterface:
+ def test_returns_correct_values(self, handler_cisco):
+ poegroup_mock = Mock(index=1)
+ poeport_mock = Mock(poegroup=poegroup_mock, index=2)
+ handler_cisco._get_poeport = Mock(return_value=poeport_mock)
+ interface = Mock(ifname="interface")
+ unit_number, if_number = handler_cisco._get_poe_indexes_for_interface(interface)
+ assert unit_number == poegroup_mock.index
+ assert if_number == poeport_mock.index
+
+ def test_raises_exception_if_no_poeport_exists(self, handler_cisco):
+ handler_cisco._get_poeport = Mock(side_effect=POEIndexNotFoundError("Fail"))
+ interface = Mock(ifname="interface")
+ with pytest.raises(POEIndexNotFoundError):
+ handler_cisco._get_poe_indexes_for_interface(interface)