import logging
from pprint import pformat

import pytest

from pbcommand.models.report import PlotGroup, Plot, PbReportError

log = logging.getLogger(__name__)


class TestPlotGroup:

    def test_init(self):
        """Test constructor with kwargs"""
        plot = Plot('a_plot', 'path/to/image.png', caption="My Image")
        p = PlotGroup('my_pg', plots=[plot])
        assert p is not None

    def test_plotgroup_null_id(self):
        """Can't create an plotGroup without an id."""
        def _test():
            p = PlotGroup(None)

        with pytest.raises(PbReportError):
            _test()

    def test_plotgroup_add_duplicate_plot(self):
        """Can't add plots with duplicate ids."""
        def _test():
            pg = PlotGroup('foo')
            pg.add_plot(Plot('id', 'i1'))
            pg.add_plot(Plot('id', 'i2'))

        with pytest.raises(PbReportError):
            _test()

    def test_to_dict(self):
        """Test plotGroup to_dict function."""
        a = PlotGroup('123', title='foo title', legend='foo legend',
                      thumbnail='foo thumbnail')
        a.add_plot(Plot('id', 'i1', caption='a caption'))

        d = a.to_dict()
        log.debug(pformat(d))

        assert '123' == d['id']
        assert 'foo title' == d['title']
        assert 'foo legend' == d['legend']
        assert 'foo thumbnail' == d['thumbnail']
        assert 1 == len(d['plots'])
        log.info(a)
        assert repr(a) is not None

    def test_adding_incorrect_type(self):
        """Validate type when adding Plots."""
        def _test():
            plots = ['Not a plot instance', 'Another bad plot.']
            p = PlotGroup('my_plotgroup', plots=plots)

        with pytest.raises(TypeError):
            _test()