Class TestMsfClientFunctions

tests/test_helpers.py:61–104

Test MSF client initialization and management functions.

class TestMsfClientFunctions:
    """Test MSF client initialization and management functions."""

    @patch('MetasploitMCP.MSF_PASSWORD', 'test-password')
    @patch('MetasploitMCP.MSF_SERVER', '127.0.0.1')
    @patch('MetasploitMCP.MSF_PORT_STR', '55553')
    @patch('MetasploitMCP.MSF_SSL_STR', 'false')
    def test_initialize_msf_client_success(self):
        """Test successful MSF client initialization."""
        with patch('MetasploitMCP._msf_client_instance', None):
            with patch('MetasploitMCP.MsfRpcClient') as mock_client_class:
                mock_client = Mock()
                mock_client.core.version = {'version': '6.3.0'}
                mock_client_class.return_value = mock_client
                
                result = initialize_msf_client()
                
                assert result is mock_client
                mock_client_class.assert_called_once_with(
                    password='test-password',
                    server='127.0.0.1', 
                    port=55553,
                    ssl=False
                )

    @patch('MetasploitMCP.MSF_PORT_STR', 'invalid-port')
    def test_initialize_msf_client_invalid_port(self):
        """Test MSF client initialization with invalid port."""
        with patch('MetasploitMCP._msf_client_instance', None):
            with pytest.raises(ValueError, match="Invalid MSF connection parameters"):
                initialize_msf_client()

    def test_get_msf_client_not_initialized(self):
        """Test get_msf_client when client not initialized."""
        with patch('MetasploitMCP._msf_client_instance', None):
            with pytest.raises(ConnectionError, match="not been initialized"):
                get_msf_client()

    def test_get_msf_client_initialized(self):
        """Test get_msf_client when client is initialized."""
        mock_client = Mock()
        with patch('MetasploitMCP._msf_client_instance', mock_client):
            result = get_msf_client()
            assert result is mock_client


class TestGetModuleObject: