summaryrefslogtreecommitdiff
path: root/tests/messages_tests/utils.py
blob: f00ea4585b3b409f7aa5004ab2e1e3d322bf76e4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from django.contrib.messages import Message


class DummyStorage:
    """Dummy message-store to test the API methods."""

    def __init__(self):
        self.store = []

    def add(self, level, message, extra_tags=""):
        self.store.append(Message(level, message, extra_tags))

    def __iter__(self):
        return iter(self.store)