thefuck/tests/rules/test_apt_unable_to_locate.py

from io import BytesIO

import pytest
from thefuck.rules.apt_unable_to_locate import match, get_new_command, _get_search_results
from thefuck.types import Command

invalid_operation = 'E: Unable to locate package {}'.format
apt_rabbitmq_search_results = b''' kamailio-rabbitmq-modules/bionic 5.1.2-1ubuntu2 amd64
  RabbitMQ and AMQP integration modules for the Kamailio SIP server

libanyevent-rabbitmq-perl/bionic 1.19+dfsg-1 all
  asynchronous and multi channel Perl AMQP client

libmojo-rabbitmq-client-perl/bionic 0.1.0-1 all
  Mojo::IOLoop based RabbitMQ client

libmono-messaging-rabbitmq4.0-cil/bionic 4.6.2.7+dfsg-1ubuntu1 all
  Mono Messaging RabbitMQ library (for CLI 4.0)

libmono-rabbitmq4.0-cil/bionic 4.6.2.7+dfsg-1ubuntu1 all
  Mono RabbitMQ.Client library (for CLI 4.0)

librabbitmq-client-java/bionic 5.0.0-1 all
  RabbitMQ Java client

librabbitmq-dbg/bionic-updates,bionic-security 0.8.0-1ubuntu0.18.04.2 amd64
  AMQP client library written in C - Debug Files

librabbitmq-dev/bionic-updates,bionic-security 0.8.0-1ubuntu0.18.04.2 amd64
  AMQP client library written in C - Dev Files

librabbitmq4/bionic-updates,bionic-security 0.8.0-1ubuntu0.18.04.2 amd64
  AMQP client library written in C

nagios-plugins-rabbitmq/bionic-updates 1:1.2.0-2.2ubuntu0.18.04.1 all
  Set of Nagios checks useful for monitoring a RabbitMQ server

opensips-rabbitmq-module/bionic 2.2.2-3build4 amd64
  Interface module to interact with a RabbitMQ server

puppet-module-puppetlabs-rabbitmq/bionic 5.3.1-2 all
  Puppet module for rabbitmq, manage everything from vhosts to exchanges

rabbitmq-server/bionic 3.6.10-1 all
  AMQP server written in Erlang

'''
apt_get_rabbitmq_search_results = b'''
kamailio-rabbitmq-modules - RabbitMQ and AMQP integration modules for the Kamailio SIP server
libanyevent-rabbitmq-perl - asynchronous and multi channel Perl AMQP client
libmojo-rabbitmq-client-perl - Mojo::IOLoop based RabbitMQ client
libmono-messaging-rabbitmq4.0-cil - Mono Messaging RabbitMQ library (for CLI 4.0)
libmono-rabbitmq4.0-cil - Mono RabbitMQ.Client library (for CLI 4.0)
librabbitmq-client-java - RabbitMQ Java client
librabbitmq-dbg - AMQP client library written in C - Debug Files
librabbitmq-dev - AMQP client library written in C - Dev Files
librabbitmq4 - AMQP client library written in C
nagios-plugins-rabbitmq - Set of Nagios checks useful for monitoring a RabbitMQ server
opensips-rabbitmq-module - Interface module to interact with a RabbitMQ server
puppet-module-puppetlabs-rabbitmq - Puppet module for rabbitmq, manage everything from vhosts to exchanges
rabbitmq-server - AMQP server written in Erlang
'''
rabbitmq_search_search_results = [
    'kamailio-rabbitmq-modules',
    'libanyevent-rabbitmq-perl',
    'libmojo-rabbitmq-client-perl',
    'libmono-messaging-rabbitmq4.0-cil',
    'libmono-rabbitmq4.0-cil',
    'librabbitmq-client-java',
    'librabbitmq-dbg',
    'librabbitmq-dev',
    'librabbitmq4',
    'nagios-plugins-rabbitmq',
    'opensips-rabbitmq-module',
    'puppet-module-puppetlabs-rabbitmq',
    'rabbitmq-server'
]


@pytest.mark.parametrize(
    'command',
    [
        (
            Command('apt install rabbitmq', invalid_operation('rabbitmq'))
        ),
        (
            Command('apt-get install rabbitmq', invalid_operation('rabbitmq'))
        )
    ]
)
def test_match(command):
    assert match(command)


@pytest.mark.parametrize(
    'command',
    [
        (
            Command('yarn install reactjs', 'a_bad_cmd: command not found')
        ),
        (
            Command('npm install reactjs', 'a_bad_cmd: command not found')
        ),
        (
            Command('apt upgrade', 'a_bad_cmd: command not found')
        )
    ]
)
def test_not_match(command):

    assert not match(command)


@pytest.fixture
def set_search(mocker):
    mock = mocker.patch('subprocess.Popen')

    def _set_text(text):
        mock.return_value.stdout = BytesIO(text)

    return _set_text


@pytest.mark.parametrize(
    'app, command, search_text, search_results',
    [
        ('apt', 'rabbitmq', apt_rabbitmq_search_results, rabbitmq_search_search_results),
        ('apt-get', 'rabbitmq', apt_get_rabbitmq_search_results, rabbitmq_search_search_results),
    ]
)
def test_get_search_results(set_search, app, command, search_text, search_results):
    set_search(search_text)
    assert _get_search_results(app, command) == search_results


@pytest.mark.parametrize(
    'command, expected_command, search_text',
    [
        (
            Command('sudo apt install rabbitmq', invalid_operation('rabbitmq')),
            [
                'sudo apt install librabbitmq4',
                'sudo apt install rabbitmq-server',
                'sudo apt install librabbitmq-dev'
            ],
            apt_rabbitmq_search_results,
        )
    ]
)
def test_get_new_command(set_search, command, expected_command, search_text):
    set_search(search_text)
    actual_command = get_new_command(command)
    assert actual_command == expected_command