0
0
mirror of https://github.com/django/django.git synced 2024-11-27 17:16:33 +01:00
django/tests/messages_tests/utils.py

15 lines
341 B
Python

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)