Back to Repositories

Testing Sudo Command Support Decorator Implementation in thefuck

This test suite validates the sudo_support decorator functionality in thefuck project, ensuring proper handling of sudo commands and their non-sudo equivalents. The tests verify command transformations and return value preservation across different scenarios.

Test Coverage Overview

The test suite provides comprehensive coverage of the sudo_support decorator functionality.

Key areas tested include:

Command transformation with and without sudo prefix
Various return value types (strings, lists, booleans)
Command object handling
Edge cases with different input/output combinations

Implementation Analysis

The implementation uses pytest’s parametrize feature to efficiently test multiple scenarios with a single test function. The approach systematically validates the decorator’s behavior across different input patterns and return value types, using Command objects to represent shell commands.

Technical implementation details:

Parametrized test cases for comprehensive coverage
Mock function injection for command handling
Command object validation

Technical Details

Testing tools and configuration:

pytest framework for test execution
pytest.mark.parametrize for test case variation
Command class from thefuck.types
sudo_support decorator from thefuck.specific.sudo
Function mocking for command verification

Best Practices Demonstrated

The test implementation showcases several testing best practices for Python decorator validation.

Notable practices include:

Comprehensive parameter coverage using pytest.mark.parametrize
Clear separation of test cases
Explicit assertion checking
Clean mock function implementation
Efficient test organization with single responsibility principle

nvbn/thefuck

tests/specific/test_sudo.py

            
import pytest
from thefuck.specific.sudo import sudo_support
from thefuck.types import Command


@pytest.mark.parametrize('return_value, command, called, result', [
    ('ls -lah', 'sudo ls', 'ls', 'sudo ls -lah'),
    ('ls -lah', 'ls', 'ls', 'ls -lah'),
    (['ls -lah'], 'sudo ls', 'ls', ['sudo ls -lah']),
    (True, 'sudo ls', 'ls', True),
    (True, 'ls', 'ls', True),
    (False, 'sudo ls', 'ls', False),
    (False, 'ls', 'ls', False)])
def test_sudo_support(return_value, command, called, result):
    def fn(command):
        assert command == Command(called, '')
        return return_value

    assert sudo_support(fn)(Command(command, '')) == result