import unittest

from gemini import urljoin

class TestUrljoin(unittest.TestCase):

    def test_relative(self):
        self.assertEqual(urljoin('gemini://example.com:1965/foo/', '/bar/'),
                         'gemini://example.com:1965/bar/')
        self.assertEqual(urljoin('gemini://example.com:1965/foo/', 'bar/'),
                         'gemini://example.com:1965/foo/bar/')
        self.assertEqual(urljoin('gemini://example.com:1965/foo/', 'baz.gem'),
                         'gemini://example.com:1965/foo/baz.gem')
        self.assertEqual(urljoin('gemini://example.com:1965/foo/bar.gem', '/bar/'),
                         'gemini://example.com:1965/bar/')
        self.assertEqual(urljoin('gemini://example.com:1965/foo/bar.gem', 'bar/'),
                         'gemini://example.com:1965/foo/bar/')
        self.assertEqual(urljoin('gemini://example.com:1965/foo/bar.gem', 'baz.gem'),
                         'gemini://example.com:1965/foo/baz.gem')