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


@pytest.mark.parametrize('command', [
    Command(script='open foo.com'),
    Command(script='open foo.ly'),
    Command(script='open foo.org'),
    Command(script='open foo.net'),
    Command(script='open foo.se'),
    Command(script='open foo.io'),
    Command(script='xdg-open foo.com'),
    Command(script='gnome-open foo.com'),
    Command(script='kde-open foo.com')])
def test_match(command):
    assert match(command)


@pytest.mark.parametrize('command, new_command', [
    (Command('open foo.com'), 'open http://foo.com'),
    (Command('open foo.ly'), 'open http://foo.ly'),
    (Command('open foo.org'), 'open http://foo.org'),
    (Command('open foo.net'), 'open http://foo.net'),
    (Command('open foo.se'), 'open http://foo.se'),
    (Command('open foo.io'), 'open http://foo.io'),
    (Command('xdg-open foo.io'), 'xdg-open http://foo.io'),
    (Command('gnome-open foo.io'), 'gnome-open http://foo.io'),
    (Command('kde-open foo.io'), 'kde-open http://foo.io')])
def test_get_new_command(command, new_command):
    assert get_new_command(command) == new_command