import unittest
from markdownify import markdownify as md


class BasicTests(unittest.TestCase):

    def test_single_tag(self):
        self.assertEqual(md('<span>Hello</span>'), 'Hello')

    def test_soup(self):
        self.assertEqual(md('<div><span>Hello</div></span>'), 'Hello')

    def test_whitespace(self):
        self.assertEqual(md(' a  b \n\n c '), ' a b c ')


class ArgTests(unittest.TestCase):

    def test_strip(self):
        self.assertEqual(
            md('<a href="https://github.com/matthewwithanm">Some Text</a>', strip=['a']),
            'Some Text')

    def test_do_not_strip(self):
        self.assertEqual(
            md('<a href="https://github.com/matthewwithanm">Some Text</a>', strip=[]),
            '[Some Text](https://github.com/matthewwithanm)')

    def test_convert(self):
        self.assertEqual(
            md('<a href="https://github.com/matthewwithanm">Some Text</a>', convert=['a']),
            '[Some Text](https://github.com/matthewwithanm)')

    def test_do_not_convert(self):
        self.assertEqual(
            md('<a href="https://github.com/matthewwithanm">Some Text</a>', convert=[]),
            'Some Text')


class EscapeTests(unittest.TestCase):

    def test_underscore(self):
        self.assertEqual(md('_hey_dude_'), '\_hey\_dude\_')

    def test_xml_entities(self):
        self.assertEqual(md('&amp;'), '&')

    def test_named_entities(self):
        self.assertEqual(md('&raquo;'), u'\xbb')

    def test_hexadecimal_entities(self):
        # This looks to be a bug in BeautifulSoup (fixed in bs4) that we have to work around.
        self.assertEqual(md('&#x27;'), '\x27')

    def test_single_escaping_entities(self):
        self.assertEqual(md('&amp;amp;'), '&amp;')


class ConversionTests(unittest.TestCase):

    def test_a(self):
        self.assertEqual(
            md('<a href="http://google.com">Google</a>'),
            '[Google](http://google.com)'
        )

    def test_a_with_title(self):
        self.assertEqual(
            md('<a href="http://google.com" title="The &quot;Goog&quot;">Google</a>'),
            r'[Google](http://google.com "The \"Goog\"")'
        )

    def test_b(self):
        self.assertEqual(md('<b>Hello</b>'), '**Hello**')

    def test_blockquote(self):
        self.assertEqual(md('<blockquote>Hello</blockquote>').strip(), '> Hello')

    def test_nested_blockquote(self):
        self.assertEqual(
            md('<blockquote>And she was like <blockquote>Hello</blockquote></blockquote>').strip(),
            '> And she was like \n> > Hello'
        )

    def test_br(self):
        self.assertEqual(md('a<br />b<br />c'), 'a  \nb  \nc')

    def test_em(self):
        self.assertEqual(md('<em>Hello</em>'), '*Hello*')

    def test_h1(self):
        self.assertEqual(md('<h1>Hello</h1>'), 'Hello\n=====\n\n')

    def test_h2(self):
        self.assertEqual(md('<h2>Hello</h2>'), 'Hello\n-----\n\n')

    def test_hn(self):
        self.assertEqual(md('<h3>Hello</h3>'), '### Hello\n\n')
        self.assertEqual(md('<h6>Hello</h6>'), '###### Hello\n\n')

    def test_i(self):
        self.assertEqual(md('<i>Hello</i>'), '*Hello*')

    def test_ol(self):
        self.assertEqual(md('<ol><li>a</li><li>b</li></ol>'), '1. a\n2. b\n')

    def test_p(self):
        self.assertEqual(md('<p>hello</p>'), 'hello\n\n')

    def test_strong(self):
        self.assertEqual(md('<strong>Hello</strong>'), '**Hello**')

    def test_ul(self):
        self.assertEqual(md('<ul><li>a</li><li>b</li></ul>'), '* a\n* b\n')


class AdvancedTests(unittest.TestCase):

    def test_nested(self):
        self.assertEqual(
            md('<p>This is an <a href="http://example.com/">example link</a>.</p>'),
            'This is an [example link](http://example.com/).\n\n'
        )