# -*- coding: utf-8 -*-

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

"""

kallithea.tests.other.test_libs

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Package for testing various lib/helper functions in kallithea

This file was forked by the Kallithea project in July 2014.

Original author and date, and relevant copyright and licensing information is below:

:created_on: Jun 9, 2011

:author: marcink

:copyright: (c) 2013 RhodeCode GmbH, and others.

:license: GPLv3, see LICENSE.md for more details.

"""

import datetime

import hashlib

import mock

from kallithea.tests import *

from kallithea.lib.utils2 import AttributeDict

from kallithea.model.db import Repository

proto = 'http'

TEST_URLS = [

    ('%s://127.0.0.1' % proto, ['%s://' % proto, '127.0.0.1'],

     '%s://127.0.0.1' % proto),

    ('%s://username@127.0.0.1' % proto, ['%s://' % proto, '127.0.0.1'],

     '%s://127.0.0.1' % proto),

    ('%s://username:pass@127.0.0.1' % proto, ['%s://' % proto, '127.0.0.1'],

     '%s://127.0.0.1' % proto),

    ('%s://127.0.0.1:8080' % proto, ['%s://' % proto, '127.0.0.1', '8080'],

     '%s://127.0.0.1:8080' % proto),

    ('%s://example.com' % proto, ['%s://' % proto, 'example.com'],

     '%s://example.com' % proto),

    ('%s://user:pass@example.com:8080' % proto, ['%s://' % proto, 'example.com',

                                                '8080'],

     '%s://example.com:8080' % proto),

]

proto = 'https'

TEST_URLS += [

    ('%s://127.0.0.1' % proto, ['%s://' % proto, '127.0.0.1'],

     '%s://127.0.0.1' % proto),

    ('%s://username@127.0.0.1' % proto, ['%s://' % proto, '127.0.0.1'],

     '%s://127.0.0.1' % proto),

    ('%s://username:pass@127.0.0.1' % proto, ['%s://' % proto, '127.0.0.1'],

     '%s://127.0.0.1' % proto),

    ('%s://127.0.0.1:8080' % proto, ['%s://' % proto, '127.0.0.1', '8080'],

     '%s://127.0.0.1:8080' % proto),

    ('%s://example.com' % proto, ['%s://' % proto, 'example.com'],

     '%s://example.com' % proto),

    ('%s://user:pass@example.com:8080' % proto, ['%s://' % proto, 'example.com',

                                                '8080'],

     '%s://example.com:8080' % proto),

]

