from charmhelpers.contrib.templating import jinja

from etcd_lib import render_grafana_dashboard


def test_render_grafana_dashboard():
    """Test loading of Grafana dashboard."""
    datasource = 'prometheus'
    raw_template = ('{{"panels": [{{"datasource": "{} - '
                    'Juju generated source"}}]}}'.format(datasource))
    expected_dashboard = {
        'panels': [
            {'datasource': '{} - Juju generated source'.format(datasource)}
        ]}

    jinja.render.return_value = raw_template
    rendered_dashboard = render_grafana_dashboard(datasource)

    assert rendered_dashboard == expected_dashboard