@@ -19,3 +19,30 @@ def test_Logger_called_with_no_args_doesnt_raise_TypeError(self):
1919logger ("a message" )
2020except TypeError :
2121self .fail ("logger raised TypeError unexpectedly" )
22+
23+
24+ class LoggerMessageHandlerTests (unittest .TestCase ):
25+ def setUp (self ):
26+ self .message = ""
27+ self .logger = Logger ()
28+ self .logger .message = self ._echo
29+
30+ def _echo (self ,message ):
31+ self .message += message
32+
33+ def test_message_handler_with_no_short_message (self ):
34+ """Test that message handler is called"""
35+ self .logger ("a message" )
36+ self .assertEqual (self .message ,"a message" )
37+
38+ def test_only_short_True (self ):
39+ """Test that only_short=True prints only the short message"""
40+ self .logger .only_short = True
41+ self .logger ("a long message" ,"a short message" )
42+ self .assertEqual (self .message ,"a short message" )
43+
44+ def test_only_short_False (self ):
45+ """Test that only_short=False prints both messages"""
46+ self .logger .only_short = False
47+ self .logger ("a long message" ,"a short message" )
48+ self .assertEqual (self .message ,"a long messagea short message" )