from __future__ import unicode_literals

import pytest

URLS = [
    u"http://www.nytimes.com/2016/04/20/us/politics/new-york-primary-preview.html?hp&action=click&pgtype=Homepage&clickSource=story-heading&module=a-lede-package-region&region=top-news&WT.nav=top-news&_r=0",
    u"www.google.com?q=google",
    u"google.com",
    u"www.red-stars.com",
    pytest.mark.xfail(u"red-stars.com"),
    u"http://foo.com/blah_(wikipedia)#cite-1",
    u"http://www.example.com/wpstyle/?bar=baz&inga=42&quux",
    u"mailto:foo.bar@baz.com",
    u"mailto:foo-bar@baz-co.com"
]

# Punctuation we want to check is split away before the URL
PREFIXES = [
    "(", '"', "...",  ">"
]

# Punctuation we want to check is split away after the URL
SUFFIXES = [
    '"', ":", ">"]

@pytest.mark.parametrize("text", URLS)
def test_simple_url(en_tokenizer, text):
    tokens = en_tokenizer(text)
    assert tokens[0].orth_ == text
    assert len(tokens) == 1


@pytest.mark.parametrize("prefix", PREFIXES)
@pytest.mark.parametrize("url", URLS)
def test_prefixed_url(en_tokenizer, prefix, url):
    tokens = en_tokenizer(prefix + url)
    assert tokens[0].text == prefix
    assert tokens[1].text == url
    assert len(tokens) == 2
    
@pytest.mark.parametrize("suffix", SUFFIXES)
@pytest.mark.parametrize("url", URLS)
def test_suffixed_url(en_tokenizer, url, suffix):
    tokens = en_tokenizer(url + suffix)
    assert tokens[0].text == url
    assert tokens[1].text == suffix
    assert len(tokens) == 2
    
@pytest.mark.parametrize("prefix", PREFIXES)
@pytest.mark.parametrize("suffix", SUFFIXES)
@pytest.mark.parametrize("url", URLS)
def test_surround_url(en_tokenizer, prefix, suffix, url):
    tokens = en_tokenizer(prefix + url + suffix)
    assert tokens[0].text == prefix
    assert tokens[1].text == url
    assert tokens[2].text == suffix
    assert len(tokens) == 3
    
@pytest.mark.parametrize("prefix1", PREFIXES)
@pytest.mark.parametrize("prefix2", PREFIXES)
@pytest.mark.parametrize("url", URLS)
def test_two_prefix_url(en_tokenizer, prefix1, prefix2, url):
    tokens = en_tokenizer(prefix1 + prefix2 + url)
    assert tokens[0].text == prefix1
    assert tokens[1].text == prefix2
    assert tokens[2].text == url
    assert len(tokens) == 3
    
@pytest.mark.parametrize("suffix1", SUFFIXES)
@pytest.mark.parametrize("suffix2", SUFFIXES)
@pytest.mark.parametrize("url", URLS)
def test_two_prefix_url(en_tokenizer, suffix1, suffix2, url):
    tokens = en_tokenizer(url + suffix1 + suffix2)
    assert tokens[0].text == url
    assert tokens[1].text == suffix1
    assert tokens[2].text == suffix2
    assert len(tokens) == 3