class TestLibs(BaseTestCase):

    @parameterized.expand(TEST_URLS)

    def test_uri_filter(self, test_url, expected, expected_creds):

        from kallithea.lib.utils2 import uri_filter

        self.assertEqual(uri_filter(test_url), expected)

    @parameterized.expand(TEST_URLS)

    def test_credentials_filter(self, test_url, expected, expected_creds):

        from kallithea.lib.utils2 import credentials_filter

        self.assertEqual(credentials_filter(test_url), expected_creds)

    @parameterized.expand([('t', True),

                           ('true', True),

                           ('y', True),

                           ('yes', True),

                           ('on', True),

                           ('1', True),

                           ('Y', True),

                           ('yeS', True),

                           ('Y', True),

                           ('TRUE', True),

                           ('T', True),

                           ('False', False),

                           ('F', False),

                           ('FALSE', False),

                           ('0', False),

                           ('-1', False),

                           ('', False)

    ])

    def test_str2bool(self, str_bool, expected):

        from kallithea.lib.utils2 import str2bool

        self.assertEqual(str2bool(str_bool), expected)

    def test_mention_extractor(self):

        from kallithea.lib.utils2 import extract_mentioned_users

        sample = (

            "@first hi there @world here's my email username@example.com "

            "@lukaszb check @one_more22 it pls @ ttwelve @D[] @one@two@three "

            "@UPPER    @cAmEL @2one_more22 @john please see this http://org.pl "

            "@marian.user just do it @marco-polo and next extract @marco_polo "

            "user.dot  hej ! not-needed maril@example.com"

        )

        s = sorted([

            '2one_more22', 'first', 'lukaszb', 'one', 'one_more22', 'UPPER', 'cAmEL', 'john',

            'marian.user', 'marco-polo', 'marco_polo', 'world'], key=lambda k: k.lower())

        self.assertEqual(s, extract_mentioned_users(sample))

    @parameterized.expand([

        (dict(), u'just now'),

        (dict(seconds= -1), u'1 second ago'),

        (dict(seconds= -60 * 2), u'2 minutes ago'),

        (dict(hours= -1), u'1 hour ago'),

        (dict(hours= -24), u'1 day ago'),

        (dict(hours= -24 * 5), u'5 days ago'),

        (dict(months= -1), u'1 month ago'),

        (dict(months= -1, days= -2), u'1 month and 2 days ago'),

        (dict(months= -1, days= -20), u'1 month and 19 days ago'),

        (dict(years= -1, months= -1), u'1 year and 1 month ago'),

        (dict(years= -1, months= -10), u'1 year and 10 months ago'),

        (dict(years= -2, months= -4), u'2 years and 4 months ago'),

        (dict(years= -2, months= -11), u'2 years and 11 months ago'),

        (dict(years= -3, months= -2), u'3 years and 2 months ago'),

    ])

    def test_age(self, age_args, expected):

        from kallithea.lib.utils2 import age

        from dateutil import relativedelta

        n = datetime.datetime(year=2012, month=5, day=17)

        delt = lambda *args, **kwargs: relativedelta.relativedelta(*args, **kwargs)

        self.assertEqual(age(n + delt(**age_args), now=n), expected)

    @parameterized.expand([

        (dict(), u'just now'),

        (dict(seconds= -1), u'1 second ago'),

        (dict(seconds= -60 * 2), u'2 minutes ago'),

        (dict(hours= -1), u'1 hour ago'),

        (dict(hours= -24), u'1 day ago'),

        (dict(hours= -24 * 5), u'5 days ago'),

        (dict(months= -1), u'1 month ago'),

        (dict(months= -1, days= -2), u'1 month ago'),

        (dict(months= -1, days= -20), u'1 month ago'),

        (dict(years= -1, months= -1), u'13 months ago'),

        (dict(years= -1, months= -10), u'22 months ago'),

        (dict(years= -2, months= -4), u'2 years ago'),

        (dict(years= -2, months= -11), u'3 years ago'),

        (dict(years= -3, months= -2), u'3 years ago'),

        (dict(years= -4, months= -8), u'5 years ago'),

    ])

    def test_age_short(self, age_args, expected):

        from kallithea.lib.utils2 import age

        from dateutil import relativedelta

        n = datetime.datetime(year=2012, month=5, day=17)

        delt = lambda *args, **kwargs: relativedelta.relativedelta(*args, **kwargs)

        self.assertEqual(age(n + delt(**age_args), show_short_version=True, now=n), expected)

    @parameterized.expand([

        (dict(), u'just now'),

        (dict(seconds=1), u'in 1 second'),

        (dict(seconds=60 * 2), u'in 2 minutes'),

        (dict(hours=1), u'in 1 hour'),

        (dict(hours=24), u'in 1 day'),

        (dict(hours=24 * 5), u'in 5 days'),

        (dict(months=1), u'in 1 month'),

        (dict(months=1, days=1), u'in 1 month and 1 day'),

        (dict(years=1, months=1), u'in 1 year and 1 month')

    ])

    def test_age_in_future(self, age_args, expected):

        from kallithea.lib.utils2 import age

        from dateutil import relativedelta

        n = datetime.datetime(year=2012, month=5, day=17)

        delt = lambda *args, **kwargs: relativedelta.relativedelta(*args, **kwargs)

        self.assertEqual(age(n + delt(**age_args), now=n), expected)

        sample = (

            "hello pta[tag] gog [[]] [[] sda ero[or]d [me =>>< sa]"

            "[requires] [stale] [see<>=>] [see => http://example.com]"

            "[requires => url] [lang => python] [just a tag]"

            "[,d] [ => ULR ] [obsolete] [desc]]"

        )

        from kallithea.lib.helpers import urlify_text

        res = urlify_text(sample, stylize=True)

        self.assertIn('<div class="metatag" tag="tag">tag</div>', res)

        self.assertIn('<div class="metatag" tag="obsolete">obsolete</div>', res)

        self.assertIn('<div class="metatag" tag="stale">stale</div>', res)

        self.assertIn('<div class="metatag" tag="lang">python</div>', res)

        self.assertIn('<div class="metatag" tag="requires">requires =&gt; <a href="/url">url</a></div>', res)

        self.assertIn('<div class="metatag" tag="tag">tag</div>', res)

    def test_alternative_gravatar(self):

        from kallithea.lib.helpers import gravatar_url

        _md5 = lambda s: hashlib.md5(s).hexdigest()

        #mock pylons.url

        class fake_url(object):

            @classmethod

            def current(cls, *args, **kwargs):

                return 'https://example.com'

        #mock pylons.tmpl_context

        def fake_tmpl_context(_url):

            _c = AttributeDict()

            _c.visual = AttributeDict()

            _c.visual.use_gravatar = True

            _c.visual.gravatar_url = _url

            return _c

        with mock.patch('pylons.url', fake_url):

            fake = fake_tmpl_context(_url='http://example.com/{email}')

            with mock.patch('pylons.tmpl_context', fake):

                    from pylons import url

                    assert url.current() == 'https://example.com'

                    grav = gravatar_url(email_address='test@example.com', size=24)

                    assert grav == 'http://example.com/test@example.com'

            fake = fake_tmpl_context(_url='http://example.com/{email}')

            with mock.patch('pylons.tmpl_context', fake):

                grav = gravatar_url(email_address='test@example.com', size=24)

                assert grav == 'http://example.com/test@example.com'

            fake = fake_tmpl_context(_url='http://example.com/{md5email}')

            with mock.patch('pylons.tmpl_context', fake):

                em = 'test@example.com'

                grav = gravatar_url(email_address=em, size=24)

                assert grav == 'http://example.com/%s' % (_md5(em))

            fake = fake_tmpl_context(_url='http://example.com/{md5email}/{size}')

            with mock.patch('pylons.tmpl_context', fake):

                em = 'test@example.com'

                grav = gravatar_url(email_address=em, size=24)

                assert grav == 'http://example.com/%s/%s' % (_md5(em), 24)

            fake = fake_tmpl_context(_url='{scheme}://{netloc}/{md5email}/{size}')

            with mock.patch('pylons.tmpl_context', fake):

                em = 'test@example.com'

                grav = gravatar_url(email_address=em, size=24)

                assert grav == 'https://example.com/%s/%s' % (_md5(em), 24)

    @parameterized.expand([

        (Repository.DEFAULT_CLONE_URI, 'group/repo1', {}, '', 'http://vps1:8000/group/repo1'),

        (Repository.DEFAULT_CLONE_URI, 'group/repo1', {'user': 'username'}, '', 'http://username@vps1:8000/group/repo1'),

        (Repository.DEFAULT_CLONE_URI, 'group/repo1', {}, '/prefix', 'http://vps1:8000/prefix/group/repo1'),

        (Repository.DEFAULT_CLONE_URI, 'group/repo1', {'user': 'user'}, '/prefix', 'http://user@vps1:8000/prefix/group/repo1'),

        (Repository.DEFAULT_CLONE_URI, 'group/repo1', {'user': 'username'}, '/prefix', 'http://username@vps1:8000/prefix/group/repo1'),

        (Repository.DEFAULT_CLONE_URI, 'group/repo1', {'user': 'user'}, '/prefix/', 'http://user@vps1:8000/prefix/group/repo1'),

        (Repository.DEFAULT_CLONE_URI, 'group/repo1', {'user': 'username'}, '/prefix/', 'http://username@vps1:8000/prefix/group/repo1'),

        ('{scheme}://{user}@{netloc}/_{repoid}', 'group/repo1', {}, '', 'http://vps1:8000/_23'),

        ('{scheme}://{user}@{netloc}/_{repoid}', 'group/repo1', {'user': 'username'}, '', 'http://username@vps1:8000/_23'),

        ('http://{user}@{netloc}/_{repoid}', 'group/repo1', {'user': 'username'}, '', 'http://username@vps1:8000/_23'),

        ('http://{netloc}/_{repoid}', 'group/repo1', {'user': 'username'}, '', 'http://vps1:8000/_23'),

        ('https://{user}@proxy1.example.com/{repo}', 'group/repo1', {'user': 'username'}, '', 'https://username@proxy1.example.com/group/repo1'),

        ('https://{user}@proxy1.example.com/{repo}', 'group/repo1', {}, '', 'https://proxy1.example.com/group/repo1'),

        ('https://proxy1.example.com/{user}/{repo}', 'group/repo1', {'user': 'username'}, '', 'https://proxy1.example.com/username/group/repo1'),

    ])

    def test_clone_url_generator(self, tmpl, repo_name, overrides, prefix, expected):

        from kallithea.lib.utils2 import get_clone_url

        clone_url = get_clone_url(uri_tmpl=tmpl, qualified_home_url='http://vps1:8000'+prefix,

                                  repo_name=repo_name, repo_id=23, **overrides)

        self.assertEqual(clone_url, expected)

    def _quick_url(self, text, tmpl="""<a class="revision-link" href="%s">%s</a>""", url_=None):

        """

        Changes `some text url[foo]` => `some text <a href="/">foo</a>

        :param text:

        """

        import re

        # quickly change expected url[] into a link

        URL_PAT = re.compile(r'(?:url\[)(.+?)(?:\])')

        def url_func(match_obj):

            _url = match_obj.groups()[0]

            return tmpl % (url_ or '/some-url', _url)

        return URL_PAT.sub(url_func, text)

    @parameterized.expand([

      ("",

       ""),

      ("git-svn-id: https://svn.apache.org/repos/asf/libcloud/trunk@1441655 13f79535-47bb-0310-9956-ffa450edef68",

       "git-svn-id: https://svn.apache.org/repos/asf/libcloud/trunk@1441655 13f79535-47bb-0310-9956-ffa450edef68"),

      ("from rev 000000000000",

       "from rev url[000000000000]"),

      ("from rev 000000000000123123 also rev 000000000000",

       "from rev url[000000000000123123] also rev url[000000000000]"),

      ("this should-000 00",

       "this should-000 00"),

      ("longtextffffffffff rev 123123123123",

       "longtextffffffffff rev url[123123123123]"),

      ("rev ffffffffffffffffffffffffffffffffffffffffffffffffff",

       "rev ffffffffffffffffffffffffffffffffffffffffffffffffff"),

      ("ffffffffffff some text traalaa",

       "url[ffffffffffff] some text traalaa"),

       ("""Multi line

       123123123123

       some text 123123123123

       sometimes !

       """,

       """Multi line

       url[123123123123]

       some text url[123123123123]

       sometimes !

       """)

    ])

    def test_urlify_changesets(self, sample, expected):

        def fake_url(self, *args, **kwargs):

            return '/some-url'

        expected = self._quick_url(expected)

        with mock.patch('pylons.url', fake_url):

            from kallithea.lib.helpers import urlify_changesets

            self.assertEqual(urlify_changesets(sample, 'repo_name'), expected)

    @parameterized.expand([

      ("",

       "",

       ""),

      ("https://svn.apache.org/repos",

       "url[https://svn.apache.org/repos]",

       "https://svn.apache.org/repos"),

      ("http://svn.apache.org/repos",

       "url[http://svn.apache.org/repos]",

       "http://svn.apache.org/repos"),

      ("from rev a also rev http://google.com",

       "from rev a also rev url[http://google.com]",

       "http://google.com"),

       ("""Multi line

       https://foo.bar.example.com

       some text lalala""",

       """Multi line

       url[https://foo.bar.example.com]

       some text lalala""",

       "https://foo.bar.example.com")

    ])

    def test_urlify_test(self, sample, expected, url_):

        from kallithea.lib.helpers import urlify_text

        expected = self._quick_url(expected,

                                   tmpl="""<a href="%s">%s</a>""", url_=url_)

        self.assertEqual(urlify_text(sample), expected)

    @parameterized.expand([

      ("", None),

      ("/_2", '2'),

      ("_2", '2'),

      ("/_2/", '2'),

      ("_2/", '2'),

      ("/_21", '21'),

      ("_21", '21'),

      ("/_21/", '21'),

      ("_21/", '21'),

      ("/_21/foobar", '21'),

      ("_21/121", '21'),

      ("/_21/_12", '21'),

      ("_21/prefix/foo", '21'),

    ])

    def test_get_repo_by_id(self, test, expected):

        from kallithea.lib.utils import _extract_id_from_repo_name

        _test = _extract_id_from_repo_name(test)

        self.assertEqual(_test, expected, msg='url:%s, got:`%s` expected: `%s`'

                                              % (test, _test, expected))