11import pytest
22from django .contrib .auth .models import AnonymousUser
3- from django .contrib .messages import ERROR , INFO , SUCCESS , WARNING
3+ from django .contrib .messages import ERROR , INFO , SUCCESS
44from django .contrib .messages .storage .base import Message
55from django .test import RequestFactory
66from django .utils .safestring import mark_safe
@@ -73,7 +73,6 @@ def dummy_request(self):
7373 request ._messages = [
7474 Message (message = "abc" , level = INFO ),
7575 Message (message = "def" , level = SUCCESS ),
76- Message (message = "warning!" , level = WARNING ),
7776 Message (message = "error!!!" , level = ERROR ),
7877 ]
7978 return request
@@ -86,14 +85,10 @@ def test_success_banner_with_text_message(self, dummy_request):
8685 result = get_notification_banner_params (dummy_request , "success" )
8786 assert result == {"text" : "def" , "type" : "success" , "role" : "alert" }
8887
89- def test_warning_banner_with_text_message (self , dummy_request ):
90- result = get_notification_banner_params (dummy_request , "warning" )
91- assert result == {"text" : "warning!" , "type" : "warning" , "role" : "alert" }
92-
9388 def test_invalid_message_type (self , dummy_request ):
9489 with pytest .raises (
9590 ValueError ,
96- match = "message_type must be one of {info, warning, success}; got error" ,
91+ match = "message_type must be one of {info, success}; got error" ,
9792 ):
9893 get_notification_banner_params (dummy_request , "error" )
9994
0 commit comments