import unittest

from mwparserfromhell import parse

from wrpylib.lib_sledrun_wikitext_to_json import get_sledrun_description, optional_set


class TestOptionalSet(unittest.TestCase):
    def test_optional_set_string(self):
        h = {'a': 3}
        optional_set(h, 'b', 5)
        expected = {'a': 3, 'b': 5}
        self.assertDictEqual(expected, h)

    def test_optional_set_none(self):
        h = {'a': 3}
        expected = h.copy()
        optional_set(h, 'b', None)
        self.assertDictEqual(expected, h)


class TestGetSledrunDescription(unittest.TestCase):
    def test_get_sledrun_description(self):
        wikitext = "== Allgemeines ==\n{{Rodelbahnbox}}\nDas ist ''die'' Beschreibung.\n" + \
                    "Weitere Details bei https://example.com.\n\n* Position\n"
        wikicode = parse(wikitext)
        expected = "Das ist ''die'' Beschreibung.\nWeitere Details bei https://example.com."
        actual = get_sledrun_description(wikicode)
        self.assertEqual(expected, actual)