Class TestSetModuleOptions

tests/test_helpers.py:157–208

Test the _set_module_options helper function.

class TestSetModuleOptions:
    """Test the _set_module_options helper function."""

    @pytest.fixture
    def mock_module(self):
        """Fixture providing a mock module object."""
        module = Mock()
        module.fullname = 'exploit/test/module'
        module.__setitem__ = Mock()
        return module

    @pytest.mark.asyncio
    async def test_set_module_options_basic(self, mock_module):
        """Test basic option setting."""
        options = {'RHOSTS': '192.168.1.1', 'RPORT': '80'}

        await _set_module_options(mock_module, options)

        # Should be called twice, once for each option
        assert mock_module.__setitem__.call_count == 2
        mock_module.__setitem__.assert_any_call('RHOSTS', '192.168.1.1')
        mock_module.__setitem__.assert_any_call('RPORT', 80)  # Type conversion: '80' -> 80

    @pytest.mark.asyncio
    async def test_set_module_options_type_conversion(self, mock_module):
        """Test option setting with type conversion."""
        options = {
            'RPORT': '80',  # String number -> int
            'SSL': 'true',  # String boolean -> bool
            'VERBOSE': 'false',  # String boolean -> bool
            'THREADS': '10'  # String number -> int
        }
        
        await _set_module_options(mock_module, options)
        
        # Verify type conversions
        calls = mock_module.__setitem__.call_args_list
        call_dict = {call[0][0]: call[0][1] for call in calls}
        
        assert call_dict['RPORT'] == 80
        assert call_dict['SSL'] is True
        assert call_dict['VERBOSE'] is False
        assert call_dict['THREADS'] == 10

    @pytest.mark.asyncio
    async def test_set_module_options_error(self, mock_module):
        """Test option setting with error."""
        mock_module.__setitem__.side_effect = KeyError("Invalid option")
        options = {'INVALID_OPT': 'value'}
        
        with pytest.raises(ValueError, match="Failed to set option"):
            await _set_module_options(mock_module, options)


class TestGetMsfConsole: