Class TestGetModuleObject

tests/test_helpers.py:107–154

Test the _get_module_object helper function.

class TestGetModuleObject:
    """Test the _get_module_object helper function."""

    @pytest.fixture
    def mock_client(self):
        """Fixture providing a mock MSF client."""
        client = Mock()
        with patch('MetasploitMCP.get_msf_client', return_value=client):
            yield client

    @pytest.mark.asyncio
    async def test_get_module_object_success(self, mock_client):
        """Test successful module object retrieval."""
        mock_module = Mock()
        mock_client.modules.use.return_value = mock_module
        
        result = await _get_module_object('exploit', 'windows/smb/ms17_010_eternalblue')
        
        assert result is mock_module
        mock_client.modules.use.assert_called_once_with('exploit', 'windows/smb/ms17_010_eternalblue')

    @pytest.mark.asyncio
    async def test_get_module_object_full_path(self, mock_client):
        """Test module object retrieval with full path."""
        mock_module = Mock()
        mock_client.modules.use.return_value = mock_module
        
        result = await _get_module_object('exploit', 'exploit/windows/smb/ms17_010_eternalblue')
        
        assert result is mock_module
        # Should strip the module type prefix
        mock_client.modules.use.assert_called_once_with('exploit', 'windows/smb/ms17_010_eternalblue')

    @pytest.mark.asyncio
    async def test_get_module_object_not_found(self, mock_client):
        """Test module object retrieval when module not found."""
        mock_client.modules.use.side_effect = KeyError("Module not found")
        
        with pytest.raises(ValueError, match="not found"):
            await _get_module_object('exploit', 'nonexistent/module')

    @pytest.mark.asyncio
    async def test_get_module_object_msf_error(self, mock_client):
        """Test module object retrieval with MSF RPC error."""
        mock_client.modules.use.side_effect = MockMsfRpcError("RPC Error")
        
        with pytest.raises(MockMsfRpcError, match="RPC Error"):
            await _get_module_object('exploit', 'test/module')


class TestSetModuleOptions: