thefuck/tests/rules/test_rm_root.py

import pytest
from thefuck.rules.rm_root import match, get_new_command
from tests.utils import Command


def test_match():
    assert match(Command(script='rm -rf /',
                         stderr='add --no-preserve-root'), None)


@pytest.mark.parametrize('command', [
    Command(script='ls', stderr='add --no-preserve-root'),
    Command(script='rm --no-preserve-root /', stderr='add --no-preserve-root'),
    Command(script='rm -rf /', stderr='')])
def test_not_match(command):
    assert not match(command, None)


def test_get_new_command():
    assert get_new_command(Command(script='rm -rf /'), None) \
           == 'rm -rf / --no-preserve-root'