Exemplo n.º 1
0
class TestEventPictureMethods(UnitTestBase):
    """
    Test methods which operate on the event's pictures.
    """
    def setUp(self):
        from kotti_events.resources import Event
        self.event = Event(u"Frankfurt", u"<b>Awesome!</b>",
                           datetime.now, None, None, None)

        from kotti_events.resources import EventPicture
        for name in ['foo', 'bar', 'cruz']:
            self.event[name] = EventPicture(name=name, mimetype=u'image/jpeg')

        self.emty_event = Event(u"Frankfurt", u"<b>Awesome!</b>",
                                datetime.now, None, None, None)

    def test_get_icon(self):
        icon = self.event.get_icon()
        assert icon.name == 'foo'

        icon = self.emty_event.get_icon()
        assert icon is None

    def test_get_pictures(self):
        pictures = list(self.event.get_pictures())
        assert len(pictures) == 3
        assert pictures[2].name == 'cruz'

        pictures = list(self.emty_event.get_pictures())
        assert len(pictures) == 0
Exemplo n.º 2
0
    def setUp(self):
        from kotti_events.resources import Event
        self.event = Event(u"Frankfurt", u"<b>Awesome!</b>",
                           datetime.now, None, None, None)

        from kotti_events.resources import EventPicture
        for name in ['foo', 'bar', 'cruz']:
            self.event[name] = EventPicture(name=name, mimetype=u'image/jpeg')

        self.emty_event = Event(u"Frankfurt", u"<b>Awesome!</b>",
                                datetime.now, None, None, None)