fix: Add get_port_by_mac for DellOS9

Fixes #4
Signed-off-by: Rafał Głombiowski <[email protected]>

Author: rafalglx

mfd_switchmanagement/vendors/dell/dell_os9/base.py

@@ -154,6 +154,33 @@ def default_ports(self, ports: str) -> None:
             ]
         )
 
+    def get_port_by_mac(self, mac: str) -> str:
+        """
+        Get port of switch with the specified MAC address.
+        :param mac: device MAC address in AA:BB:CC:DD:EE:FF form
+        :return: port name
+        :raises SwitchException: if port not found
+        :raises ValueError: if provided MAC address is incorrect
+        """
+        if self.is_mac_address(mac):
+            output = self._connection.send_command(f"show mac-address-table address {mac.lower()}")
+            """
+            GK6031-DR12-S6000-19505#show mac-address-table address 68:05:ca:c1:c8:ea
+            Codes: *N - VLT Peer Synced MAC
+            *I - Internal MAC Address used for Inter Process Communication
+            VlanId     Mac Address           Type          Interface        State
+             1      68:05:ca:c1:c8:ea       Dynamic         Te 0/32         Active
+            """
+            pattern = r"\s+\d+\s+(?:[0-9a-f]{2}:){5}[0-9a-f]{2}\s+\w+\s+(.+?)\s+Active"
+            match = re.search(pattern, output, re.IGNORECASE)
+
+            if match:
+                return match.group(1)  # return the port name
+            else:
+                raise SwitchException(f"Could not find port for MAC address {mac}")
+        else:
+            raise ValueError(f"Incorrect MAC address: {mac}")
+
     def get_vlan_by_mac(self, mac: str) -> int:
         """
         Get VLAN of port with the specified MAC address.

tests/unit/test_mfd_switchmanagement/test_vendors/test_dell/test_dell_os9/test_base.py

@@ -0,0 +1,47 @@
+# Copyright (C) 2025 Intel Corporation
+# SPDX-License-Identifier: MIT
+
+from pytest import fixture, raises
+from textwrap import dedent
+
+from mfd_switchmanagement import DellOS9
+from mfd_switchmanagement.exceptions import SwitchException
+
+
+class TestDellOS9:
+    """Class for DellOS9 tests."""
+
+    @fixture
+    def switch(self, mocker) -> DellOS9:
+        switch = DellOS9.__new__(DellOS9)
+        switch.__init__ = mocker.create_autospec(switch.__init__, return_value=None)
+        return switch
+
+    def test_get_port_by_mac(self, switch, mocker):
+        """Test get_port_by_mac method."""
+        switch._connection = mocker.Mock()
+        out = """
+        Codes: *N - VLT Peer Synced MAC
+        *I - Internal MAC Address used for Inter Process Communication
+        VlanId     Mac Address           Type          Interface        State
+         1      68:05:ca:c1:c8:ea       Dynamic         Te 0/32         Active
+        """
+        switch._connection.send_command = mocker.Mock(return_value=dedent(out))
+        assert switch.get_port_by_mac("68:05:ca:c1:c8:ea") == "Te 0/32"
+
+    def test_get_port_by_mac_not_found(self, switch, mocker):
+        """Test get_port_by_mac method when MAC address is not found."""
+        switch._connection = mocker.Mock()
+        out = """
+        Codes: *N - VLT Peer Synced MAC
+        *I - Internal MAC Address used for Inter Process Communication
+        VlanId     Mac Address           Type          Interface        State
+        """
+        switch._connection.send_command = mocker.Mock(return_value=dedent(out))
+        with raises(SwitchException, match="Could not find port for MAC address 68:05:ca:c1:c8:ea"):
+            switch.get_port_by_mac("68:05:ca:c1:c8:ea")
+
+    def test_get_port_by_mac_invalid_mac(self, switch):
+        """Test get_port_by_mac method with invalid MAC address."""
+        with raises(ValueError, match="Incorrect MAC address: ZZ:ZZ:ZZ:ZZ:ZZ:ZZ"):
+            switch.get_port_by_mac("ZZ:ZZ:ZZ:ZZ:ZZ:ZZ")