import os

from django.test.html import parse_html
from crispy_forms_foundation.layout import (Layout, ButtonGroup,
                                            Submit, Button,
                                            ButtonElement, ButtonSubmit)

from tests.forms import BasicInputForm
# from tests.utils import write_output


def test_buttongroup(output_test_path, render_output, rendered_template,
                     helper, client):
    form = BasicInputForm()
    pack = helper.template_pack

    helper.layout = Layout(
        'simple',
        ButtonGroup(
            Submit('Save', 'Save'),
            Button('Cancel', 'Cancel'),
        )
    )

    rendered = rendered_template(form, helper=helper)

    attempted = render_output(os.path.join(output_test_path, pack,
                                           "test_buttongroup.html"))
    # write_output(output_test_path, pack, "test_buttongroup.html", rendered)

    assert parse_html(attempted) == parse_html(rendered)


def test_buttonelement(output_test_path, render_output, rendered_template,
                       helper, client):
    form = BasicInputForm()
    pack = helper.template_pack

    helper.layout = Layout(
        ButtonSubmit('Save', 'Save'),
        ButtonElement('Foo', 'Foo', content="""<span>&lt;Pong/&gt;</span>"""),
    )

    rendered = rendered_template(form, helper=helper)

    attempted = render_output(os.path.join(output_test_path, pack,
                                           "test_buttonelement.html"))
    # write_output(output_test_path, pack, "test_buttonelement.html", rendered)

    assert attempted == rendered